答应我，别再 if/else 走天下了可以吗？

点击上方 蓝字 设为星标

想当年，其实我也特别钟情于 if/else 连环写法，上来就是一顿 SAO操作 ，比如 举个好理解的简单栗子 ：

一般来说我们正常的后台管理系统都有所谓的角色的概念，不同管理员权限不一样，能够行使的操作也不一样，比如：

• 系统管理员（ ROLE_ROOT_ADMIN ）：有 A 操作权限
• 订单管理员（ ROLE_ORDER_ADMIN ）：有 B 操作权限

• 普通用户（ ROLE_NORMAL ）：有 C 操作权限

比如一个用户进来，我们需要根据不同用户的角色来判断其有哪些行为，这时候 SAO代码 出现了：

publicclassJudgeRole{

public




    
String judge( String roleName ) {
String result = "";
if(roleName.equals("ROLE_ROOT_ADMIN")) { // 系统管理员有A权限
            result = "ROLE_ROOT_ADMIN: "+ "has AAA permission";
} elseif( roleName.equals("ROLE_ORDER_ADMIN") ) { // 订单管理员有B权限
            result = "ROLE_ORDER_ADMIN: "+ "has BBB permission";
} elseif( roleName.equals("ROLE_NORMAL") ) { // 普通用户有C权限
            result = "ROLE_NORMAL: "+ "has CCC permission";
} else{
            result = "XXX";
}
return result;
}

}

这样当系统里有 几十个角色 时，那几十个 if/else 嵌套可以说是非常酸爽了…… 这样 一来 非常不优雅，别人阅读起来很费劲； 二来 则是以后如果再复杂一点，或者想要再加条件的话 不好扩展 ；而且代码一改，以前的老功能肯定还得重测，岂不疯了……

所以，如果在不看下文的情况下，你一般会如何去对付这些令人头痛的if/else语句呢？

当然有人会说用 switch/case 来写是否会优雅一些呢？答案是： 毛区别都没有 ！

接下来简单讲几种改进方式，别再 if/else 走天下了

什么角色能干什么事，这很明显有一个对应关系，所以学过的枚举为啥不用呢？

首先定义一个公用接口 RoleOperation ，表示不同角色所能做的操作：

publicinterfaceRoleOperation{
String op();  // 表示某个角色可以做哪些op操作
}

接下来我们将不同角色的情况全部交由枚举类来做，定义一个不同角色有不同权限的枚举类 RoleEnum ：

publicenum




    
RoleEnumimplementsRoleOperation{

// 系统管理员(有A操作权限)
    ROLE_ROOT_ADMIN {
@Override
publicString op() {
return"ROLE_ROOT_ADMIN:"+ " has AAA permission";
}
},

// 订单管理员(有B操作权限)
    ROLE_ORDER_ADMIN {
@Override
publicString op() {
return"ROLE_ORDER_ADMIN:"+ " has BBB permission";
}
},

// 普通用户(有C操作权限)
    ROLE_NORMAL {
@Override
publicString




    
 op() {
return"ROLE_NORMAL:"+ " has CCC permission";
}
};
}

接下来调用就变得异常简单了，一行代码就行了， if/else 也灰飞烟灭了：

publicclassJudgeRole{
publicString judge( String roleName ) {
// 一行代码搞定！之前的if/else没了！
returnRoleEnum.valueOf(roleName).op();
}
}

而且这样一来，以后假如我想扩充条件，只需要去枚举类中加代码即可，而不是去改以前的代码，这岂不很稳！

除了用枚举来消除 if/else ，工厂模式也可以实现

不同分 支做不同的事情，很明显就提供了使用工厂模式的契机，我们只需要将不同情况单独定义好，然后去工厂类里面 聚合 即可。

首先，针对不同的角色，单独定义其业务类：

// 系统管理员(有A操作权限)
publicclassRootAdminRoleimplementsRoleOperation{

privateString roleName;

publicRootAdminRole( String roleName ) {
this.roleName = roleName;
}

@Override
publicString op() {
return roleName + " has AAA permission";
}
}

// 订单管理员(有B操作权限)
publicclass




    
OrderAdminRoleimplementsRoleOperation{

privateString roleName;

publicOrderAdminRole( String roleName ) {
this.roleName = roleName;
}

@Override
publicString op() {
return roleName + " has BBB permission";
}
}

// 普通用户(有C操作权限)
publicclassNormalRoleimplementsRoleOperation{

privateString roleName;

publicNormalRole( String roleName ) {
this.roleName = roleName;
}




    

@Override
publicString op() {
return roleName + " has CCC permission";
}
}

接下来再写一个 工厂类 RoleFactory 对上面不同角色进行聚合：

publicclassRoleFactory{
staticMap<String, RoleOperation> roleOperationMap = newHashMap<>();

// 在静态块中先把初始化工作全部做完
static{
        roleOperationMap.put( "ROLE_ROOT_ADMIN", newRootAdminRole("ROLE_ROOT_ADMIN") );
        roleOperationMap.put( "ROLE_ORDER_ADMIN", newOrderAdminRole("ROLE_ORDER_ADMIN") );
        roleOperationMap.put( "ROLE_NORMAL", newNormalRole("ROLE_NORMAL") );
}

publicstaticRoleOperation getOp( String roleName ) {
return roleOperationMap.get( roleName );
}
}

接下来借助上面这个工厂，业务代码调用也只需一行代码， if/else 同样被消除了：

publicclassJudgeRole{
publicString judge( String roleName ) {
// 一行代码搞定！之前的 if/else也没了！
returnRoleFactory.getOp(roleName).op();
}
}