Examples with StatefulKnowledgeSessionImpl org.drools.impl.StatefulKnowledgeSessionImpl used on opensource projects

Example 1 with StatefulKnowledgeSessionImpl

use of org.drools.impl.StatefulKnowledgeSessionImpl in project jBPM5-Developer-Guide by Salaboy.

the class AdvancedProcessAndRulesIntegrationTest method processEventsTest.

@Test
public void processEventsTest() throws InterruptedException {
    KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
    kbuilder.add(new ClassPathResource("evolution/mapping.drl"), ResourceType.DRL);
    kbuilder.add(new ClassPathResource("evolution/scoring_processVariables_wider.drl"), ResourceType.DRL);
    kbuilder.add(new ClassPathResource("evolution/process-events-decision.bpmn"), ResourceType.BPMN2);
    if (kbuilder.hasErrors()) {
        for (KnowledgeBuilderError error : kbuilder.getErrors()) {
            System.out.println(">>> Error:" + error.getMessage());
        }
        fail(">>> Knowledge couldn't be parsed! ");
    }
    KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
    kbase.addKnowledgePackages(kbuilder.getKnowledgePackages());
    final StatefulKnowledgeSession ksession = kbase.newStatefulKnowledgeSession();
    ksession.getWorkItemManager().registerWorkItemHandler("Human Task", new WorkItemHandler() {

        public void executeWorkItem(WorkItem wi, WorkItemManager wim) {
            System.out.println(" >>> Completing Task! -> " + wi.getName() + " - id: " + wi.getId());
            wim.completeWorkItem(wi.getId(), null);
        }

        public void abortWorkItem(WorkItem wi, WorkItemManager wim) {
            throw new UnsupportedOperationException("Not supported yet.");
        }
    });
    KnowledgeRuntimeLoggerFactory.newConsoleLogger(ksession);
    ksession.addEventListener(new DefaultAgendaEventListener() {

        @Override
        public void activationCreated(ActivationCreatedEvent event) {
            ((StatefulKnowledgeSession) event.getKnowledgeRuntime()).fireAllRules();
        }

        @Override
        public void afterRuleFlowGroupActivated(RuleFlowGroupActivatedEvent event) {
            ((StatefulKnowledgeSession) event.getKnowledgeRuntime()).fireAllRules();
        }
    });
    Person person = new Person("Salaboy", 28);
    RatesToday ratesToday = new RatesToday(1, 100);
    Map<String, Object> params = new HashMap<String, Object>();
    params.put("person", person);
    params.put("ratesToday", ratesToday);
    ProcessInstance processInstance = ksession.createProcessInstance("com.salaboy.process.SimpleDecision", params);
    System.out.println("Variables: " + ((WorkflowProcessInstanceImpl) processInstance).getVariables());
    assertEquals(processInstance.getState(), ProcessInstance.STATE_PENDING);
    final FactHandle processHandle = ksession.insert(processInstance);
    ((StatefulKnowledgeSessionImpl) ksession).addEventListener(new DefaultProcessEventListener() {

        @Override
        public void beforeProcessStarted(ProcessStartedEvent event) {
            ((StatefulKnowledgeSession) event.getKnowledgeRuntime()).fireAllRules();
        }

        @Override
        public void afterProcessStarted(ProcessStartedEvent event) {
            ((StatefulKnowledgeSession) event.getKnowledgeRuntime()).fireAllRules();
        }

        @Override
        public void afterProcessCompleted(ProcessCompletedEvent event) {
            ((StatefulKnowledgeSession) event.getKnowledgeRuntime()).retract(processHandle);
        }
    });
    ksession.startProcessInstance(processInstance.getId());
    Thread.sleep(1000);
    // // If you want to query the process variables while the process Instance is running you can do:
    // QueryResults queryResults = ksession.getQueryResults("allProcessVariables", new Object[]{});
    // Iterator<QueryResultsRow> iterator = queryResults.iterator();
    // 
    // QueryResultsRow ratesRow = iterator.next();
    // assertEquals(ratesToday, ((ProcessVariable) ratesRow.get("$pv")).getValue());
    // 
    // QueryResultsRow personRow = iterator.next();
    // assertEquals(person, ((ProcessVariable) personRow.get("$pv")).getValue());
    assertEquals(processInstance.getState(), ProcessInstance.STATE_COMPLETED);
    QueryResults queryResults = ksession.getQueryResults("allProcessVariables", new Object[] {});
    // The Process Variables are automatically retracted when the Process Instance is Completed
    assertEquals(0, queryResults.size());
}

