利用策略模式优化过多 if else 代码
前言
不出意外,这应该是年前最后一次分享,本次来一点实际开发中会用到的小技巧。
比如平时大家是否都会写类似这样的代码:
-
if(a){
-
//dosomething
-
}else if(b){
-
//doshomething
-
}else if(c){
-
//doshomething
-
} else{
-
////doshomething
-
}
条件少还好,一旦 elseif
过多这里的逻辑将会比较混乱,并很容易出错。
比如这样:
摘自 cim 中的一个客户端命令的判断条件。
刚开始条件较少,也就没管那么多直接写的;现在功能多了导致每次新增一个 else
条件我都得仔细核对,生怕影响之前的逻辑。
这次终于忍无可忍就把他重构了,重构之后这里的结构如下:
最后直接变为两行代码,简洁了许多。
而之前所有的实现逻辑都单独抽取到其他实现类中。
这样每当我需要新增一个 else
逻辑,只需要新增一个类实现同一个接口便可完成。每个处理逻辑都互相独立互不干扰。
实现
按照目前的实现画了一个草图。
整体思路如下:
- 定义一个
InnerCommand
接口,其中有一个 process
函数交给具体的业务实现。 - 根据自己的业务,会有多个类实现
InnerCommand
接口;这些实现类都会注册到 SpringBean
容器中供之后使用。 - 通过客户端输入命令,从
SpringBean
容器中获取一个 InnerCommand
实例。 - 执行最终的
process
函数。
主要想实现的目的就是不在有多个判断条件,只需要根据当前客户端的状态动态的获取 InnerCommand
实例。
从源码上来看最主要的就是 InnerCommandContext
类,他会根据当前客户端命令动态获取 InnerCommand
实例。
- 第一步是获取所有的
InnerCommand
实例列表。 - 根据客户端输入的命令从第一步的实例列表中获取类类型。
- 根据类类型从
Spring
容器中获取具体实例对象。
因此首先第一步需要维护各个命令所对应的类类型。
所以在之前的枚举中就维护了命令和类类型的关系,只需要知道命令就能知道他的类类型。
这样才能满足只需要两行代码就能替换以前复杂的 ifelse
,同时也能灵活扩展。
-
InnerCommand instance = innerCommandContext.getInstance(msg);
-
instance.process(msg) ;
总结
当然还可以做的更灵活一些,比如都不需要显式的维护命令和类类型的对应关系。
只需要在应用启动时扫描所有实现了 InnerCommand
接口的类即可,在 cicada 中有类似实现,感兴趣的可以自行查看。
这样一些小技巧希望对你有所帮助。
以上所有源码可以在这里查看:
https://github.com/crossoverJie/cim