現(xiàn)在有這么個(gè)需求，網(wǎng)上購(gòu)物，需要根據(jù)不同的規(guī)則計(jì)算商品折扣，比如VIP客戶增加5%的折扣，購(gòu)買金額超過(guò)1000元的增加10%的折扣等，而且這些規(guī)則可能隨時(shí)發(fā)生變化，甚至增加新的規(guī)則。面對(duì)這個(gè)需求，你該怎么實(shí)現(xiàn)呢？難道是計(jì)算規(guī)則一變，就要修改業(yè)務(wù)代碼，重新測(cè)試，上線嗎。

其實(shí)，我們可以通過(guò)規(guī)則引擎來(lái)實(shí)現(xiàn)，Drools 就是一個(gè)開源的業(yè)務(wù)規(guī)則引擎，可以很容易地與 spring boot 應(yīng)用程序集成，那本文就用Drools來(lái)實(shí)現(xiàn)一下上面說(shuō)的需求吧。

引入依賴

我們創(chuàng)建一個(gè)spring boot應(yīng)用程序，pom中添加drools相關(guān)的依賴，如下：

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配置類

創(chuàng)建一個(gè)名為DroolsConfig的配置 java 類。

@Configuration
publicclassDroolsConfig{
//制定規(guī)則文件的路徑
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;
}
}

定義了一個(gè) KieContainer的Spring Bean ，KieContainer用于通過(guò)加載應(yīng)用程序的/resources文件夾下的規(guī)則文件來(lái)構(gòu)建規(guī)則引擎。

創(chuàng)建KieFileSystem實(shí)例并配置規(guī)則引擎并從應(yīng)用程序的資源目錄加載規(guī)則的 DRL 文件。

使用KieBuilder實(shí)例來(lái)構(gòu)建 drools 模塊。我們可以使用KieSerive單例實(shí)例來(lái)創(chuàng)建 KieBuilder 實(shí)例。

最后，使用 KieService 創(chuàng)建一個(gè) KieContainer 并將其配置為 spring bean。

添加業(yè)務(wù)Model

創(chuàng)建一個(gè)訂單對(duì)象OrderRequest，這個(gè)類中的字段后續(xù)回作為輸入信息發(fā)送給定義的drools規(guī)則中，用來(lái)計(jì)算給定客戶訂單的折扣金額。

@Getter
@Setter
publicclassOrderRequest{
/**
*客戶號(hào)
*/
privateStringcustomerNumber;
/**
*年齡
*/
privateIntegerage;
/**
*訂單金額
*/
privateIntegeramount;
/**
*客戶類型
*/
privateCustomerTypecustomerType;
}

此外，定義一個(gè)客戶類型CustomerType 的枚舉，規(guī)則引擎會(huì)根據(jù)該值計(jì)算客戶訂單折扣百分比，如下所示。

publicenumCustomerType{
LOYAL,NEW,DISSATISFIED;

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

最后，創(chuàng)建一個(gè)訂單折扣類 OrderDiscount ，用來(lái)表示計(jì)算得到的最終的折扣，如下所示。

@Getter
@Setter
publicclassOrderDiscount{

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

我們將使用上述響應(yīng)對(duì)象返回計(jì)算出的折扣。

定義drools 規(guī)則

前面的DroolsConfig類中指定drools規(guī)則的目錄，現(xiàn)在我們?cè)?src/main/resources/rules目錄下添加customer-discount.drl文件，在里面定義對(duì)應(yīng)的規(guī)則。

這個(gè)drl文件雖然不是java文件，但還是很容易看懂的。

我們使用了一個(gè)名為orderDiscount 的全局參數(shù)，可以在多個(gè)規(guī)則之間共享。關(guān)注工眾號(hào)：碼猿技術(shù)專欄，回復(fù)關(guān)鍵詞：1111 獲取阿里內(nèi)部java性能調(diào)優(yōu)手冊(cè)！

drl 文件可以包含一個(gè)或多個(gè)規(guī)則。我們可以使用mvel語(yǔ)法來(lái)指定規(guī)則。此外，每個(gè)規(guī)則使用rule關(guān)鍵字進(jìn)行描述。

每個(gè)規(guī)則when-then語(yǔ)法來(lái)定義規(guī)則的條件。

根據(jù)訂單請(qǐng)求的輸入值，我們正在為結(jié)果添加折扣。如果規(guī)則表達(dá)式匹配，每個(gè)規(guī)則都會(huì)向全局結(jié)果變量添加額外的折扣。

完整的規(guī)則源碼如下：

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

dialect"mvel"

//規(guī)則1:根據(jù)年齡判斷
rule"Agebaseddiscount"
when
//當(dāng)客戶年齡在20歲以下或者50歲以上
OrderRequest(age50)
then
//則添加10%的折扣
System.out.println("==========Adding10%discountforKids/seniorcustomer=============");
orderDiscount.setDiscount(orderDiscount.getDiscount()+10);
end

//規(guī)則2：根據(jù)客戶類型的規(guī)則
rule"Customertypebaseddiscount-Loyalcustomer"
when
//當(dāng)客戶類型是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層

創(chuàng)建一個(gè)名為OrderDiscountService 的服務(wù)類，如下：。

@Service
publicclassOrderDiscountService{

@Autowired
privateKieContainerkieContainer;

publicOrderDiscountgetDiscount(OrderRequestorderRequest){
OrderDiscountorderDiscount=newOrderDiscount();
//開啟會(huì)話
KieSessionkieSession=kieContainer.newKieSession();
//設(shè)置折扣對(duì)象
kieSession.setGlobal("orderDiscount",orderDiscount);
//設(shè)置訂單對(duì)象
kieSession.insert(orderRequest);
//觸發(fā)規(guī)則
kieSession.fireAllRules();
//中止會(huì)話
kieSession.dispose();
returnorderDiscount;
}
}

注入KieContainer實(shí)例并創(chuàng)建一個(gè)KieSession實(shí)例。

設(shè)置了一個(gè)OrderDiscount類型的全局參數(shù)，它將保存規(guī)則執(zhí)行結(jié)果。

使用insert()方法將請(qǐng)求對(duì)象傳遞給 drl 文件。

調(diào)用fireAllRules()方法觸發(fā)所有規(guī)則。

最后通過(guò)調(diào)用KieSession 的dispose()方法終止會(huì)話。

添加Controller

創(chuàng)建一個(gè)名為OrderDiscountController 的Controller類，具體代碼如下：

@RestController
publicclassOrderDiscountController{

@Autowired
privateOrderDiscountServiceorderDiscountService;

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

測(cè)試一下

運(yùn)行 spring boot 應(yīng)用程序并通過(guò)發(fā)送客戶訂單請(qǐng)求 JSON 來(lái)訪問(wèn) REST API 端點(diǎn)。

對(duì)于年齡 < 20 且金額 > 1000 的 LOYAL 客戶類型，我們應(yīng)該根據(jù)我們定義的規(guī)則獲得 20% 的折扣。

總結(jié)

我們通過(guò)drools規(guī)則引擎簡(jiǎn)單實(shí)現(xiàn)了這樣一個(gè)折扣的業(yè)務(wù)，現(xiàn)在產(chǎn)品經(jīng)理說(shuō)要你加一條規(guī)則，比如地址是杭州的折扣加10%，你就直接改這個(gè)drl文件，其他時(shí)間用來(lái)摸魚就好了，哈哈~~。更多關(guān)于drools的用法大家可以去官網(wǎng)探索。

審核編輯：劉清