当前位置：首页 > article >正文

重构代码之提取子类

article 2024/12/27 1:44:15

提取子类主要用于精简类结构和增强代码的灵活性。这个方法适用于当类中的某些字段或方法只适用于特定对象的情况。通过提取子类，可以将这些特定的字段和方法分离到一个新的子类中，使代码结构更清晰。

一、为什么要使用提取子类

提高代码的可读性：当类包含与特定需求无关的字段和方法时，类会显得冗杂。提取子类后，代码的职责会变得更加清晰。
增强扩展性和复用性：将特定功能放入子类，便于在不修改原始类的情况下进行扩展。
减少条件判断：避免主类中使用大量条件语句（如 if 和 switch）来处理不同的对象类型。

二、示例

假设我们有一个 Employee 类，其中包含了部分只适用于销售人员的属性和方法，例如 salesTarget 和 achieveSalesTarget() 方法：

public class Employee
{
    public string Name { get; set; }
    public decimal Salary { get; set; }
    public decimal SalesTarget { get; set; }

    public Employee(string name, decimal salary)
    {
        Name = name;
        Salary = salary;
    }

    public void AchieveSalesTarget()
    {
        Console.WriteLine($"{Name} 已经完成了销售目标");
    }
}

对于非销售人员的员工对象，这些特定的字段和方法可能毫无意义。此时我们可以考虑提取一个 SalesEmployee 子类，来包含这些特定于销售的逻辑。

2.1 重构后的代码

通过提取 SalesEmployee 子类，将销售相关的内容与一般员工类分离：

public class Employee
{
    public string Name { get; set; }
    public decimal Salary { get; set; }

    public Employee(string name, decimal salary)
    {
        Name = name;
        Salary = salary;
    }
}

public class SalesEmployee : Employee
{
    public decimal SalesTarget { get; set; }

    public SalesEmployee(string name, decimal salary, decimal salesTarget)
        : base(name, salary)
    {
        SalesTarget = salesTarget;
    }

    public void AchieveSalesTarget()
    {
        Console.WriteLine($"{Name} has achieved the sales target!");
    }
}