update

徐烜
1 parent 7605d4b2
Showing 5 changed files with 210 additions and 0 deletions
pom.xml
src/main/java/com/bsth/service/schedule/rules/Message.java
src/main/java/com/bsth/service/schedule/rules/MyDroolsConfiguration.java
src/main/resources/rules/HelloWorld.drl
src/test/java/com/bsth/service/schedule/rules/DroolsRulesTest.java
@@ -98,7 +98,39 @@
 			<artifactId>guava</artifactId>
 			<version>19.0</version>
 		</dependency>
+
+        <!-- drools 6依赖 -->
+        <dependency>
+            <groupId>org.kie</groupId>
+            <artifactId>kie-api</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.drools</groupId>
+            <artifactId>drools-compiler</artifactId>
+        </dependency>
+
+        <!-- springboot测试 -->
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-test</artifactId>
+            <scope>test</scope>
+        </dependency>
+
 	</dependencies>
+
+    <dependencyManagement>
+        <dependencies>
+            <!-- drools 6依赖 -->
+            <dependency>
+                <groupId>org.drools</groupId>
+                <artifactId>drools-bom</artifactId>
+                <type>pom</type>
+                <version>6.2.0.Final</version>
+                <scope>import</scope>
+            </dependency>
+        </dependencies>
+    </dependencyManagement>
+
 	<build>
 		<plugins>
 			<plugin>
