如何消除if...else代码块

青旅半醒 2022-09-15 05:53 180阅读 0赞

## 前言    ##

最近在做代码重构，发现了很多代码的烂味道。其他的不多说，今天主要说说那些又臭又长的if...else要如何重构。

在介绍更更优雅的编程之前，让我们一起回顾一下，不好的if...else代码

## 一、又臭又长的if...else ##

废话不多说，先看看下面的代码。

![8a8f9c525a0738cab820ccfb58e3202a.png][]

![37a36655b139d4b3dd3d0915d6016726.png][]

PayService类的toPay方法主要是为了发起支付，根据不同的code，决定调用用不同的支付类（比如：aliaPay）的pay方法进行支付。

这段代码有什么问题呢？也许有些人就是这么干的。

试想一下，如果支付方式越来越多，比如：又加了百度支付、美团支付、银联支付等等，就需要改toPay方法的代码，增加新的else...if判断，判断多了就会导致逻辑越来越多？

很明显，这里违法了设计模式六大原则的：`开闭原则` 和 `单一职责原则`。

![7c8b0d04383a125cd00dda70badec8ae.png][]

## 二、消除if...else的锦囊妙计 ##

### 1、使用注解 ###

代码中之所以要用code判断使用哪个支付类，是因为code和支付类没有一个绑定关系，如果绑定关系存在了，就可以不用判断了。

我们先定义一个注解。

![a1453d2fe5b2a5d4d7842ce724697aac.png][]

在所有的支付类上都加上该注解

![445ce80b2f16a4ad45a2bf0526d07c32.png][]

![2e979649a5baa8fc244d2a1bf9ff8b80.png][]

然后增加最关键的类：

![474e0a04d3e58db5a4e0fea7de40ab85.png][]

PayService2类实现了`ApplicationListener`接口，这样在`onApplicationEvent`方法中，就可以拿到`ApplicationContext`的实例。我们再获取打了PayCode注解的类，放到一个map中，map中的key就是PayCode注解中定义的value，跟code参数一致，value是支付类的实例。

这样，每次就可以每次直接通过code获取支付类实例，而不用if...else判断了。如果要加新的支付方法，只需在支付类上面打上PayCode注解定义一个新的code即可。

注意：这种方式的code可以没有业务含义，可以是纯数字，只有不重复就行。

### 2、动态拼接名称 ###

该方法主要针对code是有业务含义的场景。

![1e7cb60e3eeca423c0423e28c5cc51e4.png][]

我们可以看到，支付类bean的名称是由code和后缀拼接而成，比如：aliaPay、weixinPay和jingDongPay。这就要求支付类取名的时候要特别注意，前面的一段要和code保持一致。调用的支付类的实例是直接从`ApplicationContext`实例中获取的，默认情况下`bean`是单例的，放在内存的一个`map`中，所以不会有性能问题。

特别说明一下，这种方法实现了`ApplicationContextAware`接口跟上面的`ApplicationListener`接口不一样，是想告诉大家获取`ApplicationContext`实例的方法不只一种。

### 3、模板方法判断 ###

当然除了上面介绍的两种方法之外，spring的源码实现中也告诉我们另外一种思路，解决if...else问题。

我们先一起看看`spring AOP`的部分源码，看一下`DefaultAdvisorAdapterRegistry`的`wrap`方法

![e51fbb3cd1ac157483e249d3b363f8e7.png][]

重点看看`supportAdvice`方法，有三个类实现了这个方法。我们随便抽一个类看看

![cb53a074b5d77815025fb92d497269b5.png][]

该类的`supportsAdvice`方法非常简单，只是判断了一下`advice`的类型是不是`AfterReturningAdvice`。

我们看到这里应该有所启发。

其实，我们可以这样做，定义一个接口或者抽象类，里面有个`support`方法判断参数传的code是否自己可以处理，如果可以处理则走支付逻辑。

![8524db282f398e87099325dd29edf2c6.png][]

![5a52854379dbb8279447e22a8951a46f.png][]

每个支付类都有一个support方法，判断传过来的code是否和自己定义的相等。

![9d3b9535d645491bac9137b417f1df2f.png][]

这段代码中先把实现了IPay接口的支付类实例初始化到一个list集合中，返回在调用支付接口时循环遍历这个list集合，如果code跟自己定义的一样，则调用当前的支付类实例的pay方法。

### 4.策略+工厂模式 ###

这种方式也是用于code是有业务含义的场景。

*  `策略模式`定义了一组算法，把它们一个个封装起来, 并且使它们可相互替换。
 *  `工厂模式`用于封装和管理对象的创建，是一种创建型模式。

![7114054826a33f3487889ea320e53302.png][]

![d35172986860cbad42836c25d843d5f3.png][]

![4f07dae0bf90427a69143cc34704c193.png][]

这段代码的关键是`PayStrategyFactory`类，它是一个策略工厂，里面定义了一个全局的map，在所有IPay的实现类中注册当前实例到map中，然后在调用的地方通过`PayStrategyFactory`类根据code从map获取支付类实例即可。

### 5.责任链模式 ###

这种方式在代码重构时用来消除if...else非常有效。

*  责任链模式：将请求的处理对象像一条长链一般组合起来，形成一条对象链。请求并不知道具体执行请求的对象是哪一个，这样就实现了请求与处理对象之间的解耦。

常用的`filter`、`spring aop`就是使用了责任链模式，这里我稍微改良了一下，具体代码如下：

![2a9f22d9c7af3e90513e7b565ffcd657.png][]

