package com.sample;

import java.io.FileReader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

import javax.rules.ConfigurationException;
import javax.rules.RuleRuntime;
import javax.rules.RuleServiceProvider;
import javax.rules.RuleServiceProviderManager;
import javax.rules.StatelessRuleSession;
import javax.rules.admin.LocalRuleExecutionSetProvider;
import javax.rules.admin.RuleAdministrator;
import javax.rules.admin.RuleExecutionSet;

import org.drools.jsr94.rules.RuleServiceProviderImpl;

public class JSR94Sample {
    private static RuleServiceProvider ruleProvider;
    
    private static void initProvider(){
        String uri = RuleServiceProviderImpl.RULE_SERVICE_PROVIDER;
        Class providerClass = RuleServiceProviderImpl.class;
        
        try{
            //注冊ruleProvider
            RuleServiceProviderManager.registerRuleServiceProvider(uri, providerClass);
            
            //從RuleServiceProviderManager獲取ruleProvider
            ruleProvider = RuleServiceProviderManager.getRuleServiceProvider(uri);
        }catch(ConfigurationException e){
            e.printStackTrace();
        }
    }
    
    private static void adminSample(){
        

        try{
            //獲取RuleAdministrator實(shí)例
            RuleAdministrator admin = ruleProvider.getRuleAdministrator();
            
            //獲取RuleExectuionSetProvider
            HashMap properties = new HashMap();   
            properties.put("name", "My Rules");   
            properties.put("description", "A trivial rulebase");
            
            LocalRuleExecutionSetProvider ruleExecutionSetProvider = admin.getLocalRuleExecutionSetProvider(properties);
            
            //創(chuàng)建RuleExecutionSet
            FileReader reader = new FileReader("bin/sample.drl");
            RuleExecutionSet reSet = ruleExecutionSetProvider.createRuleExecutionSet(reader, properties);
            
            //注冊RuleExecutionSet
            admin.registerRuleExecutionSet("mysample",reSet,properties);
        }catch(Exception e){
            e.printStackTrace();
        }
        
    }
    
    
    private static void runtimeSampe(){
        try{
            //獲取RuleRuntime, 創(chuàng)建會(huì)話
            RuleRuntime runtime = ruleProvider.getRuleRuntime();
            StatelessRuleSession ruleSession = (StatelessRuleSession)runtime.createRuleSession("mysample",null,RuleRuntime.STATELESS_SESSION_TYPE);
            
            //初始化輸入數(shù)據(jù)
            Message message1 = new Message();
            message1.setMessage("Hello World");
            message1.setStatus(Message.HELLO);
            
            Message message2 = new Message();
            message2.setMessage("Goodbye World");
            message2.setStatus(Message.GOODBYE);
            
            
            List inputs = new ArrayList();
            inputs.add(message1);
            inputs.add(message2);
            
            //執(zhí)行規(guī)則
            List<Message> results = ruleSession.executeRules(inputs);
            for(int i=0;i<results.size();i++){
                Message msg = results.get(i);
                System.out.println(msg.message);
            }
            
            
            //釋放會(huì)話資源
            ruleSession.release();
            
        }catch(Exception e){
            e.printStackTrace();
        }
    }
    
    
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        initProvider();
        adminSample();
        runtimeSampe();
    }
    
    
    public static class Message {

        public static final int HELLO = 0;
        public static final int GOODBYE = 1;

        private String message;

        private int status;

        public String getMessage() {
            return this.message;
        }

        public void setMessage(String message) {
            this.message = message;
        }

        public int getStatus() {
            return this.status;
        }

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

    }

}