Example 2 with StatefulKnowledgeSessionImpl

use of org.drools.impl.StatefulKnowledgeSessionImpl in project jBPM5-Developer-Guide by Salaboy.

the class ProcessAndEventMultiThreadIntegrationTest method processEventsWithListenerTest.

@Test
public void processEventsWithListenerTest() throws InterruptedException {
    // Let's create a kbuilder, kbase and ksession for running the process
    final StatefulKnowledgeSession processKsession = createProcessSession();
    // Let's create a kbuilder, kbase and ksession for analyzing the process events
    final StatefulKnowledgeSession eventsKsession = createEventsSession();
    final TaskSpeed taskSpeed = new TaskSpeed(1000L);
    eventsKsession.setGlobal("taskSpeed", taskSpeed);
    processKsession.getWorkItemManager().registerWorkItemHandler("Human Task", new WorkItemHandler() {

        public void executeWorkItem(WorkItem wi, WorkItemManager wim) {
            try {
                System.out.println(" >>> Working on Task! it will take: " + taskSpeed.getAmount() / 1000 + " seconds.");
                Thread.sleep(taskSpeed.getAmount());
            } catch (InterruptedException ex) {
                Logger.getLogger(ProcessAndEventMultiThreadIntegrationTest.class.getName()).log(Level.SEVERE, null, ex);
            }
            System.out.println(" >>> Completing Task! -> " + wi.getName() + " - id: " + wi.getId());
            wim.completeWorkItem(wi.getId(), null);
        }

        public void abortWorkItem(WorkItem wi, WorkItemManager wim) {
        // do nothing
        }
    });
    ((StatefulKnowledgeSessionImpl) processKsession).addEventListener(new DefaultProcessEventListener() {

        @Override
        public void beforeProcessStarted(ProcessStartedEvent event) {
            System.out.println(" >>> Before Process Started: " + event.getProcessInstance().getProcessName());
            eventsKsession.insert(event);
        }

        @Override
        public void afterProcessCompleted(ProcessCompletedEvent event) {
            System.out.println(" >>> After Process Completed: " + event.getProcessInstance().getProcessName());
            eventsKsession.insert(event);
        }

        @Override
        public void beforeNodeLeft(org.drools.event.process.ProcessNodeLeftEvent event) {
            System.out.println(" >>> Before Node Left: " + event.getNodeInstance().getNodeName());
            eventsKsession.insert(event);
        }
    });
    for (int i = 0; i < 5; i++) {
        for (int j = 0; j < 5; j++) {
            final int id1 = i;
            final int id2 = j;
            new Thread(new Runnable() {

                public void run() {
                    Person person = new Person("Salaboy-" + id1 + "_" + id2, id2);
                    Map<String, Object> params = new HashMap<String, Object>();
                    params.put("person", person);
                    ProcessInstance processInstance = processKsession.createProcessInstance("com.salaboy.process.SimpleProcess", params);
                    assertEquals(processInstance.getState(), ProcessInstance.STATE_PENDING);
                    FactHandle processtHandle = processKsession.insert(processInstance);
                    System.out.println("Starting Process Instance: " + processInstance.getId());
                    processKsession.startProcessInstance(processInstance.getId());
                    assertEquals(processInstance.getState(), ProcessInstance.STATE_COMPLETED);
                    processKsession.retract(processtHandle);
                }
            }).start();
        }
        Thread.sleep(3000);
    }
}