《设计模式的运用》使用策略模式+工厂模式优化代码中的if else

Dear 丶 2024-03-17 19:48 79阅读 0赞

## 使用策略模式优化if else ##

有这样一段逻辑

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>function{ </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>for{ </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>if()</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>if()</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>if(</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>if(</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>)</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>)</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>...</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

公司有的祖传的代码，是一段规则校验，校验的越多，每一个请求都会进来校验，加上后来的开发人员也不注意，每次多加校验就多加一个if,只要命中就在if里面写逻辑，好一点的会抽到一个方法中，看起来稍好些，但是if越加越多，一个循环最终发展到了数千行，可读性越来越差，领导点名重构；

解决方案： 策略模式+工厂模式

工厂模式大家都很熟悉了，所以简单介绍一下策略模式

策略模式，顾名思义，组要是依靠策略来完成类的具体行为，所以，策略模式的核心就是编写不同的策略类，我们可以通过Java的接口编写抽象策略，用不同的实现类来完成策略，当然策略模式还需要一个上下文的类，用来选择策略；现实生活中也有很好的例子，例如支付方式，你可以选择微信支付，支付宝支付，银行卡支付等，  
首先我们编写抽象策略

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>public interface Strategy { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>Boolean check(Object obj);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

然后我们编写具体的实现

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>public class StrategyImpl01 implements Strategy{ </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>@Override</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public int check(Object obj) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>// 实现具体的业务逻辑</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return "";</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public class StrategyImpl02 implements Strategy{}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public class StrategyImpl03 implements Strategy{}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>...</code></td> 
  </tr> 
 </tbody> 
</table>

实现上下文类

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>public class Context { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>private Strategy strategy;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public Context(Strategy strategy){ </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>this.strategy = strategy;</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public int executeStrategy(Object obj){ </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return strategy.check(obj);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

调用策略类

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>public static void main(String[] args) { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>Context context = new Context(new StrategyImpl01()); </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>context.check(obj)</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>context = new Context(new StrategyImpl02()); </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>context.check(obj)</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>context = new Context(new StrategyImpl03()); </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>context.check(obj)</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

到此，策略模式已经实现了，但是不是特别优雅，所以我们加上了工厂模式  
编写策略工厂

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>public clss Factory{ </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>// 申明一个全局的map,key 为if中的条件，value为策略类</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>private static final Map&lt;String,Strategy&gt; strategyMap = new HashMa&lt;String,Strategy&gt;();</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>// 填充map</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>static { </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>strategyMap.put("aaa",new StrategyImpl01())</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>strategyMap.put("bbb",new StrategyImpl02())</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>strategyMap.put("bbb",new StrategyImpl03())</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>...</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>// 暴露一个获取map的方法</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>public static Strategy getStrategy(String key){ </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>return strategyMap.get(key).check(obj)</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

这样，我们只需要在用的时候

<table> 
 <tbody> 
  <tr> 
   <td></td> 
   <td><code>public static void main(String[] args){ </code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>Factory.getStrategy(key);</code></td> 
  </tr> 
  <tr> 
   <td></td> 
   <td><code>}</code></td> 
  </tr> 
 </tbody> 
</table>

TYPESCRIPT 复制 全屏

看起来优雅多了