还在写大量 if 判断？试试规则执行器，真香！

2024-04-25码农

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

if (是否海外用户) { returnfalse; } if (刷单用户) { returnfalse; } if (未付费用户 && 不再服务时段) { returnfalse } if (转介绍用户 || 付费用户 || 内推用户) { returntrue; }

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

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

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

规则执行器

针对这个需求，我首先梳理了一下咱们规则执行器大概的设计，然后我设计了一个 V1 版本和大家一起分享一下，如果大家也有这样的 case 可以给我分享留言，下面部分主要是设计和实现的流程和 code。

规则执行器的设计

对于规则的抽象并实现规则

// 业务数据 @Data public classRuleDto{ private String address; privateint age; } // 规则抽象 publicinterfaceBaseRule{ booleanexecute(RuleDto dto); } // 规则模板 publicabstract classAbstractRuleimplementsBaseRule{ protected <T> T convert(RuleDto dto){ return (T) dto; } @Override publicbooleanexecute(RuleDto dto){ return executeRule(convert(dto)); } protected <T> booleanexecuteRule(T t){ returntrue; } } // 具体规则- 例子1 public classAddressRuleextendsAbstractRule{ @Override publicbooleanexecute(RuleDto dto){ System.out.println("AddressRule invoke!"); if (dto.getAddress().startsWith(MATCH_ADDRESS_START)) { returntrue; } returnfalse; } } // 具体规则- 例子2 public classNationalityRuleextendsAbstractRule{ @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> booleanexecuteRule(T t){ System.out.println("NationalityRule invoke!"); NationalityRuleDto nationalityRuleDto = (NationalityRuleDto) t; if (nationalityRuleDto.getNationality().startsWith(MATCH_NATIONALITY_START)) { returntrue; } returnfalse; } } // 常量定义 public classRuleConstant{ publicstaticfinal String MATCH_ADDRESS_START= "北京"; publicstaticfinal String MATCH_NATIONALITY_START= "中国"; }

执行器构建

public classRuleService{ private Map<Integer, List<BaseRule>> hashMap = new HashMap<>(); privatestaticfinalint AND = 1; privatestaticfinalint OR = 0; publicstatic RuleService create(){ returnnew RuleService(); } public RuleService and(List<BaseRule> ruleList){ hashMap.put(AND, ruleList); returnthis; } public RuleService or(List<BaseRule> ruleList){ hashMap.put(OR, ruleList); returnthis; } publicbooleanexecute(RuleDto dto){ for (Map.Entry<Integer, List<BaseRule>> item : hashMap.entrySet()) { List<BaseRule> ruleList = item.getValue(); switch (item.getKey()) { case AND: // 如果是 and 关系，同步执行 System.out.println("execute key = " + 1); if (!and(dto, ruleList)) { returnfalse; } break; case OR: // 如果是 or 关系，并行执行 System.out.println("execute key = " + 0); if (!or(dto, ruleList)) { returnfalse; } break; default: break; } } returntrue; } privatebooleanand(RuleDto dto, List<BaseRule> ruleList){ for (BaseRule rule : ruleList) { boolean execute = rule.execute(dto); if (!execute) { // and 关系匹配失败一次，返回 false returnfalse; } } // and 关系全部匹配成功，返回 true returntrue; } privatebooleanor(RuleDto dto, List<BaseRule> ruleList){ for (BaseRule rule : ruleList) { boolean execute = rule.execute(dto); if (execute) { // or 关系匹配到一个就返回 true returntrue; } } // or 关系一个都匹配不到就返回 false returnfalse; } }

执行器的调用

public classRuleServiceTest{ @org.junit.Test publicvoidexecute(){ //规则执行器 //优点：比较简单，每个规则可以独立，将规则，数据，执行器拆分出来，调用方比较规整 //缺点：数据依赖公共传输对象 dto //1. 定义规则 init rule AgeRule ageRule = new AgeRule(); NameRule nameRule = new NameRule(); NationalityRule nationalityRule = new NationalityRule(); AddressRule addressRule = new AddressRule(); SubjectRule subjectRule = new SubjectRule(); //2. 构造需要的数据 create dto RuleDto dto = new RuleDto(); dto.setAge(5); dto.setName("张三"); dto.setAddress("北京"); dto.setSubject("数学");; //3. 通过以链式调用构建和执行 rule execute boolean ruleResult = RuleService .create() .and(Arrays.asList(nationalityRule, nameRule, addressRule)) .or(Arrays.asList(ageRule, subjectRule)) .execute(dto); System.out.println("this student rule execute result :" + ruleResult); } }