一文帶你學會規則引擎Drools的應用

Posted on by

前言

現在有這麼個需求，網上購物，需要根據不同的規則計算商品折扣，比如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的用法大傢可以去官網探索。

到此這篇關於一文帶你學會規則引擎Drools的應用的文章就介紹到這瞭,更多相關規則引擎Drools內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：