Examples with InternalWorkingMemory org.drools.core.common.InternalWorkingMemory used on opensource projects

Example 36 with InternalWorkingMemory

use of org.drools.core.common.InternalWorkingMemory in project drools by kiegroup.

the class ParallelEvaluationTest method getMultipleParallelKieSessionsWithUpdatesCallable.

private Callable<Void> getMultipleParallelKieSessionsWithUpdatesCallable(KieBase kBase) {
    return new Callable<Void>() {

        @Override
        public Void call() {
            KieSession ksession = kBase.newKieSession();
            assertThat(((InternalWorkingMemory) ksession).getAgenda().isParallelAgenda()).as("Parallel agenda has to be enabled").isTrue();
            List<Integer> list = new DebugList<Integer>();
            ksession.setGlobal("list", list);
            FactHandle[] fhs = new FactHandle[10];
            fhs = insertFacts(ksession, 10);
            ksession.fireAllRules();
            assertThat(list.size()).isEqualTo(10);
            list.clear();
            for (int i = 0; i < 10; i++) {
                ksession.update(fhs[i], i);
            }
            ksession.fireAllRules();
            assertThat(list.size()).isEqualTo(10);
            return null;
        }
    };
}

Example 37 with InternalWorkingMemory

use of org.drools.core.common.InternalWorkingMemory in project drools by kiegroup.

the class ParallelEvaluationTest method testDisableParallelismOnSinglePartition.

@Test(timeout = 10000L)
public void testDisableParallelismOnSinglePartition() {
    String drl = "rule R1 when\n" + "    $i : Integer( this == 4 )" + "    String( length > $i )\n" + "then end \n" + "rule R2 when\n" + "    $i : Integer( this == 4 )" + "    String( length == $i )\n" + "then end \n" + "rule R3 when\n" + "    $i : Integer( this == 4 )" + "    String( length < $i )\n" + "then end \n";
    KieSession ksession = new KieHelper().addContent(drl, ResourceType.DRL).build(MultithreadEvaluationOption.YES).newKieSession();
    InternalWorkingMemory session = (InternalWorkingMemory) ksession;
    // since there is only one partition the multithread evaluation should be disabled and run with the DefaultAgenda
    assertFalse(((InternalWorkingMemory) ksession).getAgenda().isParallelAgenda());
}

Example 38 with InternalWorkingMemory

use of org.drools.core.common.InternalWorkingMemory in project drools by kiegroup.

the class ParallelEvaluationTest method testImmediateEventsExpiration.

@Test(timeout = 10000L)
public void testImmediateEventsExpiration() {
    StringBuilder sb = new StringBuilder(400);
    sb.append("global java.util.List list;\n");
    sb.append("import " + MyEvent.class.getCanonicalName() + ";\n");
    sb.append("declare MyEvent @role( event ) @expires( 1ms ) @timestamp( timestamp ) end\n");
    for (int i = 0; i < 10; i++) {
        sb.append(getRuleWithEvent(i));
    }
    KieSessionConfiguration sessionConfig = KnowledgeBaseFactory.newKnowledgeSessionConfiguration();
    KieSession ksession = new KieHelper().addContent(sb.toString(), ResourceType.DRL).build(EventProcessingOption.STREAM, MultithreadEvaluationOption.YES).newKieSession(sessionConfig, null);
    assertTrue(((InternalWorkingMemory) ksession).getAgenda().isParallelAgenda());
    List<Integer> list = new DebugList<Integer>();
    ksession.setGlobal("list", list);
    for (int i = 0; i < 10; i++) {
        ksession.insert(new MyEvent(i, i * 2L));
    }
    ksession.fireAllRules();
    assertEquals(10, list.size());
}

Example 39 with InternalWorkingMemory

use of org.drools.core.common.InternalWorkingMemory in project drools by kiegroup.

the class ParallelEvaluationTest method testFireUntilHalt.

@Test(timeout = 10000L)
public void testFireUntilHalt() {
    StringBuilder sb = new StringBuilder(400);
    sb.append("global java.util.List list;\n");
    for (int i = 0; i < 10; i++) {
        sb.append(getRule(i, ""));
    }
    KieSession ksession = new KieHelper().addContent(sb.toString(), ResourceType.DRL).build(MultithreadEvaluationOption.YES).newKieSession();
    assertTrue(((InternalWorkingMemory) ksession).getAgenda().isParallelAgenda());
    CountDownLatch done = new CountDownLatch(1);
    DebugList<Integer> list = new DebugList<Integer>();
    list.onItemAdded = (l -> {
        if (l.size() == 10) {
            ksession.halt();
            done.countDown();
        }
    });
    ksession.setGlobal("list", list);
    new Thread(() -> ksession.fireUntilHalt()).start();
    for (int i = 0; i < 10; i++) {
        ksession.insert(i);
        ksession.insert("" + i);
    }
    try {
        done.await();
    } catch (InterruptedException e) {
        throw new RuntimeException(e);
    }
    assertEquals(10, list.size());
}

Example 40 with InternalWorkingMemory

use of org.drools.core.common.InternalWorkingMemory in project drools by kiegroup.

the class ParallelEvaluationTest method testWithDeletes.

@Test(timeout = 10000L)
public void testWithDeletes() {
    StringBuilder sb = new StringBuilder(400);
    sb.append("global java.util.List list;\n");
    for (int i = 1; i < 11; i++) {
        sb.append(getRule(i, "delete( $i );\n"));
    }
    for (int i = 1; i < 11; i++) {
        sb.append(getNotRule(i));
    }
    KieSession ksession = new KieHelper().addContent(sb.toString(), ResourceType.DRL).build(MultithreadEvaluationOption.YES).newKieSession();
    assertTrue(((InternalWorkingMemory) ksession).getAgenda().isParallelAgenda());
    List<Integer> list = new DebugList<Integer>();
    ksession.setGlobal("list", list);
    for (int i = 1; i < 11; i++) {
        ksession.insert(i);
        ksession.insert("" + i);
    }
    ksession.fireAllRules();
    assertEquals(20, list.size());
}