前言
現在有這麼個需求,網上購物,需要根據不同的規則計算商品折扣,比如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>
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
。
添加業務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
的全局參數,可以在多個規則之間共享。 -
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
的用法大家可以去官網探索。