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

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

引入依賴

我們創(chuàng)建一個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)建一個名為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;
}
}

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

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

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

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

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

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

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

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

publicenumCustomerType{
LOYAL,NEW,DISSATISFIED;

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

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

@Getter
@Setter
publicclassOrderDiscount{

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

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

定義drools 規(guī)則

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

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

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

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

每個規(guī)則when-then語法來定義規(guī)則的條件。

根據(jù)訂單請求的輸入值，我們正在為結(jié)果添加折扣。如果規(guī)則表達式匹配，每個規(guī)則都會向全局結(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
//當客戶年齡在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
//當客戶類型是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)建一個名為OrderDiscountService 的服務(wù)類，如下：。

@Service
publicclassOrderDiscountService{

@Autowired
privateKieContainerkieContainer;

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

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

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

使用insert()方法將請求對象傳遞給 drl 文件。

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

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

添加Controller

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

@RestController
publicclassOrderDiscountController{

@Autowired
privateOrderDiscountServiceorderDiscountService;

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

測試一下

運行 spring boot 應(yīng)用程序并通過發(fā)送客戶訂單請求 JSON 來訪問 REST API 端點。

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

總結(jié)

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

審核編輯：劉清