现在有这么个需求，网上购物，需要根据不同的规则计算商品折扣，比如VIP客户增加5%的折扣，购买金额超过1000元的增加10%的折扣等，而且这些规则可能随时发生变化，甚至增加新的规则。面对这个需求，你该怎么实现呢？难道是计算规则一变，就要修改业务代码，重新测试，上线吗。
其实，我们可以通过规则引擎来实现，Drools 就是一个开源的业务规则引擎，可以很容易地与 spring boot 应用程序集成，那本文就用Drools来实现一下上面说的需求吧。
引入依赖
我们创建一个spring boot应用程序，pom中添加drools相关的依赖，如下：
<dependency> <groupId>org.drools</groupId> <artifactId>drools-core</artifactId> <version>7.59.0.Final</version> </dependency> <dependency> <groupId>org.drools</groupId> <artifactId>drools-compiler</artifactId> <version>7.59.0.Final</version> </dependency> <dependency> <groupId>org.drools</groupId> <artifactId>drools-decisiontables</artifactId> <version>7.59.0.Final</version> </dependency>
基于 Spring Boot + MyBatis Plus + Vue & Element 实现的后台管理系统 + 用户小程序，支持 RBAC 动态权限、多租户、数据权限、工作流、三方登录、支付、短信、商城等功能
Drools配置类
创建一个名为DroolsConfig的配置 java 类。
@Configuration public class DroolsConfig { // 制定规则文件的路径 private static final String RULES_CUSTOMER_RULES_DRL = "rules/customer-discount.drl"; private static final KieServices kieServices = KieServices.Factory.get(); @Bean public KieContainer kieContainer() { KieFileSystem kieFileSystem = kieServices.newKieFileSystem(); kieFileSystem.write(ResourceFactory.newClassPathResource(RULES_CUSTOMER_RULES_DRL)); KieBuilder kb = kieServices.newKieBuilder(kieFileSystem); kb.buildAll(); KieModule kieModule = kb.getKieModule(); KieContainer kieContainer = kieServices.newKieContainer(kieModule.getReleaseId()); return kieContainer; } }
- 定义了一个
KieContainer的Spring Bean，KieContainer用于通过加载应用程序的/resources文件夹下的规则文件来构建规则引擎。 - 创建
KieFileSystem实例并配置规则引擎并从应用程序的资源目录加载规则的DRL文件。 - 使用
KieBuilder实例来构建drools模块。我们可以使用KieSerive单例实例来创建KieBuilder实例。 - 最后，使用
KieService创建一个KieContainer并将其配置为spring bean。
基于 Spring Cloud Alibaba + Gateway + Nacos + RocketMQ + Vue & Element 实现的后台管理系统 + 用户小程序，支持 RBAC 动态权限、多租户、数据权限、工作流、三方登录、支付、短信、商城等功能
添加业务Model
创建一个订单对象OrderRequest，这个类中的字段后续回作为输入信息发送给定义的drools规则中，用来计算给定客户订单的折扣金额。
@Getter @Setter public class OrderRequest { /** * 客户号 */ private String customerNumber; /** * 年龄 */ private Integer age; /** * 订单金额 */ private Integer amount; /** * 客户类型 */ private CustomerType customerType; }
此外，定义一个客户类型CustomerType 的枚举，规则引擎会根据该值计算客户订单折扣百分比，如下所示。
public enum CustomerType { LOYAL, NEW, DISSATISFIED; public String getValue() { return this.toString(); } }
最后，创建一个订单折扣类 OrderDiscount ，用来表示计算得到的最终的折扣，如下所示。
@Getter @Setter public class OrderDiscount { /** * 折扣 */ private Integer discount = 0; }
我们将使用上述响应对象返回计算出的折扣。
定义drools 规则
前面的DroolsConfig类中指定drools规则的目录，现在我们在/src/main/resources/rules目录下添加customer-discount.drl文件，在里面定义对应的规则。
这个drl文件虽然不是java文件，但还是很容易看懂的。
- 我们使用了一个名为
orderDiscount的全局参数，可以在多个规则之间共享。关注工众号：码猿技术专栏，回复关键词：1111 获取阿里内部java性能调优手册！ drl文件可以包含一个或多个规则。我们可以使用mvel语法来指定规则。此外，每个规则使用rule关键字进行描述。- 每个规则
when-then语法来定义规则的条件。 - 根据订单请求的输入值，我们正在为结果添加折扣。如果规则表达式匹配，每个规则都会向全局结果变量添加额外的折扣。
完整的规则源码如下：
import com.alvin.drools.model.OrderRequest; import com.alvin.drools.model.CustomerType; global com.alvin.drools.model.OrderDiscount orderDiscount; dialect "mvel" // 规则1: 根据年龄判断 rule "Age based discount" when // 当客户年龄在20岁以下或者50岁以上 OrderRequest(age < 20 || age > 50) then // 则添加10%的折扣 System.out.println("==========Adding 10% discount for Kids/ senior customer============="); orderDiscount.setDiscount(orderDiscount.getDiscount() + 10); end // 规则2： 根据客户类型的规则 rule "Customer type based discount - Loyal customer" when // 当客户类型是LOYAL OrderRequest(customerType.getValue == "LOYAL") then // 则增加5%的折扣 System.out.println("==========Adding 5% discount for LOYAL customer============="); orderDiscount.setDiscount(orderDiscount.getDiscount() + 5); end rule "Customer type based discount - others" when OrderRequest(customerType.getValue != "LOYAL") then System.out.println("==========Adding 3% discount for NEW or DISSATISFIED customer============="); orderDiscount.setDiscount(orderDiscount.getDiscount() + 3); end rule "Amount based discount" when OrderRequest(amount > 1000L) then System.out.println("==========Adding 5% discount for amount more than 1000$============="); orderDiscount.setDiscount(orderDiscount.getDiscount() + 5); end
添加Service层
创建一个名为OrderDiscountService 的服务类，如下：。
@Service public class OrderDiscountService { @Autowired private KieContainer kieContainer; public OrderDiscount getDiscount(OrderRequest orderRequest) { OrderDiscount orderDiscount = new OrderDiscount(); // 开启会话 KieSession kieSession = kieContainer.newKieSession(); // 设置折扣对象 kieSession.setGlobal("orderDiscount", orderDiscount); // 设置订单对象 kieSession.insert(orderRequest); // 触发规则 kieSession.fireAllRules(); // 中止会话 kieSession.dispose(); return orderDiscount; } }
- 注入
KieContainer实例并创建一个KieSession实例。 - 设置了一个
OrderDiscount类型的全局参数，它将保存规则执行结果。 - 使用
insert()方法将请求对象传递给drl文件。 - 调用
fireAllRules()方法触发所有规则。 - 最后通过调用
KieSession的dispose()方法终止会话。
添加Controller
创建一个名为OrderDiscountController 的Controller类，具体代码如下：
@RestController public class OrderDiscountController { @Autowired private OrderDiscountService orderDiscountService; @PostMapping("/get-discount") public ResponseEntity<OrderDiscount> getDiscount(@RequestBody OrderRequest orderRequest) { OrderDiscount discount = orderDiscountService.getDiscount(orderRequest); return new ResponseEntity<>(discount, HttpStatus.OK); } }
测试一下
运行 spring boot 应用程序并通过发送客户订单请求 JSON 来访问 REST API 端点。
- 对于年龄 < 20 且金额 > 1000 的
LOYAL客户类型，我们应该根据我们定义的规则获得20%的折扣。
总结
我们通过drools规则引擎简单实现了这样一个折扣的业务，现在产品经理说要你加一条规则，比如地址是杭州的折扣加10%，你就直接改这个drl文件，其他时间用来摸鱼就好了，哈哈~~。更多关于drools的用法大家可以去官网探索。
