使用规则执行器代替 if else 判断

业务场景

近日有个需求，需要对之前已有的用户申请规则进行拓展。场景大概如下所示:

if (是否海外用户) {
 return false;
}

if (刷单用户) {
  return false;
}

if (未付费用户 && 不再服务时段) {
  return false
}

if (转介绍用户 || 付费用户 || 内推用户) {
  return true;
}

按照上述的条件我们可以得出的结论是：

• 流程主要是基于 and 或者 or 的关系。
• 如果有一个不匹配的话，后续的流程是不用执行的，就是需要具备一个短路的功能。
• 对于目前的现状来说，如果在原有的基础上来改，只要稍微注意一下解决需求不是很大的问题，但是说后面可维护性非常差。

后面进过权衡过后，还是决定将这个部分进行重构一下。

规则执行器

规则执行器设计

规则的抽象和实现

/**
 * 基础规则业务数据
 * @author LiuHuan
 * @date 2021/4/23 10:24 上午
 */
@Data
public class RuleDTO {

    private String address;

    private int age;

}

/**
 * 具体业务数据
 * @author LiuHuan
 * @date 2021/4/23 10:47 上午
 */
@Data
public class NationalityRuleDTO extends RuleDTO{

    private String nationality;

}

/**
 * 规则接口
 * @author LiuHuan
 * @date 2021/4/23 10:33 上午
 */
public interface Rule {

    boolean execute(RuleDTO dto);

}

/**
 * 基础规则
 * @author LiuHuan
 * @date 2021/4/23 10:35 上午
 */
public abstract class BaseRule implements Rule{

    public static final String MATCH_ADDRESS_START= "北京";

    public static final String MATCH_NATIONALITY_START= "中国";

    protected <T> T convert(RuleDTO dto) {
        return (T) dto;
    }

    @Override
    public boolean execute(RuleDTO dto) {
        return executeRule(convert(dto));
    }

    protected <T> boolean executeRule(T t) {
        return true;
    }

}

/**
 * 地址规则
 * @author LiuHuan
 * @date 2021/4/23 10:38 上午
 */
public class AddressRule extends BaseRule{

    @Override
    public boolean execute(RuleDTO dto) {
        System.out.println("AddressRule invoke!");
        return dto.getAddress().startsWith(MATCH_ADDRESS_START);
    }

}

/**
 * 国家规则
 * @author LiuHuan
 * @date 2021/4/23 10:39 上午
 */
public class NationalityRule extends BaseRule{

    @Override
    protected <T> T convert(RuleDTO dto) {
        NationalityRuleDTO nationalityRuleDto = new NationalityRuleDTO();
        if (dto.getAddress().startsWith(MATCH_ADDRESS_START)) {
            nationalityRuleDto.setNationality(MATCH_NATIONALITY_START);
        }
        return (T) nationalityRuleDto;
    }


    @Override
    protected <T> boolean executeRule(T t) {
        System.out.println("NationalityRule invoke!");
        NationalityRuleDTO nationalityRuleDto = (NationalityRuleDTO) t;
        return nationalityRuleDto.getNationality().startsWith(MATCH_NATIONALITY_START);
    }

}

执行器

/**
 * 规则执行
 * @author LiuHuan
 * @date 2021/4/23 10:41 上午
 */
public class RuleManage {

    private Map<Integer, List<Rule>> hashMap = new HashMap<>();
    private static final int AND = 1;
    private static final int OR = 0;

    public static RuleManage create() {
        return new RuleManage();
    }


    public RuleManage and(List<Rule> ruleList) {
        hashMap.put(AND, ruleList);
        return this;
    }

    public RuleManage or(List<Rule> ruleList) {
        hashMap.put(OR, ruleList);
        return this;
    }

    public boolean execute(RuleDTO dto) {
        for (Map.Entry<Integer, List<Rule>> item : hashMap.entrySet()) {
            List<Rule> ruleList = item.getValue();
            switch (item.getKey()) {
                case AND:
                    // 如果是 and 关系，同步执行
                    System.out.println("execute key = " + 1);
                    if (!and(dto, ruleList)) {
                        return false;
                    }
                    break;
                case OR:
                    // 如果是 or 关系，并行执行
                    System.out.println("execute key = " + 0);
                    if (!or(dto, ruleList)) {
                        return false;
                    }
                    break;
                default:
                    break;
            }
        }
        return true;
    }

    private boolean and(RuleDTO dto, List<Rule> ruleList) {
        for (Rule rule : ruleList) {
            boolean execute = rule.execute(dto);
            if (!execute) {
                // and 关系匹配失败一次，返回 false
                return false;
            }
        }
        // and 关系全部匹配成功，返回 true
        return true;
    }

    private boolean or(RuleDTO dto, List<Rule> ruleList) {
        for (Rule rule : ruleList) {
            boolean execute = rule.execute(dto);
            if (execute) {
                // or 关系匹配到一个就返回 true
                return true;
            }
        }
        // or 关系一个都匹配不到就返回 false
        return false;
    }

}

测试

/**
 * @author LiuHuan
 * @date 2021/4/23 10:49 上午
 */
public class RuleManageTest {

    @Test
    public void execute() {
        //规则执行器
        //优点：比较简单，每个规则可以独立，将规则，数据，执行器拆分出来，调用方比较规整
        //缺点：数据依赖公共传输对象 dto

        //1. 定义规则  init rule
        NationalityRule nationalityRule = new NationalityRule();
        AddressRule addressRule = new AddressRule();

        //2. 构造需要的数据 create dto
        RuleDTO dto = new RuleDTO();
        dto.setAge(5);
        dto.setAddress("北京");

        //3. 通过以链式调用构建和执行 rule execute
        boolean ruleResult = RuleManage
            .create()
            .and(Arrays.asList(nationalityRule, addressRule))
            //.or(Arrays.asList(addressRule, nationalityRule))
            .execute(dto);
        System.out.println("this student rule execute result :" + ruleResult);
    }

}

总结

优点

• 比较简单，每个规则可以独立，将规则，数据，执行器拆分出来，调用方比较规整
• Rule 模板类中定义 convert 方法做参数的转换这样可以能够，为特定 rule 需要的场景数据提供拓展

缺点

• 上下 rule 有数据依赖性，如果直接修改公共传输对象 dto 这样设计不是很合理，建议提前构建数据