Spring Boot+规则引擎Drools

现在有这么个需求，网上购物，需要根据不同的规则计算商品折扣，比如VIP客户增加5%的折扣，购买金额超过1000元的增加10%的折扣等，而且这些规则可能随时发生变化，甚至增加新的规则。面对这个需求，你该怎么实现呢？难道是计算规则一变，就要修改业务代码，重新测试，上线吗。

其实，我们可以通过规则引擎来实现，Drools 就是一个开源的业务规则引擎，可以很容易地与 spring boot 应用程序集成，那本文就用Drools来实现一下上面说的需求吧。

引入依赖

我们创建一个spring boot应用程序，pom中添加drools相关的依赖，如下：

org.drools
drools-core
7.59.0.Final


org.drools
drools-compiler
7.59.0.Final


org.drools
drools-decisiontables
7.59.0.Final

Drools配置类

创建一个名为DroolsConfig的配置 java 类。

@Configuration
publicclassDroolsConfig{
//制定规则文件的路径
privatestaticfinalStringRULES_CUSTOMER_RULES_DRL="rules/customer-discount.drl";
privatestaticfinalKieServiceskieServices=KieServices.Factory.get();

@Bean
publicKieContainerkieContainer(){
KieFileSystemkieFileSystem=kieServices.newKieFileSystem();
kieFileSystem.write(ResourceFactory.newClassPathResource(RULES_CUSTOMER_RULES_DRL));
KieBuilderkb=kieServices.newKieBuilder(kieFileSystem);
kb.buildAll();
KieModulekieModule=kb.getKieModule();
KieContainerkieContainer=kieServices.newKieContainer(kieModule.getReleaseId());
returnkieContainer;
}
}

定义了一个 KieContainer的Spring Bean ，KieContainer用于通过加载应用程序的/resources文件夹下的规则文件来构建规则引擎。

创建KieFileSystem实例并配置规则引擎并从应用程序的资源目录加载规则的 DRL 文件。

使用KieBuilder实例来构建 drools 模块。我们可以使用KieSerive单例实例来创建 KieBuilder 实例。

最后，使用 KieService 创建一个 KieContainer 并将其配置为 spring bean。

添加业务Model

创建一个订单对象OrderRequest，这个类中的字段后续回作为输入信息发送给定义的drools规则中，用来计算给定客户订单的折扣金额。

@Getter
@Setter
publicclassOrderRequest{
/**
*客户号
*/
privateStringcustomerNumber;
/**
*年龄
*/
privateIntegerage;
/**
*订单金额
*/
privateIntegeramount;
/**
*客户类型
*/
privateCustomerTypecustomerType;
}

此外，定义一个客户类型CustomerType 的枚举，规则引擎会根据该值计算客户订单折扣百分比，如下所示。

publicenumCustomerType{
LOYAL,NEW,DISSATISFIED;

publicStringgetValue(){
returnthis.toString();
}
}

最后，创建一个订单折扣类 OrderDiscount ，用来表示计算得到的最终的折扣，如下所示。

@Getter
@Setter
publicclassOrderDiscount{

/**
*折扣
*/
privateIntegerdiscount=0;
}

我们将使用上述响应对象返回计算出的折扣。

定义drools 规则

前面的DroolsConfig类中指定drools规则的目录，现在我们在/src/main/resources/rules目录下添加customer-discount.drl文件，在里面定义对应的规则。

这个drl文件虽然不是java文件，但还是很容易看懂的。

我们使用了一个名为orderDiscount 的全局参数，可以在多个规则之间共享。关注工众号：码猿技术专栏，回复关键词：1111 获取阿里内部java性能调优手册！

drl 文件可以包含一个或多个规则。我们可以使用mvel语法来指定规则。此外，每个规则使用rule关键字进行描述。

每个规则when-then语法来定义规则的条件。

根据订单请求的输入值，我们正在为结果添加折扣。如果规则表达式匹配，每个规则都会向全局结果变量添加额外的折扣。

完整的规则源码如下：

importcom.alvin.drools.model.OrderRequest;
importcom.alvin.drools.model.CustomerType;
globalcom.alvin.drools.model.OrderDiscountorderDiscount;

dialect"mvel"

//规则1:根据年龄判断
rule"Agebaseddiscount"
when
//当客户年龄在20岁以下或者50岁以上
OrderRequest(age< 20 || age >50)
then
//则添加10%的折扣
System.out.println("==========Adding10%discountforKids/seniorcustomer=============");
orderDiscount.setDiscount(orderDiscount.getDiscount()+10);
end

//规则2：根据客户类型的规则
rule"Customertypebaseddiscount-Loyalcustomer"
when
//当客户类型是LOYAL
OrderRequest(customerType.getValue=="LOYAL")
then
//则增加5%的折扣
System.out.println("==========Adding5%discountforLOYALcustomer=============");
orderDiscount.setDiscount(orderDiscount.getDiscount()+5);
end

rule"Customertypebaseddiscount-others"
when
OrderRequest(customerType.getValue!="LOYAL")
then
System.out.println("==========Adding3%discountforNEWorDISSATISFIEDcustomer=============");
orderDiscount.setDiscount(orderDiscount.getDiscount()+3);
end

rule"Amountbaseddiscount"
when
OrderRequest(amount>1000L)
then
System.out.println("==========Adding5%discountforamountmorethan1000$=============");
orderDiscount.setDiscount(orderDiscount.getDiscount()+5);
end

添加Service层

创建一个名为OrderDiscountService 的服务类，如下：。

@Service
publicclassOrderDiscountService{

@Autowired
privateKieContainerkieContainer;

publicOrderDiscountgetDiscount(OrderRequestorderRequest){
OrderDiscountorderDiscount=newOrderDiscount();
//开启会话
KieSessionkieSession=kieContainer.newKieSession();
//设置折扣对象
kieSession.setGlobal("orderDiscount",orderDiscount);
//设置订单对象
kieSession.insert(orderRequest);
//触发规则
kieSession.fireAllRules();
//中止会话
kieSession.dispose();
returnorderDiscount;
}
}

注入KieContainer实例并创建一个KieSession实例。

设置了一个OrderDiscount类型的全局参数，它将保存规则执行结果。

使用insert()方法将请求对象传递给 drl 文件。

调用fireAllRules()方法触发所有规则。

最后通过调用KieSession 的dispose()方法终止会话。

添加Controller

创建一个名为OrderDiscountController 的Controller类，具体代码如下：

@RestController
publicclassOrderDiscountController{

@Autowired
privateOrderDiscountServiceorderDiscountService;

@PostMapping("/get-discount")
publicResponseEntitygetDiscount(@RequestBodyOrderRequestorderRequest){
OrderDiscountdiscount=orderDiscountService.getDiscount(orderRequest);
returnnewResponseEntity<>(discount,HttpStatus.OK);
}
}

测试一下

运行 spring boot 应用程序并通过发送客户订单请求 JSON 来访问 REST API 端点。

对于年龄 < 20 且金额 > 1000 的 LOYAL 客户类型，我们应该根据我们定义的规则获得 20% 的折扣。

总结

我们通过drools规则引擎简单实现了这样一个折扣的业务，现在产品经理说要你加一条规则，比如地址是杭州的折扣加10%，你就直接改这个drl文件，其他时间用来摸鱼就好了，哈哈~~。更多关于drools的用法大家可以去官网探索。

审核编辑：刘清