+package com.bsth.service.schedule.rules;
+
+/**
+ * Created by xu on 16/6/15.
+ */
+public class Message {
+    public static final int HELLO = 0;
+    public static final int GOODBYE = 1;
+
+    private String message;
+    private int status;
+    public String getMessage() {
+        return message;
+    }
+    public void setMessage(String message) {
+        this.message = message;
+    }
+    public int getStatus() {
+        return status;
+    }
+    public void setStatus(int status) {
+        this.status = status;
+    }
+}
+package com.bsth.service.schedule.rules;
+
+import org.kie.api.KieBase;
+import org.kie.api.KieBaseConfiguration;
+import org.kie.api.KieServices;
+import org.kie.api.builder.*;
+import org.kie.api.builder.Message;
+import org.kie.api.builder.model.KieBaseModel;
+import org.kie.api.builder.model.KieModuleModel;
+import org.kie.api.builder.model.KieSessionModel;
+import org.kie.api.conf.EqualityBehaviorOption;
+import org.kie.api.conf.EventProcessingOption;
+import org.kie.api.runtime.KieContainer;
+import org.kie.api.runtime.conf.ClockTypeOption;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+/**
+ * Drools 6配置类。
+ */
+@Configuration
+public class MyDroolsConfiguration {
+    /**
+     * 返回一个kiebase知识库，直接冲文件系统读入drl规则文件，
+     * TODO：以后需要从数据库读入规则文件，并重新创建kbase知识库。
+     */
+    @Bean
+    public KieBase myKieBase() {
+        // Drools 6开始引入kie统一接口（jboss的jbpm工作流也使用kie接口了），整个定义方式和5差别很大
+        // 这里使用全api方式创建知识库对象，不使用xml的方式，提供最大的灵活性
+
+        // 1、创建kieservices
+        KieServices kieServices = KieServices.Factory.get();
+        // 2、创建KieModuleModel，默认是由kmodule.xml的方式创建，这里使用api方式闯将
+        KieModuleModel kieModuleModel = kieServices.newKieModuleModel();
+        // 2.1、创建KieBaseModel，类似kmodule.xml中的kbase标签
+        KieBaseModel kieBaseModel1 = kieModuleModel.newKieBaseModel("KBase1")
+                .setDefault(true)
+                .setEqualsBehavior(EqualityBehaviorOption.EQUALITY)
+                .setEventProcessingMode(EventProcessingOption.STREAM);
+        // 2.2、创建与kbase关联的KieSessionModel，类似kmodule.xml中的kbase内的ksession标签
+        kieBaseModel1.newKieSessionModel("KSession1")
+                .setDefault(true)
+                .setType(KieSessionModel.KieSessionType.STATEFUL)
+                .setClockType(ClockTypeOption.get("realtime"));
+
+        // 3、创建KieFileSystem，将模型xml，drl等写入，TODO：以后考虑从数据库中获取
+        KieFileSystem kfs = kieServices.newKieFileSystem();
+        // 3.1、写入KieBaseModel（内部包含了KieSessionModel的内容了，注意之前的KieSessionModel的创建方式）
+        kfs.writeKModuleXML(kieModuleModel.toXML());
+
+        // 3.2、写入drl（写法超多，有点混乱）
+        // 这里使用文件的形式写入，TODO：以后考虑从数据库中读drl写入
+        // 注意kfs写的时候如果指定path，强制为src/main/resources/加上文件名，还有就是文件名不要重复否则会覆盖的
+        kfs.write("src/main/resources/HelloWorld.drl", kieServices.getResources()
+                .newInputStreamResource(this.getClass().getResourceAsStream(
+                        "/rules/HelloWorld.drl"), "UTF-8"));
+        // TODO：还有其他drl....
+
+        // 4、创建KieBuilder，使用KieFileSystem构建
+        KieBuilder kieBuilder = kieServices.newKieBuilder(kfs).buildAll();
+        Results results = kieBuilder.getResults();
+        if (results.hasMessages(Message.Level.ERROR))
+            throw new IllegalStateException("构建drools6错误：" + results.getMessages());
+
+        // 5、获取KieContainer
+        // TODO：ReleaseId用处很大，以后再议
+        ReleaseId releaseId = kieServices.getRepository().getDefaultReleaseId();
+        KieContainer kieContainer = kieServices.newKieContainer(releaseId);
+
+        // 6、创建kbase
+        KieBaseConfiguration kieBaseConfiguration = kieServices.newKieBaseConfiguration();
+        KieBase kieBase = kieContainer.newKieBase("KBase1", kieBaseConfiguration);
+
+        return kieBase;
+    }
+}
+package com.bsth.service.schedule
+
+//list any import classes here.
+
+import com.bsth.service.schedule.rules.Message;
+
+//declare any global variables here
+
+global java.util.List list
+
+
+rule "Hello World"
+	dialect "mvel" 
+    when
+        $m : Message( status == Message.HELLO, $message : message )
+    then
+        System.out.println($message);
+        $m.status = com.bsth.service.schedule.rules.Message.GOODBYE;
+        $m.message = "Goodbye cruel world";
+        update($m);
+end
+
+rule "Good bye"
+	dialect "mvel" 
+    when
+    	$m : Message( status == Message.GOODBYE, $message : message )
+    then
+		System.out.println($message);
+		list.add("you come on!");
+end
+package com.bsth.service.schedule.rules;
+
+import com.bsth.Application;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.kie.api.KieBase;
+import org.kie.api.runtime.KieSession;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.SpringApplicationConfiguration;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+
+import java.util.ArrayList;
+import java.util.List;
+
+@RunWith(SpringJUnit4ClassRunner.class)
+@SpringApplicationConfiguration(classes = {Application.class})
+public class DroolsRulesTest {
+
+    @Autowired
+    private KieBase kieBase;
+
+    @Test
+    public void helloWorldDrlTest() throws Exception {
+        // 1、创建session，内部配置的是stateful
+        KieSession session = kieBase.newKieSession();
+
+        // 1.1 设置gloable对象，在drl中通过别名使用
+        List<String> gloableList = new ArrayList<String>();
+        session.setGlobal("list", gloableList);
+
+        // 1.2 可以设置一些监听器，再议
+
+        // 2、创建fact对象
+        Message message = new Message();
+        message.setMessage("Hello World");
+        message.setStatus(Message.HELLO);
+        session.insert(message);
+
+        // 3、执行rule
+        session.fireAllRules();
+
+        System.out.println(gloableList);
+
+        // 4、执行完毕销毁，有日志的也要关闭
+        session.dispose();
+    }
+}