策略模式+Map字典消除if else

前言

之前由于疫情的原因，公司让我做一个疫情物资申请的功能，其实系统里面以前就有很多种类的申请了，比如礼品申请，酒水申请等等，功能跟以前一个申请类似，复制下稍微改下就可以了，大致流程是这样的：公司内部人员进入系统，填写表单-->提交申请发起流程-->各节点领导审批-->审批通过后线下发放物资。前端页面其实是有一个申请列表的，列表可以有很多种申请，也可以选择多条未提交的申请进行列表提交，列表提交的有段代码是这样的：

if (ApplyTypeEnum.JS.getType().equals(apply.getApplyType())) {
    // 酒水申请
    // 获取流程所需参数
    
}else if (ApplyTypeEnum.LPSQ.getType().equals(apply.getApplyType())) {
    // 礼品申请
    // 获取流程所需参数
    
}
// 还有很多种申请类型.......

// 提交申请

列表提交可以有很多种申请类型，每种类型申请的流程参数是不一样的，以前使用if else来判断类型，进行流程参数的设置，可以看到if else是有很多的，而且我新写的疫情物资申请功能，又在下面加了一个else if，这种代码其实很low的，每次新增其他的申请，又得改代码，所以决定有策略模式来消除这种恶心的if else。

什么是策略模式

策略模式是对算法的包装，把使用算法的责任和算法本身分隔开，委派给不同的对象管理。策略模式通常把一系列的算法包装到一系列的策略类里面，作为一个抽象策略类的子类。简单来说就是就定义一个策略接口，子类策略去实现该接口去定义不同的策略。然后定义一个环境（Context）类，以策略接口作为成员变量，根据环境来使用具体的策略。

实现

1.定义策略接口

/**
 * 申请处理接口
 */
public interface IApplyHandler {

    /**
     * 设置流程所需参数
     */
    String getProcessParameters(Apply apply, Map map);

    /**
     * 获取申请类型
     * @return
     */
    ApplyTypeEnum getApplyTypeEnum();
}

2.创建策略接口实现类

/**
 * 酒水申请参数处理类
 */
@Component
public class DrinksApplyHandlerImpl implements IApplyHandler {
    /**
     * 设置流程所需参数
     * @param apply
     * @param map
     */
    @Override
    public String getProcessParameters(Apply apply, Map map) {
        // map.put("","");
        System.out.println("酒水申请流程所需参数设置成功");
        return null;
    }

    @Override
    public ApplyTypeEnum getApplyTypeEnum() {
        return ApplyTypeEnum.JS;
    }
}

/**
 * 礼品申请参数处理类
 */
@Component
public class PresentApplyHandlerImpl implements IApplyHandler {

    /**
     * 设置流程所需参数
     * @param apply
     * @param map
     */
    @Override
    public String getProcessParameters(Apply apply, Map map) {
        // map.put("","");
        System.out.println("礼品申请流程所需参数设置成功");
        return null;
    }

    @Override
    public ApplyTypeEnum getApplyTypeEnum() {
        return ApplyTypeEnum.LPSQ;
    }
}

3.创建策略工厂类

/**
 * 资产申请：策略工厂
 */
@Component
public class ApplyHandlerStrategyFactory implements InitializingBean{

    public static Map<String, IApplyHandler> applyHandlerMap = new ConcurrentHashMap<>();

    @Resource
    private ApplicationContext applicationContext;

    /**
     * 类初始化自动获取所有申请策略，放入Map中
     * @throws Exception
     */
    @Override
    public void afterPropertiesSet() throws Exception {
        Map<String, IApplyHandler> beans = applicationContext.getBeansOfType(IApplyHandler.class);
        if (beans != null && beans.size() > 0) {
            for (Map.Entry<String, IApplyHandler> entry : beans.entrySet()) {
                applyHandlerMap.put(entry.getValue().getApplyTypeEnum().getType(), entry.getValue());
            }
        }
    }

    /**
     * 从Map中根据key获取策略
     * @param applyType
     * @return
     */
    public static IApplyHandler getApplyHandlerStrategy(String applyType){
        return applyHandlerMap.get(applyType);
    }
}

4.创建策略上下文类

/**
 * 资产申请：策略上下文
 */
public class ApplyHandlerStrategyContext {

    private IApplyHandler applyHandler;

    public ApplyHandlerStrategyContext(String applyType) {
        this.applyHandler = ApplyHandlerStrategyFactory.getApplyHandlerStrategy(applyType);
    }

    public String getProcessParameters(Apply apply, Map map) {
        if (this.applyHandler != null) {
            return this.applyHandler.getProcessParameters(apply, map);
        }
        return "获取对应申请的实体策略类失败！";
    }
}

5.业务代码

    /**
     * 
     * @param ids
     * @return
     * @throws Exception
     */
    @Override
    public SimpleReturnVo submitApplyList(String ids) throws Exception {
        Map<String, Object> map = new HashMap<String, Object>();
        SimpleReturnVo returnVo = new SimpleReturnVo(AssetsConstant.ERROR, "提交失败！");
        String[] idsArr = ids.trim().split(",");
        for (String id : idsArr) {
            Apply apply = getApplyById(id);
            ApplyHandlerStrategyContext context = new ApplyHandlerStrategyContext(apply.getApplyType());
            String parameters = context.getProcessParameters(apply, map);
            if (StringUtils.isNotEmpty(parameters)) {
                returnVo.setResponseMsg(parameters);
                return returnVo;
            }
            // 提交流程


        }
        return returnVo;
    }

6.测试结果

image