想当年,其实我也特别钟情于 if/else
连环写法,上来就是一顿SAO操作,比如举个好理解的简单栗子:
一般来说我们正常的后台管理系统都有所谓的角色的概念,不同管理员权限不一样,能够行使的操作也不一样,比如:
-
系统管理员(
ROLE_ROOT_ADMIN
):有A
操作权限 -
订单管理员(
ROLE_ORDER_ADMIN
):有B
操作权限 -
普通用户(
ROLE_NORMAL
):有C
操作权限
比如一个用户进来,我们需要根据不同用户的角色来判断其有哪些行为,这时候SAO代码出现了:
publicclassJudgeRole{
publicString 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
:
publicenumRoleEnumimplementsRoleOperation{
// 系统管理员(有A操作权限) ROLE_ROOT_ADMIN {@OverridepublicString op() {return"ROLE_ROOT_ADMIN:"+ " has AAA permission";}},
// 订单管理员(有B操作权限) ROLE_ORDER_ADMIN {@OverridepublicString op() {return"ROLE_ORDER_ADMIN:"+ " has BBB permission";}},
// 普通用户(有C操作权限) ROLE_NORMAL {@OverridepublicString 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;}
@OverridepublicString op() {return roleName + " has AAA permission";}}复制代码
// 订单管理员(有B操作权限)publicclassOrderAdminRoleimplementsRoleOperation{
privateString roleName;
publicOrderAdminRole( String roleName ) {this.roleName = roleName;}
@OverridepublicString op() {return roleName + " has BBB permission";}}复制代码
// 普通用户(有C操作权限)publicclassNormalRoleimplementsRoleOperation{
privateString roleName;
publicNormalRole( String roleName ) {this.roleName = roleName;}
@OverridepublicString 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();}}复制代码
这样的话以后想扩展条件也很容易,只需要增加新代码,而不需要动以前的业务代码,非常符合“开闭原则”。