码迷,mamicode.com
首页 > 其他好文 > 详细

重构:越来越长的 switch ... case 和 if ... else if ... else

时间:2015-11-29 18:02:05      阅读:224      评论:0      收藏:0      [点我收藏+]

标签:

在代码中,时常有就一类型码(Type Code)而展开的如 switch ... case 或 if ... else if ... else 的条件表达式。随着项目业务逻辑的增加及代码经年累月的修改,这些条件判断逻辑往往变得越来越冗长。特别是当同样的逻辑判断出现在多个地方的时候(结构示意如下),代码的可读性和维护难易程度将变得非常的糟糕。每次修改时,你必须找到所有有逻辑分支的地方,并修改它们。

 1 switch(type)
 2 {
 3     case "1":
 4         ...
 5         break;
 6     case "2":
 7         ...
 8         break;
 9     case default:
10         ...
11         break;
12 }
13 
14 ... ...
15 ... ...
16 
17 switch(type)
18 {
19     case "1":
20         ...
21         break;
22     case "2":
23         ...
24         break;
25     case default:
26         ...
27         break;
28 }

当代码中出现这样情况的时候,你就应考虑该重构它了(又有一种说法:要极力的重构 switch ... case 语句,避免它在你的代码中出现)。

 

重构掉类型码(Type Code)判断的条件表达式,我们需要引入面向对象的多态机制。第一种方式:使用子类来替换条件表达式

我们以计算不同 role 的员工薪水的 Employee 类为例,一步步讲解重构的过程。原始待重构的类结构如下:

技术分享

 1 public class Employee_Ori
 2 {
 3     private int _type;
 4 
 5     private const int ENGINEER = 1;
 6     private const int SALESMAN = 2;
 7     private const int MANAGER = 3;
 8 
 9     private decimal _baseSalary = 10000;
10     private decimal _royalty = 100;
11     private decimal _bonus = 400;
12 
13     public Employee_Ori(int type)
14     {
15         this._type = type;
16     }
17 
18     public decimal getEmployeeSalary()
19     {
20         var monthlySalary = 0m;
21 
22         switch (this._type)
23         { 
24             case ENGINEER:
25                 monthlySalary = _baseSalary;
26                 break;
27             case SALESMAN:
28                 monthlySalary = _baseSalary + _royalty;
29                 break;
30             case MANAGER:
31                 monthlySalary = _baseSalary + _bonus;
32                 break;
33         }
34 
35         return monthlySalary;
36     }
37 }

其中方法 getEmployeeSalary() 根据员工不同的角色返回当月薪水,当然,此处逻辑仅为示意,勿深究。

用子类方法重构后的类结构是这样的:

技术分享

下面详细展开。首先,以类型码(Type Code)的宿主类 Employee 类为基类,为每个角色建立子类。

 1 public class Engineer_Sub : Employee_Sub
 2 {
 3     ...
 4 }
 5 
 6 public class Salesman_Sub : Employee_Sub
 7 {
 8     ...
 9 }
10 
11 public class Manager_Sub : Employee_Sub
12 {
13     ...
14 }

同时,将 Employee 基类的构造函数改造为静态 Create 工厂函数。

 1 public static Employee_Sub Create(int type)
 2 {
 3     Employee_Sub employee = null;
 4 
 5     switch (type)
 6     {
 7         case ENGINEER:
 8             employee = new Engineer_Sub();
 9             break;
10         case SALESMAN:
11             employee = new Salesman_Sub();
12             break;
13         case MANAGER:
14             employee = new Manager_Sub();
15             break;
16     }
17 
18     return employee;
19 }

该静态工厂方法中也含有一 switch 逻辑。而重构后的代码中 switch 判断只有这一处,且只在对象创建的过程中涉及,不牵扯任何的业务逻辑,所以是可以接受的。

每个角色子类均覆写基类中的 getEmployeeSalary() 方法,应用自己的规则来计算薪水。

1 private decimal _baseSalary = 10000;
2 
3 public override decimal getEmployeeSalary()
4 {
5     return _baseSalary;
6 }

这样,基类 Employee 中的 getEmployeeSalary() 方法已无实际意义,将其变为 abstract 方法。

1 public abstract decimal getEmployeeSalary();

子类已经完全取代了臃肿的 switch 表达式。如果 Engineer, Salesman 或者 Manager 有新的行为,可在各自的子类中添加方法。或后续有新的 Employee Type, 完全可以通过添加新的子类来实现。在这里,通过多态实现了服务与其使用这的分离。

 

但是,在一些情况下,如对象类型在生命周期中需要变化(细化到本例,如 Engineer 晋升为 Manager)或者类型宿主类已经有子类,则使用子类重构的办法就无法奏效了。这时应用第二种方法:用 State/Strategy 来取代条件表达式

同样,先上一张该方法重构后的类结构:

技术分享

抽出一 EmployeeType 类,类型码的宿主类 Employee 对其进行引用。

1 public class Employee_State
2 {
3     // employee‘s type can be changed
4     private EmployeeType_State _employeeType = null;
5 }

EmployeeType 为基类,具体员工类型作为其子类。EmployeeType 类中含有一创建员工类型的静态工厂方法 Create。

 1 public static EmployeeType_State Create(int type)
 2 {
 3     EmployeeType_State empType = null;
 4 
 5     switch (type)
 6     {
 7         case ENGINEER:
 8             empType = new EngineerType_State();
 9             break;
10         case SALESMAN:
11             empType = new SalesmanType_State();
12             break;
13         case MANAGER:
14             empType = new ManagerType_State();
15             break;
16     }
17 
18     return empType;
19 }

将员工薪水的计算方法 getEmployeeSalary() 从 Employee 类迁徙到员工类型基类 EmployeeType 中。各具体员工类型类均 override 该方法。考虑到 EmployeeType 已无具体业务逻辑意义,将 EmployeeType 中的 getEmployeeSalary() 方法改为抽象方法。

1 public class EngineerType_State : EmployeeType_State
2 {
3     private decimal _baseSalary = 10000;
4 
5     public override decimal getEmployeeSalary()
6     {
7         return _baseSalary;
8     }
9 }

 

最后,附上示意代码,点这里下载。

 


参考资料:

《重构 改善既有代码的设计》 Martin Fowler

 

重构:越来越长的 switch ... case 和 if ... else if ... else

标签:

原文地址:http://www.cnblogs.com/isun/p/5003685.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!