还在用策略模式解决 if-else？Map+函数式接口方法才是王道

本文链接：https://blog.csdn.net/qq_39712282/article/details/131397221

使用Map+函数式接口替换if-else及其策略模式

最近写了一个服务：根据优惠券的类型resourceType和编码resourceId来查询发放方式和领取规则；

实现方式：

根据优惠券类型resourceType -> 确定查询哪个数据表
根据编码resourceId -> 到对应的数据表里边查询优惠券的派发方式grantType和领取规则
优惠券有多种类型，分别对应了不同的数据库表：
红包 —— 红包发放规则表
购物券 —— 购物券表
QQ会员
外卖会员
。。。等

实际的优惠券远不止这些，这个需求是要我们写一个业务分派的逻辑;
第一个能想到的思路就是if-else或者switch case:

switch(resourceType){
 case "红包": 
  查询红包的派发方式 
  break;
 case "购物券": 
  查询购物券的派发方式
  break;
 case "QQ会员" :
  break;
 case "外卖会员" :
  break;
 ......
 default : logger.info("查找不到该优惠券类型resourceType以及对应的派发方式");
  break;
}

如果要这么写的话，一个方法的代码可就太长了，影响了可读性。（别看着上面case里面只有一句话，但实际情况是有很多行的）

而且由于整个 if-else的代码有很多行，也不方便修改，可维护性低。

第二种就是使用策略模式，具体的实现方式可以参考我上一篇文章：

https://blog.csdn.net/qq_39712282/article/details/129625513?spm=1001.2014.3001.5501

策略模式在业务逻辑分派的时候还是if-else，只是说比第一种思路的if-else 更好维护一点。

但缺点也明显：

如果 if-else的判断情况很多，那么对应的具体策略实现类也会很多，上边的具体的策略实现类还只是2个，查询红包发放方式写在类RedPaper里边，购物券写在另一个类Shopping里边；那资源类型多个QQ会员和外卖会员，不就得再多写两个类？有点麻烦了
没法俯视整个分派的业务逻辑

Map+函数式接口

用上了Java8的新特性lambda表达式

判断条件放在key中
对应的业务逻辑放在value中

这样子写的好处是非常直观，能直接看到判断条件对应的业务逻辑

需求：根据优惠券(资源)类型resourceType和编码resourceId查询派发方式grantType

上代码：
如果单个 if 语句块的业务逻辑有很多行的话，我们可以把这些业务操作抽出来，写成一个单独的Service，即：

package com.test.serves;

/**
 * <p>
 * Title:
 * </p>
 * <p>
 *
 * @Description:com.test.serves </p>
 * <p>
 * @Class_Name:GrantTypeServe </p>
 * <p>
 * @Create_Date:2023/6/26 </p>
 * <p>
 * <p>
 * @Author:YYF</p>
 */
public interface GrantTypeServe {
    /**
     * 红包发放的方式
     * @param resourceId
     * @return
     */
    String redPaper(String resourceId);

    /**
     * 购物券的发放方式
     * @param resourceId
     * @return
     */
    String shopping(String resourceId);

    /**
     * qq会员的发放方式
     * @param resourceId
     * @return
     */
    String QQVip(String resourceId);
}

实现类：

package com.test.serves.impl;

import com.test.serves.GrantTypeServe;
import org.springframework.stereotype.Service;

/**
 * <p>
 * Title:
 * </p>
 * <p>
 *
 * @Description:com.test.serves.impl </p>
 * <p>
 * @Class_Name:GrantTypeServeImpl </p>
 * <p>
 * @Create_Date:2023/6/26 </p>
 * <p>
 * <p>
 * @Author:YYF </p>
 */
@Service
public class GrantTypeServeImpl implements GrantTypeServe {


    @Override
    public String redPaper(String resourceId) {
        return "每周末9点发放";
    }

    @Override
    public String shopping(String resourceId) {
        return "每周三9点发放";
    }

    @Override
    public String QQVip(String resourceId) {
        return "每周一0点开始秒杀";
    }
}

统一接口：

package com.test.serves.impl;

import com.test.serves.GrantTypeServe;
import lombok.RequiredArgsConstructor;
import org.springframework.cglib.core.internal.Function;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;
import java.util.HashMap;
import java.util.Map;

/**
 * <p>
 * Title:
 * </p>
 * <p>
 *
 * @Description:com.test.serves.impl </p>
 * <p>
 * @Class_Name:QueryGrantTypeServeImpl </p>
 * <p>
 * @Create_Date:2023/6/26 </p>
 * <p>
 * <p>
 * @Author:YYF </p>
 */
@Service
@RequiredArgsConstructor
public class QueryGrantTypeServeImpl {

    private final GrantTypeServe grantTypeServe;

    private Map<String, Function<String, String>> grantTypeMap = new HashMap<>();

    /**
     * 初始化业务分派逻辑,代替了if-else部分
     * key: 优惠券类型
     * value: lambda表达式,最终会获得该优惠券的发放方式
     */
    @PostConstruct
    public void dispatcherInit() {
        grantTypeMap.put("红包", resourceId -> grantTypeServe.redPaper(resourceId));
        grantTypeMap.put("购物券", resourceId -> grantTypeServe.shopping(resourceId));
        grantTypeMap.put("qq会员", resourceId -> grantTypeServe.QQVip(resourceId));
    }

    public String getResult(String resourceType, String resourceId) {
        //Controller根据 优惠券类型resourceType、编码resourceId 去查询 发放方式grantType
        Function<String, String> result = grantTypeMap.get(resourceType);
        if (result != null) {
            //传入resourceId 执行这段表达式获得String型的grantType
            return result.apply(resourceId);
        }
        return "查询不到该优惠券的发放方式";
    }
}

入参String resourceId是用来查数据库的，这里简化了，传参之后不做处理。

用http调用的结果：

package com.test.controller;

import com.test.serves.impl.QueryGrantTypeServeImpl;
import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * <p>
 * Title:
 * </p>
 * <p>
 *
 * @Description:com.test.controller </p>
 * <p>
 * @Class_Name:GrantTypeController </p>
 * <p>
 * @Create_Date:2023/6/26 </p>
 * <p>
 * <p>
 * @Author:YYF </p>
 */
@RestController
@RequiredArgsConstructor
public class GrantTypeController {
    private final QueryGrantTypeServeImpl queryGrantTypeServe;

    @PostMapping("/grantType")
    public String test(String resourceName) {
        return queryGrantTypeServe.getResult(resourceName, "1");
    }
}