策略模式替代if-else繁琐判断

怼烎@ 2023-10-09 16:19 106阅读 0赞

###  ###

经常在网上看到一些名为“别再if-else走天下了”，“教你干掉if-else”等之类的文章，大部分都会讲到用策略模式去代替if-else。策略模式实现的方式也大同小异。主要是定义统一行为（接口或[抽象类][Link 1]），并实现不同策略下的处理逻辑（对应实现类）。客户端使用时自己选择相应的处理类，利用工厂或其他方式。

本文要说的是实现策略模式的一个方法，代替了传统的if-else判断。话不多说，直接上代码

首先创建一个策略接口

public interface StrategyService {
    
        /**
         * 选择使用的算法
         * @param type 类型
         * @return 策略算法
         */
        Strategy render(String type);
    
    }

然后实现这个service，构建策略

@Service
    public class StrategyServiceImpl implements StrategyService {
    
        private final Map<String, Strategy> strategyMap = new ConcurrentHashMap<>();
    
        /**
         * 构建策略信息
         * @param strategyList 策略信息
         */
        @Autowired
        public StrategyServiceImpl(List<Strategy> strategyList) {
             for(Strategy strategy : strategyList) {
                if (strategy instanceof DeleteMaterialServiceImpl) {
                    strategyMap.put(InterfaceTypeEnum.deleteMaterial.getCode(), strategy);
                }
    
                if (strategy instanceof InsertMaterialServiceImpl) {
                    strategyMap.put(InterfaceTypeEnum.insertMaterial.getCode(), strategy);
                }
    
                 if (strategy instanceof UpdateMaterialServiceImpl) {
                     strategyMap.put(InterfaceTypeEnum.updateMaterial.getCode(), strategy);
                 }
    
                if (strategy instanceof MaterialInwareServiceImpl) {
                    strategyMap.put(InterfaceTypeEnum.materialInware.getCode(), strategy);
                }
    
                if (strategy instanceof MaterialOutwareServiceImpl) {
                    strategyMap.put(InterfaceTypeEnum.materialOutware.getCode(), strategy);
                }
    
                 if (strategy instanceof AutoMaterialServiceImpl) {
                    strategyMap.put(InterfaceTypeEnum.autoMaterial.getCode(), strategy);
                }
    
                if (strategy instanceof InventoryStockServiceImpl) {
                    strategyMap.put(InterfaceTypeEnum.inventoryStock.getCode(), strategy);
                }
            }
        }
    
        @Override
        public Strategy render(String type) {
            return this.strategyMap.get(type);
        }
    }

其中 Strategy 是一个接口，根据不同的业务对Strategy得对实现

StrategyServiceImpl（）方法是为了注入Strategy的实现类，根据类型匹配实现类

render(String type)方法是为了根据类型匹配带具体的实现类

Strategy接口

public interface Strategy {
    
        AjaxResult materialInfoRomaToMes(MaterialRequestVo materialRequestVo) throws IllegalAccessException, IntrospectionException, InstantiationException;
    
        
        AjaxResult materialInfoMesToRoma(MaterialRequestVo materialRequestVo);
    }

然后简单扔出一个实现Strategy的例子

@Service
    public class AutoMaterialServiceImpl implements Strategy {
    
    
    
        @Override
        @DSTransactional
        public AjaxResult materialInfoRomaToMes(MaterialRequestVo materialRequestVo) {
            String result = "";
            return AjaxResult.success(result);
        }
    
        @Override
        @RequiredLog(methodType="1")
        public AjaxResult materialInfoMesToRoma(MaterialRequestVo materialRequestVo) {
    
            return AjaxResult.success(result);
        }
    }

![b29e036c74874cde96b75f276f0f3827.png][]

![a3f0eaf8fe264cd4aa5232d332f2738a.png][]

![bddac0da40904775b921e8a501681d77.png][]

![7d76b19d24de45b899ca36e788269e85.png][]

再一个接口多个实现都写好了之后，就是引用的问题了，下面看看如何利用render(String type)方法完成实现类的引用

@Service
    public class MesProvideRomaServiceImpl {
    
        @Autowired
        private StrategyService strategyService;
    
        @Override
        @RequiredLog(methodType="2")
        public AjaxResult materialInfoRomaToMes(MaterialRequestVo materialRequestVo) throws IllegalAccessException, IntrospectionException, InstantiationException {
            
            return strategyService.render(materialRequestVo.getType()).materialInfoRomaToMes(materialRequestVo);
        }
    }

至此，到这里一个简单的策略模式就算是完成了，简单记录，也希望可以帮到有需要的人。

[Link 1]: https://so.csdn.net/so/search?q=%E6%8A%BD%E8%B1%A1%E7%B1%BB&spm=1001.2101.3001.7020
[b29e036c74874cde96b75f276f0f3827.png]: https://img-blog.csdnimg.cn/b29e036c74874cde96b75f276f0f3827.png
[a3f0eaf8fe264cd4aa5232d332f2738a.png]: https://img-blog.csdnimg.cn/a3f0eaf8fe264cd4aa5232d332f2738a.png
[bddac0da40904775b921e8a501681d77.png]: https://img-blog.csdnimg.cn/bddac0da40904775b921e8a501681d77.png
[7d76b19d24de45b899ca36e788269e85.png]: https://img-blog.csdnimg.cn/7d76b19d24de45b899ca36e788269e85.png