![e2318b4ce7c4bef27e78b40ce80b2c5d.png][]

![25fce36d049cd8e1725d993dda4eb0cf.png][]

这段代码的关键是每个PayHandler的子类，都定义了下一个需要执行的PayHandler子类，构成一个链式调用，通过PayHandlerChain把这种链式结构组装起来。

### 6、其他的消除if...else的方法 ###

当然实际项目开发中使用if...else判断的场景非常多，上面只是其中几种场景。下面再列举一下，其他常见的场景。

#### 1.根据不同的数字返回不同的字符串 ####

![6fdb85164156d8f782dd39f48acd3237.png][]

其实，这种判断没有必要，用一个枚举就可以搞定。

![b8b122b388559555f78db80740e290e6.png][]

再把调用方法稍微调整一下

![17edfb8ab3d2bff7298ab4086ab63cc3.png][]

完美。

#### 2.集合中的判断 ####

上面的枚举MessageEnum中的getMessageEnum方法，如果不用java8的语法的话，可能要这样写

![d7df6d090c68de9704fd13ed76dd8ce0.png][]

对于集合中过滤数据，或者查找方法，java8有更简单的方法消除if...else判断。

![ef8bec8155bb4855e563745cf2203fb7.png][]

#### 3.简单的判断 ####

其实有些简单的if...else完全没有必要写，可以用三目运算符代替，比如这种情况：

![9fc3b042e3f812184cf264c49d714dad.png][]

改成三目运算符：

![8bd5b0713350a7265b9a2c573ba52fef.png][]

![194d51d0ace06cc2c42cd5bd41579ab9.png][]

若有收获，请记得三连发，**分享**，**点赞**，**在看**

[8a8f9c525a0738cab820ccfb58e3202a.png]: /images/20220828/0cde3fe6e32d41bda159af0f3903fd81.png
[37a36655b139d4b3dd3d0915d6016726.png]: /images/20220828/324d04aac5a74e46a979bad1a3650f64.png
[7c8b0d04383a125cd00dda70badec8ae.png]: /images/20220828/c9dfc4099a2c41498184fdbb8d531745.png
[a1453d2fe5b2a5d4d7842ce724697aac.png]: /images/20220828/63571e64abd34e48a8eb0e1432cbe19f.png
[445ce80b2f16a4ad45a2bf0526d07c32.png]: /images/20220828/c5b670408cb146ec80d8267adcf78e71.png
[2e979649a5baa8fc244d2a1bf9ff8b80.png]: /images/20220828/869404a913da43158f0000b2064741a0.png
[474e0a04d3e58db5a4e0fea7de40ab85.png]: /images/20220828/d1f15d50be3a4eef856cec11645fc415.png
[1e7cb60e3eeca423c0423e28c5cc51e4.png]: /images/20220828/e5c716e9c7644442b22163eb668c4774.png
[e51fbb3cd1ac157483e249d3b363f8e7.png]: /images/20220828/8d9836e23fbf403bb8bf1c2dca09f727.png
[cb53a074b5d77815025fb92d497269b5.png]: /images/20220828/35db9f72d8f04ea7989f6083c3ef4054.png
[8524db282f398e87099325dd29edf2c6.png]: /images/20220828/3eabca37221e4ce7aed310465a885298.png
[5a52854379dbb8279447e22a8951a46f.png]: /images/20220828/df66533c2f7d45dba9b22502db2d6770.png
[9d3b9535d645491bac9137b417f1df2f.png]: /images/20220828/335f2e9411744536a407cf121961ecdc.png
[7114054826a33f3487889ea320e53302.png]: /images/20220828/1e38cecc6a904944af6a7f12744ec28d.png
[d35172986860cbad42836c25d843d5f3.png]: /images/20220828/f50187e25e314e3c85ef006032fe26f5.png
[4f07dae0bf90427a69143cc34704c193.png]: /images/20220828/e535800678d94cca85e76f4e3fe49820.png
[2a9f22d9c7af3e90513e7b565ffcd657.png]: /images/20220828/5e92258d186441488fbccab97d6d60ab.png
[e2318b4ce7c4bef27e78b40ce80b2c5d.png]: /images/20220828/a6962c44b59649f2b3485d7c689f8555.png
[25fce36d049cd8e1725d993dda4eb0cf.png]: /images/20220828/631c705c87504c109c4c722650490851.png
[6fdb85164156d8f782dd39f48acd3237.png]: /images/20220828/b77ace5a0e884f22bfd4761494933226.png
[b8b122b388559555f78db80740e290e6.png]: /images/20220828/be2ecfad2bfd433b8e45d0cd15cb1cbe.png
[17edfb8ab3d2bff7298ab4086ab63cc3.png]: /images/20220828/7575634b2c5b4e51989456bd21cd3836.png
[d7df6d090c68de9704fd13ed76dd8ce0.png]: /images/20220828/402b116a0922420f99354d2cc42de929.png
[ef8bec8155bb4855e563745cf2203fb7.png]: /images/20220828/3012b9374693429ebd2c063449643bc2.png
[9fc3b042e3f812184cf264c49d714dad.png]: /images/20220828/d8427e94c8b743b7af141ad327c8c78d.png
[8bd5b0713350a7265b9a2c573ba52fef.png]: /images/20220828/ae1b621ce88941beb09379fd2e73dd79.png
[194d51d0ace06cc2c42cd5bd41579ab9.png]: /images/20220828/e97405124fec4adfaa47262382148567.png