[Drools]JAVA規則引擎 -- Drools 2

上一篇文章 http://blog.csdn.net/quzishen/archive/2011/01/25/6163012.aspx 描述了一些常用的drools的語法標籤和一個模擬實例即發送積分的場景，這一片優化了一下代碼，在此貼一下，希望有這方面使用經驗的朋友多多交流溝通，指正不足。

通常而言，習慣上我們將規則放到文件系統中，比如以drl結尾的規則文件，現在我們要擴充一下，使其放到數據庫中，以供多臺服務器同時使用，同時依然保留文件系統的支持。

先看下一個接口：

 

/**
 * 規則接口
 * @author quzishen
 */
public interface PointRuleEngine {

    /**
     * 初始化規則引擎
     */
    public void initEngine();

    /**
     * 刷新規則引擎中的規則
     */
    public void refreshEnginRule();

    /**
     * 執行規則引擎
     * @param pointDomain 積分Fact
     */
    public void executeRuleEngine(final PointDomain pointDomain);
}

 

實現過程沒有任何難度，兩種方式封裝過程只在於讀取規則的方式不同，代碼很簡單：

 

package com.drools.demo.point;

import <a href="http://lib.csdn.net/base/17" class='replace_word' title="Java EE知識庫" target='_blank' style='color:#df3434; font-weight:bold;'>Java</a>.io.File;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.Reader;
import java.io.StringReader;
import java.util.ArrayList;
import java.util.List;
import org.drools.RuleBase;
import org.drools.StatefulSession;
import org.drools.compiler.PackageBuilder;
import org.drools.spi.Activation;

/**
 * 規則接口實現類
 *
 * @author quzishen
 */
public class PointRuleEngineImpl implements PointRuleEngine {
    // ~~~ instance filed begin
    /** RuleBase */
    private RuleBase ruleBase;
    // ~~~ instance filed end

