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

重构复杂简单变量之用类替换类型码

article 2024/10/24 19:12:47

用类替换类型码 是一种重构手法，用于将代码中使用的类型码（Type Code）替换为类。这种类型码通常是一些预定义的常量（如 int、string 等），表示不同的类型、状态或类别。使用类代替类型码有助于减少错误，增强代码的可读性和可扩展性。

一、什么时候使用

当代码中存在多处判断逻辑依赖于类型码，比如使用 if-else 或 switch-case 语句来处理不同的类型。
当类型码携带的不只是标识，还需要附加行为或数据。
当类型码可能会在未来增加新的类型或需要更多扩展。

二、优化步骤

识别类型码：首先找到代码中使用类型码（常常是 int、string 或枚举）来表示不同类别的地方。
创建类封装类型码：为每个类型码创建一个类，封装类型码所代表的行为和数据。
替换类型码：将代码中所有使用类型码的地方替换为类的实例。
移植行为：将与类型码相关的行为移动到新创建的类中。

三、示例

假设你有一个类 Employee，其中有一个类型码 EmployeeType 来表示员工的类型（如 1 表示全职员工，2 表示兼职员工）：

public class Employee
{
    public int EmployeeType { get; set; } // 1: Full-time, 2: Part-time

    public decimal GetBonus()
    {
        if (EmployeeType == 1)
        {
            return 1000;
        }
        else if (EmployeeType == 2)
        {
            return 500;
        }
        else
        {
            throw new ArgumentException("Invalid employee type.");
        }
    }
}

4.1 优化后的代码

首先为每种员工类型创建一个类，并将行为迁移到类中：

public abstract class EmployeeType
{
    public abstract decimal GetBonus();
}

public class FullTimeEmployee : EmployeeType
{
    public override decimal GetBonus()
    {
        return 1000;
    }
}

public class PartTimeEmployee : EmployeeType
{
    public override decimal GetBonus()
    {
        return 500;
    }
}

然后将 Employee 类中的类型码替换为 EmployeeType 对象：

public class Employee
{
    public EmployeeType Type { get; set; }

    public Employee(EmployeeType type)
    {
        Type = type;
    }

    public decimal GetBonus()
    {
        return Type.GetBonus();
    }
}

在客户端代码中使用时：

var fullTimeEmployee = new Employee(new FullTimeEmployee());
var partTimeEmployee = new Employee(new PartTimeEmployee());

Console.WriteLine(fullTimeEmployee.GetBonus()); // 输出 1000
Console.WriteLine(partTimeEmployee.GetBonus()); // 输出 500