使用规则执行器替换IF条件判断

业务场景

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

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

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

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

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

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

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

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

规则执行器

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

规则执行器的设计

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

业务数据

@EqualsAndHashCode
@Data
@Builder
@AllArgsConstructor
@NoArgsConstructor
@ToString
public class RuleDTO {

    private String address;

    private Integer age;

    private String name;

    private String subject;
}

@EqualsAndHashCode
@Data
@SuperBuilder
@AllArgsConstructor
@NoArgsConstructor
@ToString
public class SubjectDTO extends RuleDTO {

    private String sort;

}

@EqualsAndHashCode
@Data
@SuperBuilder
@AllArgsConstructor
@NoArgsConstructor
@ToString
public class NationalityRuleDTO extends RuleDTO {
    private String nationality;

}

public class RuleConstant {

    public static final String MATCH_ADDRESS_START = "北京";

    public static final String MATCH_NATIONALITY_START = "中国";

    public static final String MATCH_SUBJECT_MATH = "数学";

    public static final String MATCH_SUBJECT_SCIENCE = "理科";
}

规则抽象

public interface BaseRule {

    boolean execute(RuleDTO dto);
}

规则模板

public abstract class AbstractRule implements BaseRule {

    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;
    }
}

具体规则

实例一

public class AddressRule extends AbstractRule {
    @Override
    public boolean execute(RuleDTO dto) {
        System.out.println("AddressRule invoke!");
        if (dto.getAddress().startsWith(RuleConstant.MATCH_ADDRESS_START)) {
            return true;
        }
        return false;
    }
}

实例二

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


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

实例三

public class AgeRule extends AbstractRule {

    @Override
    protected <T> boolean executeRule(T t) {
        System.out.println("AgeRule invoke!");
        RuleDTO ruleDTO = (RuleDTO) t;
        if (ruleDTO.getAge() > 18) {
            return true;
        }
        return false;
    }
}

实例四

public class NameRule extends AbstractRule {

    @Override
    protected <T> boolean executeRule(T t) {
        System.out.println("NameRule invoke!");
        RuleDTO ruleDTO = (RuleDTO) t;
        if (ruleDTO.getName().contains("张")) {
            return true;
        }
        return false;
    }
}

实例五

public class SubjectRule extends AbstractRule {

    @Override
    protected <T> T convert(RuleDTO dto) {
        SubjectDTO subjectDTO = new SubjectDTO();
        if (StringUtils.equals(dto.getSubject(), RuleConstant.MATCH_SUBJECT_MATH)) {
            subjectDTO.setSort(RuleConstant.MATCH_SUBJECT_SCIENCE);
        }
        return (T) subjectDTO;
    }

    @Override
    protected <T> boolean executeRule(T t) {
        SubjectDTO subjectDTO = (SubjectDTO) t;
        System.out.println("SubjectRule invoke!");
        if (StringUtils.equals(subjectDTO.getSort(), RuleConstant.MATCH_SUBJECT_SCIENCE)) {
            return true;
        }
        return false;
    }
}

执行器构建

public class RuleService {

    private Map<Integer, List<BaseRule>> hashMap = new HashMap<>();

    private static final int AND = 1;
    private static final int OR = 0;

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


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

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

    public boolean execute(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)) {
                        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<BaseRule> ruleList) {
        for (BaseRule rule : ruleList) {
            boolean execute = rule.execute(dto);
            if (!execute) {
                // and 关系匹配失败一次，返回 false
                return false;
            }
        }
        // and 关系全部匹配成功，返回 true
        return true;
    }

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

执行器调用

public class RuleServiceTest {
    @Test
    public void execute() {
        //规则执行器
        //优点：比较简单，每个规则可以独立，将规则，数据，执行器拆分出来，调用方比较规整
        //缺点：数据依赖公共传输对象 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.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);
    }
}

执行结果

总结

规则执行器的优点和缺点

优点：

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

缺点：

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

使用规则执行器替换IF条件判断

规则执行器的设计

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

业务数据

规则抽象

规则模板

具体规则

执行器构建

执行器调用

总结

优点：

缺点：

继续阅读

CSU 1561 (More) Multiplication

CSU 1563 Lexicography

HDU 4721 Food and Productivity

ZOJ 1041 Transmitters

CSU 1562 Fun House

CodeChef PALPROB Palindromeness

UVA 10344- 23 out of 5

ZOJ 1104 Leaps Tall Buildings

HDU 2821 Pusher

UVA 1401 Remember the Word

ZOJ 2748 Free Kick

CSU 1567 Reverse Rot

JAVA 系列——>开发工具IntelliJ IDEA的安装以及配置、快捷键IDEA 简介

UVA 519 Puzzle (II)

磁盘结构及在Linux中的命名

详解STM32单片机的堆栈