    /*
     * (non-Javadoc)
     * @see com.drools.demo.point.PointRuleEngine#initEngine()
     */
    public void initEngine() {
        // 設置時間格式
        System.setProperty("drools.dateformat", "yyyy-MM-dd HH:mm:ss");
        try {
            synchronized (this) {
                ruleBase = RuleBaseFacatory.getRuleBase();
                // 優先從DB加載規則，如果沒有加載到或者加載錯誤，則從文件系統加載
                PackageBuilder backageBuilder = getPackBuilderFromDrlDB();
                backageBuilder = null == backageBuilder ? getPackageBuilderFromDrlFile()
                        : backageBuilder;

                ruleBase.addPackages(backageBuilder.getPackages());
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /*
     * (non-Javadoc)
     * @see com.drools.demo.point.PointRuleEngine#refreshEnginRule()
     */
    public void refreshEnginRule() {
        ruleBase = RuleBaseFacatory.getRuleBase();
        synchronized (ruleBase) {
            // 刪除所有的添加的Package
            org.drools.rule.Package[] packages = ruleBase.getPackages();
            for (org.drools.rule.Package pg : packages) {
                ruleBase.removePackage(pg.getName());
            }

            // 重新初始化規則引擎
            initEngine();
        }
    }

    /*
     * (non-Javadoc)
     * @see com.drools.demo.point.PointRuleEngine#executeRuleEngine(com.drools.demo.point.PointDomain)
     */
    public void executeRuleEngine(final PointDomain pointDomain) {
        if (null == ruleBase.getPackages() || 0 == ruleBase.getPackages().length) {
            return;
        }

        StatefulSession statefulSession = ruleBase.newStatefulSession();
        statefulSession.insert(pointDomain);

        // fire
        statefulSession.fireAllRules(new org.drools.spi.AgendaFilter() {
            public boolean accept(Activation activation) {
                return !activation.getRule().getName().contains("_test");
            }
        });

        statefulSession.dispose();
    }

    /**
     * 從Drl規則文件中讀取規則
     *
     * @return
     * @throws Exception
     */
    private PackageBuilder getPackageBuilderFromDrlFile() {
        // 裝載規則文件
        List<Reader> readers;
        try {
            readers = buildReadersFromDrlFile();
            // 裝載PackageBuilder
            return buildPackageBuilder(readers);
        } catch (FileNotFoundException e) {
            e.printStackTrace();
            return null;
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    /**
     * 從Drl規則DB中讀取規則
     *
     * @return
     * @throws Exception
     */
    private PackageBuilder getPackBuilderFromDrlDB() {
        // 裝載規則
        List<Reader> readers = buildReadersFromDrlDB();

        // 裝載PackageBuilder
        try {
            return buildPackageBuilder(readers);
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    /**
     * 裝載db中的規則到List<Reader>
     *
     * @return
     */
    private List<Reader> buildReadersFromDrlDB() {
        List<Reader> readers = new ArrayList<Reader>();
        // 獲取腳本
        List<DroolsRuleDomain> drlRuleDomains = getRuleFromDB();

        if (null == drlRuleDomains) {
            return readers;
        }

        for (DroolsRuleDomain droolsRuleDomain : drlRuleDomains) {
            String ruleContext = droolsRuleDomain.getRuleContext();

            Reader br = new StringReader(ruleContext);
            readers.add(br);
        }
        return readers;
    }

    /**
     * 裝載PackageBuilder
     *
     * @param readers
     * @return
     * @throws Exception
     */
    private PackageBuilder buildPackageBuilder(List<Reader> readers)
            throws Exception {
        if (null == readers || 0 == readers.size()) {
            return null;
        }

        PackageBuilder backageBuilder = new PackageBuilder();
        for (Reader r : readers) {
            backageBuilder.addPackageFromDrl(r);
        }

        // 檢查腳本是否有問題
        if (backageBuilder.hasErrors()) {
            throw new Exception(backageBuilder.getErrors().toString());
        }

        return backageBuilder;
    }

    /**
     * 裝載規則文件到Reader中
     *
     * @return
     * @throws FileNotFoundException
     */
    private List<Reader> buildReadersFromDrlFile() throws FileNotFoundException {
        // 獲取腳本文件
        List<String> drlFilePath = getRuleDrlFile();
        // 裝載腳本文件
        return readRuleFromDrlFile(drlFilePath);
    }

    /**
     * 從規則文件中讀取規則
     *
     * @param drlFilePath 腳本文件路徑
     * @return
     * @throws FileNotFoundException
     */
    private List<Reader> readRuleFromDrlFile(List<String> drlFilePath)
            throws FileNotFoundException {
        if (null == drlFilePath || 0 == drlFilePath.size()) {
            return null;
        }

        List<Reader> readers = new ArrayList<Reader>();

        for (String ruleFilePath : drlFilePath) {
            readers.add(new FileReader(new File(ruleFilePath)));
        }

        return readers;
    }

    /**
     * 從數據庫中獲取規則腳本內容
     *
     * @return
     */
    private List<DroolsRuleDomain> getRuleFromDB() {
        // 測試代碼
        List<DroolsRuleDomain> droolsRuleDomains = new ArrayList<DroolsRuleDomain>();

        DroolsRuleDomain d1 = new DroolsRuleDomain();
        d1.setId(1);
        d1.setRuleContext("package com.drools.demo.point" + "/n" +
                "import com.drools.demo.point.PointDomain;" + "/n" +
                "rule birthdayPoint" + "/n" +
                "// 過生日，則加10分，並且將當月交易比數翻倍後再計算積分" + "/n" +
                "salience 100" + "/n" +
                "lock-on-active true" + "/n" +
                "when" + "/n" +
                "$pointDomain : PointDomain(birthDay == true)" + "/n" +
                "then" + "/n" +
                "pointDomain.setPoint(

pointDomain.getPoint()+10);" + "/n" +                   "pointDomain.recordPointLog(

pointDomain.getUserName(),/"birthdayPoint/");" + "/n" +
                "end");

        d1.setRuleName("testRule");
        d1.setVersion(1);

        droolsRuleDomains.add(d1);

        return droolsRuleDomains;
    }

    /**
     * 獲取規則文件
     *
     * @return
     */
    private List<String> getRuleDrlFile() {
        List<String> drlFilePath = new ArrayList<String>();
        drlFilePath
                .add("D:/workspace2/DroolsDemo/src/com/drools/demo/point/addpoint.drl");
        drlFilePath
                .add("D:/workspace2/DroolsDemo/src/com/drools/demo/point/subpoint.drl");

        return drlFilePath;
    }
}

 

其中的getRuleFromDB() 和 getRuleDrlFile() 兩個方法即可以重寫以接入個人系統，現在其中編寫的是測試代碼。

其他的文件與上篇文章相同：

 

RuleBaseFacatory

 

package com.drools.demo.point;

import org.drools.RuleBase;
import org.drools.RuleBaseFactory;

/**
 * RuleBaseFacatory 單實例RuleBase生成工具
 * @author quzishen
 */
public class RuleBaseFacatory {
    private static RuleBase ruleBase;

    public static RuleBase getRuleBase(){
        return null != ruleBase ? ruleBase : RuleBaseFactory.newRuleBase();
    }
}

 

DroolsRuleDomain

 

package com.drools.demo.point;

/**
 * 規則內容domain
 *
 * @author quzishen
 */
public class DroolsRuleDomain {
    /** 數據庫記錄ID */
    private long id;
    /** 規則名稱 */
    private String ruleName;
    /** 規則正文  */
    private String ruleContext;
    /** 規則版本 */
    private int version;
    /** 規則腳本狀態 */
    private int status;

    public long getId() {
        return id;
    }

    public void setId(long id) {
        this.id = id;
    }

    public String getRuleName() {
        return ruleName;
    }

    public void setRuleName(String ruleName) {
        this.ruleName = ruleName;
    }

    public String getRuleContext() {
        return ruleContext;
    }

    public void setRuleContext(String ruleContext) {
        this.ruleContext = ruleContext;
    }

    public int getVersion() {
        return version;
    }

    public void setVersion(int version) {
        this.version = version;
    }

    public int getStatus() {
        return status;
    }

    public void setStatus(int status) {
        this.status = status;
    }

}

 

PointDomain

 

package com.drools.demo.point;


/**
 * 積分計算對象
 * @author quzishen
 */
public class PointDomain {
    // 用戶名
    private String userName;
    // 是否當日生日
    private boolean birthDay;
    // 增加積分數目
    private long point;
    // 當月購物次數
    private int buyNums;
    // 當月退貨次數
    private int backNums;
    // 當月購物總金額
    private double buyMoney;
    // 當月退貨總金額
    private double backMondy;
    // 當月信用卡還款次數
    private int billThisMonth;

    /**
     * 記錄積分發送流水，防止重複發放
     * @param userName 用戶名
     * @param type 積分發放類型
     */
    public void recordPointLog(String userName, String type){
        System.out.println("增加對"+userName+"的類型爲"+type+"的積分操作記錄.");
    }

    public String getUserName() {
        return userName;
    }

    public void setUserName(String userName) {
        this.userName = userName;
    }

    public boolean isBirthDay() {
        return birthDay;
    }

    public void setBirthDay(boolean birthDay) {
        this.birthDay = birthDay;
    }

    public long getPoint() {
        return point;
    }

    public void setPoint(long point) {
        this.point = point;
    }

    public int getBuyNums() {
        return buyNums;
    }

    public void setBuyNums(int buyNums) {
        this.buyNums = buyNums;
    }

    public int getBackNums() {
        return backNums;
    }

    public void setBackNums(int backNums) {
        this.backNums = backNums;
    }

    public double getBuyMoney() {
        return buyMoney;
    }

    public void setBuyMoney(double buyMoney) {
        this.buyMoney = buyMoney;
    }

    public double getBackMondy() {
        return backMondy;
    }

    public void setBackMondy(double backMondy) {
        this.backMondy = backMondy;
    }

    public int getBillThisMonth() {
        return billThisMonth;
    }

    public void setBillThisMonth(int billThisMonth) {
        this.billThisMonth = billThisMonth;
    }

}

 

addpoint.drl

 

package com.drools.demo.point

import com.drools.demo.point.PointDomain;

rule birthdayPoint
    // 過生日，則加10分，並且將當月交易比數翻倍後再計算積分
    salience 100
    lock-on-active true
    when
        $pointDomain : PointDomain(birthDay == true)
    then
        pointDomain.setPoint(

pointDomain.getPoint()+10);          pointDomain.setBuyNums(pointDomain.getBuyNums()*2);          pointDomain.setBuyMoney(pointDomain.getBuyMoney()*2);          pointDomain.setBillThisMonth(pointDomain.getBillThisMonth()*2);                    pointDomain.recordPointLog(pointDomain.getUserName(),"birthdayPoint");  end    rule billThisMonthPoint      // 2011-01-08 - 2011-08-08每月信用卡還款3次以上，每滿3筆贈送30分      salience 99      lock-on-active true      date-effective "2011-01-08 23:59:59"      date-expires "2011-08-08 23:59:59"      when          $pointDomain : PointDomain(billThisMonth >= 3)      then          pointDomain.setPoint(pointDomain.getPoint()+$pointDomain.getBillThisMonth()/3*30);          pointDomain.recordPointLog(pointDomain.getUserName(),"billThisMonthPoint");  end    rule buyMoneyPoint      // 當月購物總金額100以上，每100元贈送10分      salience 98      lock-on-active true      when          $pointDomain : PointDomain(buyMoney >= 100)      then          pointDomain.setPoint(pointDomain.getPoint()+ (int)$pointDomain.getBuyMoney()/100 * 10);          pointDomain.recordPointLog(pointDomain.getUserName(),"buyMoneyPoint");  end    rule buyNumsPoint      // 當月購物次數5次以上，每五次贈送50分      salience 97      lock-on-active true      when          $pointDomain : PointDomain(buyNums >= 5)      then          pointDomain.setPoint(pointDomain.getPoint()+$pointDomain.getBuyNums()/5 * 50);          pointDomain.recordPointLog(pointDomain.getUserName(),"buyNumsPoint");  end    rule allFitPoint      // 特別的，如果全部滿足了要求，則額外獎勵100分      salience 96      lock-on-active true      when          $pointDomain:PointDomain(buyNums >= 5 && billThisMonth >= 3 && buyMoney >= 100)      then          pointDomain.setPoint(pointDomain.getPoint()+ 100);          pointDomain.recordPointLog(

pointDomain.getUserName(),"allFitPoint");
end

 

subpoint.drl 與上一篇相同，請參見上一篇，此處省略篇幅略

測試代碼

Test

 

package com.drools.demo.point;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;


public class Test {

    /**
     * @param args
     * @throws IOException
     */
    public static void main(String[] args) throws IOException {
        PointRuleEngine pointRuleEngine = new PointRuleEngineImpl();
        boolean isStart = false;
        while(true){
            InputStream is = System.in;
            BufferedReader br = new BufferedReader(new InputStreamReader(is));
            String input = br.readLine();

            if (null != input && "s".equals(input)){
                System.out.println("初始化規則引擎...");
                pointRuleEngine.initEngine();
                isStart = true;
                System.out.println("初始化規則引擎結束.");
            } else if ("e".equals(input)){
                if (!isStart) {
                    System.out.println("需要輸入s啓動");
                } else {
                    final PointDomain pointDomain = new PointDomain();
                    pointDomain.setUserName("hello kity");
                    pointDomain.setBackMondy(100d);
                    pointDomain.setBuyMoney(500d);
                    pointDomain.setBackNums(1);
                    pointDomain.setBuyNums(5);
                    pointDomain.setBillThisMonth(5);
                    pointDomain.setBirthDay(true);
                    pointDomain.setPoint(0l);

                    pointRuleEngine.executeRuleEngine(pointDomain);

                    System.out.println("執行完畢BillThisMonth："+pointDomain.getBillThisMonth());
                    System.out.println("執行完畢BuyMoney："+pointDomain.getBuyMoney());
                    System.out.println("執行完畢BuyNums："+pointDomain.getBuyNums());

                    System.out.println("執行完畢規則引擎決定發送積分："+pointDomain.getPoint());
                }
            } else if ("r".equals(input)){
                System.out.println("刷新規則文件...");
                pointRuleEngine.refreshEnginRule();
                isStart = true;
                System.out.println("刷新規則文件結束.");
            } else if ("q".equals(input)) {
                System.exit(0);
            }
        }
    }

}