use of org.drools.core.reteoo.ObjectTypeNode in project drools by kiegroup.
the class LinkingTest method testNotNodeMasksWithoutConstraints.
@Test
public void testNotNodeMasksWithoutConstraints() throws Exception {
String str = "";
str += "package org.kie \n";
str += "import " + A.class.getCanonicalName() + "\n";
str += "import " + B.class.getCanonicalName() + "\n";
str += "import " + C.class.getCanonicalName() + "\n";
str += "import " + D.class.getCanonicalName() + "\n";
str += "import " + E.class.getCanonicalName() + "\n";
str += "import " + F.class.getCanonicalName() + "\n";
str += "import " + G.class.getCanonicalName() + "\n";
str += "global java.util.List list \n";
str += "rule rule1 when \n";
str += " A() \n";
str += " not( B( ) ) \n";
str += " C() \n";
str += "then \n";
str += "end \n";
KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
kbuilder.add(ResourceFactory.newByteArrayResource(str.getBytes()), ResourceType.DRL);
assertFalse(kbuilder.getErrors().toString(), kbuilder.hasErrors());
InternalKnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
kbase.addPackages(kbuilder.getKnowledgePackages());
ObjectTypeNode node = getObjectTypeNode(kbase, A.class);
InternalWorkingMemory wm = ((StatefulKnowledgeSessionImpl) kbase.newKieSession());
LeftInputAdapterNode liaNode = (LeftInputAdapterNode) node.getObjectSinkPropagator().getSinks()[0];
assertEquals(1, liaNode.getSinkPropagator().size());
wm.insert(new A());
wm.flushPropagations();
NotNode notNode = (NotNode) liaNode.getSinkPropagator().getSinks()[0];
JoinNode cNode = (JoinNode) notNode.getSinkPropagator().getSinks()[0];
RuleTerminalNode rtn = (RuleTerminalNode) cNode.getSinkPropagator().getSinks()[0];
PathMemory pmem = (PathMemory) wm.getNodeMemory(rtn);
assertEquals(1, pmem.getSegmentMemories().length);
assertEquals(1, pmem.getAllLinkedMaskTest());
SegmentMemory sm = pmem.getSegmentMemories()[0];
assertEquals(7, sm.getAllLinkedMaskTest());
assertEquals(3, sm.getLinkedNodeMask());
assertFalse(sm.isSegmentLinked());
assertFalse(pmem.isRuleLinked());
wm.insert(new C());
wm.flushPropagations();
assertEquals(7, sm.getLinkedNodeMask());
assertTrue(sm.isSegmentLinked());
assertTrue(pmem.isRuleLinked());
}
use of org.drools.core.reteoo.ObjectTypeNode in project drools by kiegroup.
the class LinkingTest method testNotNodeMasksWithConstraints.
@Test
public void testNotNodeMasksWithConstraints() throws Exception {
String str = "";
str += "package org.kie \n";
str += "import " + A.class.getCanonicalName() + "\n";
str += "import " + B.class.getCanonicalName() + "\n";
str += "import " + C.class.getCanonicalName() + "\n";
str += "import " + D.class.getCanonicalName() + "\n";
str += "import " + E.class.getCanonicalName() + "\n";
str += "import " + F.class.getCanonicalName() + "\n";
str += "import " + G.class.getCanonicalName() + "\n";
str += "global java.util.List list \n";
str += "rule rule1 when \n";
str += " $a : A() \n";
str += " not( B( value == $a.value ) ) \n";
str += " C() \n";
str += "then \n";
str += "end \n";
KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
kbuilder.add(ResourceFactory.newByteArrayResource(str.getBytes()), ResourceType.DRL);
assertFalse(kbuilder.getErrors().toString(), kbuilder.hasErrors());
InternalKnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
kbase.addPackages(kbuilder.getKnowledgePackages());
ObjectTypeNode node = getObjectTypeNode(kbase, A.class);
InternalWorkingMemory wm = ((StatefulKnowledgeSessionImpl) kbase.newKieSession());
LeftInputAdapterNode liaNode = (LeftInputAdapterNode) node.getObjectSinkPropagator().getSinks()[0];
assertEquals(1, liaNode.getSinkPropagator().size());
wm.insert(new A());
wm.flushPropagations();
NotNode notNode = (NotNode) liaNode.getSinkPropagator().getSinks()[0];
JoinNode cNode = (JoinNode) notNode.getSinkPropagator().getSinks()[0];
RuleTerminalNode rtn = (RuleTerminalNode) cNode.getSinkPropagator().getSinks()[0];
PathMemory pmem = (PathMemory) wm.getNodeMemory(rtn);
assertEquals(1, pmem.getSegmentMemories().length);
assertEquals(1, pmem.getAllLinkedMaskTest());
SegmentMemory sm = pmem.getSegmentMemories()[0];
assertEquals(5, sm.getAllLinkedMaskTest());
assertEquals(3, sm.getLinkedNodeMask());
assertFalse(sm.isSegmentLinked());
assertFalse(pmem.isRuleLinked());
wm.insert(new C());
wm.flushPropagations();
// only 5 is needed to link, the 'not' turns on but it has no unfleunce either way
assertEquals(7, sm.getLinkedNodeMask());
assertTrue(sm.isSegmentLinked());
assertTrue(pmem.isRuleLinked());
}
use of org.drools.core.reteoo.ObjectTypeNode in project drools by kiegroup.
the class LinkingTest method testNonReactiveSubNetworkOwnSegmentMasks.
@Test
public void testNonReactiveSubNetworkOwnSegmentMasks() throws Exception {
String str = "";
str += "package org.kie \n";
str += "import " + A.class.getCanonicalName() + "\n";
str += "import " + B.class.getCanonicalName() + "\n";
str += "import " + C.class.getCanonicalName() + "\n";
str += "import " + D.class.getCanonicalName() + "\n";
str += "import " + E.class.getCanonicalName() + "\n";
str += "import " + F.class.getCanonicalName() + "\n";
str += "import " + G.class.getCanonicalName() + "\n";
str += "global java.util.List list \n";
str += "rule rule1 when \n";
str += " A() \n";
str += " exists( B() and C() ) \n";
str += " exists( eval(1==1) ) \n";
str += " D() \n";
str += "then \n";
str += "end \n";
str += "rule rule2 when \n";
str += " A() \n";
str += " exists( B() and C() ) \n";
str += " exists( eval(1==1) ) \n";
str += " E() \n";
str += "then \n";
str += "end \n";
KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
kbuilder.add(ResourceFactory.newByteArrayResource(str.getBytes()), ResourceType.DRL);
assertFalse(kbuilder.getErrors().toString(), kbuilder.hasErrors());
InternalKnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
kbase.addPackages(kbuilder.getKnowledgePackages());
ObjectTypeNode node = getObjectTypeNode(kbase, A.class);
InternalWorkingMemory wm = ((StatefulKnowledgeSessionImpl) kbase.newKieSession());
LeftInputAdapterNode liaNode = (LeftInputAdapterNode) node.getObjectSinkPropagator().getSinks()[0];
assertEquals(2, liaNode.getSinkPropagator().size());
JoinNode bNode = (JoinNode) liaNode.getSinkPropagator().getSinks()[0];
JoinNode cNode = (JoinNode) bNode.getSinkPropagator().getSinks()[0];
ExistsNode exists1n = (ExistsNode) liaNode.getSinkPropagator().getSinks()[1];
EvalConditionNode evalNode = (EvalConditionNode) exists1n.getSinkPropagator().getSinks()[0];
ExistsNode exists2n = (ExistsNode) exists1n.getSinkPropagator().getSinks()[1];
JoinNode dNode = (JoinNode) exists2n.getSinkPropagator().getSinks()[0];
RuleTerminalNode rtn = (RuleTerminalNode) dNode.getSinkPropagator().getSinks()[0];
wm.insert(new A());
PathMemory pmem = (PathMemory) wm.getNodeMemory(rtn);
assertEquals(4, pmem.getSegmentMemories().length);
// the exists eval segment does not need to be linked in
assertEquals(11, pmem.getAllLinkedMaskTest());
RightInputAdapterNode.RiaNodeMemory riaMem = (RightInputAdapterNode.RiaNodeMemory) wm.getNodeMemory((MemoryFactory) exists1n.getRightInput());
// second segment must be linked in
assertEquals(2, riaMem.getRiaPathMemory().getAllLinkedMaskTest());
wm.insert(new B());
wm.insert(new C());
assertEquals(2, riaMem.getRiaPathMemory().getSegmentMemories().length);
riaMem = (RightInputAdapterNode.RiaNodeMemory) wm.getNodeMemory((MemoryFactory) exists2n.getRightInput());
// no segments to be linked in
assertEquals(0, riaMem.getRiaPathMemory().getAllLinkedMaskTest());
}
use of org.drools.core.reteoo.ObjectTypeNode in project drools by kiegroup.
the class LinkingTest method testNotNodeUnlinksWithNoConstriants.
@Test
public void testNotNodeUnlinksWithNoConstriants() {
String str = "";
str += "package org.kie \n";
str += "import " + A.class.getCanonicalName() + "\n";
str += "import " + B.class.getCanonicalName() + "\n";
str += "import " + C.class.getCanonicalName() + "\n";
str += "import " + D.class.getCanonicalName() + "\n";
str += "import " + E.class.getCanonicalName() + "\n";
str += "import " + F.class.getCanonicalName() + "\n";
str += "import " + G.class.getCanonicalName() + "\n";
str += "global java.util.List list \n";
str += "rule rule1 when \n";
str += " $a : A() \n";
str += " not B() \n";
str += " $c : C() \n";
str += "then \n";
str += " list.add( 'x' ); \n";
str += "end \n";
KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
kbuilder.add(ResourceFactory.newByteArrayResource(str.getBytes()), ResourceType.DRL);
assertFalse(kbuilder.getErrors().toString(), kbuilder.hasErrors());
InternalKnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
kbase.addPackages(kbuilder.getKnowledgePackages());
ObjectTypeNode aotn = getObjectTypeNode(kbase, A.class);
InternalWorkingMemory wm = ((StatefulKnowledgeSessionImpl) kbase.newKieSession());
List list = new ArrayList();
wm.setGlobal("list", list);
LeftInputAdapterNode aNode = (LeftInputAdapterNode) aotn.getObjectSinkPropagator().getSinks()[0];
NotNode bNode = (NotNode) aNode.getSinkPropagator().getSinks()[0];
JoinNode cNode = (JoinNode) bNode.getSinkPropagator().getSinks()[0];
SegmentUtilities.createSegmentMemory(cNode, wm);
LiaNodeMemory amem = (LiaNodeMemory) wm.getNodeMemory(aNode);
// Only NotNode is linked in
assertEquals(2, amem.getSegmentMemory().getLinkedNodeMask());
FactHandle fha = wm.insert(new A());
FactHandle fhb = wm.insert(new B());
FactHandle fhc = wm.insert(new C());
wm.fireAllRules();
assertEquals(0, list.size());
// NotNode unlinks, which is allowed because it has no variable constraints
assertEquals(5, amem.getSegmentMemory().getLinkedNodeMask());
// NotNode links back in again, which is allowed because it has no variable constraints
wm.retract(fhb);
wm.flushPropagations();
assertEquals(7, amem.getSegmentMemory().getLinkedNodeMask());
wm.fireAllRules();
assertEquals(1, list.size());
// Now try with lots of facthandles on NotNode
list.clear();
List<FactHandle> handles = new ArrayList<FactHandle>();
for (int i = 0; i < 5; i++) {
handles.add(wm.insert(new B()));
}
wm.fireAllRules();
assertEquals(0, list.size());
assertEquals(5, amem.getSegmentMemory().getLinkedNodeMask());
for (FactHandle fh : handles) {
wm.retract(fh);
}
wm.flushPropagations();
assertEquals(7, amem.getSegmentMemory().getLinkedNodeMask());
wm.fireAllRules();
assertEquals(1, list.size());
}
use of org.drools.core.reteoo.ObjectTypeNode in project drools by kiegroup.
the class LinkingTest method testSubnetwork.
@Test
public void testSubnetwork() throws Exception {
String str = "";
str += "package org.kie \n";
str += "import " + A.class.getCanonicalName() + "\n";
str += "import " + B.class.getCanonicalName() + "\n";
str += "import " + C.class.getCanonicalName() + "\n";
str += "import " + D.class.getCanonicalName() + "\n";
str += "import " + E.class.getCanonicalName() + "\n";
str += "import " + F.class.getCanonicalName() + "\n";
str += "import " + G.class.getCanonicalName() + "\n";
str += "global java.util.List list \n";
str += "rule rule1 when \n";
str += " $a : A() \n";
str += " exists ( B() and C() ) \n";
str += " $e : D() \n";
str += "then \n";
str += " list.add( 'x' ); \n";
str += "end \n";
KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
kbuilder.add(ResourceFactory.newByteArrayResource(str.getBytes()), ResourceType.DRL);
assertFalse(kbuilder.getErrors().toString(), kbuilder.hasErrors());
InternalKnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
kbase.addPackages(kbuilder.getKnowledgePackages());
ObjectTypeNode aotn = getObjectTypeNode(kbase, A.class);
ObjectTypeNode botn = getObjectTypeNode(kbase, A.class);
ObjectTypeNode cotn = getObjectTypeNode(kbase, A.class);
InternalWorkingMemory wm = ((StatefulKnowledgeSessionImpl) kbase.newKieSession());
List list = new ArrayList();
wm.setGlobal("list", list);
wm.insert(new A());
wm.insert(new B());
for (int i = 0; i < 28; i++) {
wm.insert(new C());
}
wm.insert(new D());
wm.flushPropagations();
InternalAgenda agenda = (InternalAgenda) wm.getAgenda();
InternalAgendaGroup group = (InternalAgendaGroup) agenda.getNextFocus();
AgendaItem item = (AgendaItem) group.remove();
RuleExecutor ruleExecutor = ((RuleAgendaItem) item).getRuleExecutor();
int count = ruleExecutor.evaluateNetworkAndFire(wm, null, 0, -1);
// assertEquals(3, count );
wm.fireAllRules();
assertEquals(1, list.size());
wm.fireAllRules();
// check it doesn't double fire
assertEquals(1, list.size());
}
Aggregations