use of org.drools.core.rule.Declaration in project drools by kiegroup.
the class KiePackagesBuilder method buildQueryPattern.
private RuleConditionElement buildQueryPattern(RuleContext ctx, QueryCallPattern queryPattern) {
Pattern pattern = new Pattern(ctx.getNextPatternIndex(), 0, ClassObjectType.ObjectArray_ObjectType, null);
InternalReadAccessor arrayReader = new SelfReferenceClassFieldReader(Object[].class);
QueryArgument[] arguments = new QueryArgument[queryPattern.getArguments().length];
List<Integer> varIndexList = new ArrayList<>();
List<Declaration> requiredDeclarations = new ArrayList<>();
for (int i = 0; i < queryPattern.getArguments().length; i++) {
Argument arg = queryPattern.getArguments()[i];
if (arg instanceof Variable) {
Variable var = ((Variable) arg);
Declaration decl = ctx.getDeclaration(var);
if (decl != null) {
requiredDeclarations.add(decl);
arguments[i] = new QueryArgument.Declr(decl);
} else {
ArrayElementReader reader = new ArrayElementReader(arrayReader, i, arg.getType());
pattern.addDeclaration(var.getName()).setReadAccessor(reader);
arguments[i] = QueryArgument.VAR;
varIndexList.add(i);
}
} else if (arg instanceof Value) {
arguments[i] = new QueryArgument.Literal(((Value) arg).getValue());
} else {
throw new UnsupportedOperationException();
}
}
return new QueryElement(pattern, queryPattern.getQuery().getName(), arguments, varIndexList.stream().mapToInt(i -> i).toArray(), requiredDeclarations.toArray(new Declaration[requiredDeclarations.size()]), queryPattern.isOpen(), // TODO: query.isAbductive() );
false);
}
use of org.drools.core.rule.Declaration in project drools by kiegroup.
the class RightTupleIndexHashTableIteratorTest method getConstraint.
protected BetaNodeFieldConstraint getConstraint(String identifier, Operator operator, String fieldName, Class clazz) {
ClassFieldAccessorStore store = new ClassFieldAccessorStore();
store.setClassFieldAccessorCache(new ClassFieldAccessorCache(Thread.currentThread().getContextClassLoader()));
store.setEagerWire(true);
InternalReadAccessor extractor = store.getReader(clazz, fieldName);
Declaration declaration = new Declaration(identifier, extractor, new Pattern(0, new ClassObjectType(clazz)));
String expression = fieldName + " " + operator.getOperatorString() + " " + declaration.getIdentifier();
return new MvelConstraintTestUtil(expression, declaration, extractor);
}
use of org.drools.core.rule.Declaration in project drools by kiegroup.
the class RightTupleIndexHashTableTest method testResize.
@Test
public void testResize() throws Exception {
final InternalReadAccessor extractor = store.getReader(Cheese.class, "type");
final Pattern pattern = new Pattern(0, new ClassObjectType(Cheese.class));
final Declaration declaration = new Declaration("typeOfCheese", extractor, pattern);
final FieldIndex fieldIndex = new FieldIndex(extractor, declaration, MvelConstraint.INDEX_EVALUATOR);
final TupleIndexHashTable map = new TupleIndexHashTable(16, 0.75f, new FieldIndex[] { fieldIndex }, false);
assertEquals(0, map.size());
final Cheese stilton1 = new Cheese("stilton", 35);
map.add(newRightTuple(1, stilton1));
final Cheese stilton2 = new Cheese("stilton", 81);
map.add(newRightTuple(2, stilton2));
final Cheese cheddar1 = new Cheese("cheddar", 35);
map.add(newRightTuple(3, cheddar1));
final Cheese cheddar2 = new Cheese("cheddar", 38);
map.add(newRightTuple(4, cheddar2));
final Cheese brie = new Cheese("brie", 293);
map.add(newRightTuple(5, brie));
final Cheese mozerella = new Cheese("mozerella", 15);
map.add(newRightTuple(6, mozerella));
final Cheese dolcelatte = new Cheese("dolcelatte", 284);
map.add(newRightTuple(7, dolcelatte));
final Cheese camembert1 = new Cheese("camembert", 924);
map.add(newRightTuple(8, camembert1));
final Cheese camembert2 = new Cheese("camembert", 765);
map.add(newRightTuple(9, camembert2));
final Cheese redLeicestor = new Cheese("red leicestor", 23);
map.add(newRightTuple(10, redLeicestor));
final Cheese wensleydale = new Cheese("wensleydale", 20);
map.add(newRightTuple(11, wensleydale));
final Cheese edam = new Cheese("edam", 12);
map.add(newRightTuple(12, edam));
final Cheese goude1 = new Cheese("goude", 93);
map.add(newRightTuple(13, goude1));
final Cheese goude2 = new Cheese("goude", 88);
map.add(newRightTuple(14, goude2));
final Cheese gruyere = new Cheese("gruyere", 82);
map.add(newRightTuple(15, gruyere));
final Cheese emmental = new Cheese("emmental", 98);
map.add(newRightTuple(16, emmental));
// At this point we have 16 facts but only 12 different types of cheeses
// so no table resize and thus its size is 16
assertEquals(16, map.size());
Entry[] table = map.getTable();
assertEquals(16, table.length);
final Cheese feta = new Cheese("feta", 48);
map.add(newRightTuple(2, feta));
// This adds our 13th type of cheese. The map is set with an initial capacity of 16 and
// a threshold of 75%, that after 12 it should resize the map to 32.
assertEquals(17, map.size());
table = map.getTable();
assertEquals(32, table.length);
final Cheese haloumi = new Cheese("haloumi", 48);
map.add(newRightTuple(2, haloumi));
final Cheese chevre = new Cheese("chevre", 48);
map.add(newRightTuple(2, chevre));
}
use of org.drools.core.rule.Declaration in project drools by kiegroup.
the class RightTupleIndexHashTableTest method testRemove.
@Test
public void testRemove() throws Exception {
final InternalReadAccessor extractor = store.getReader(Cheese.class, "type");
final Pattern pattern = new Pattern(0, new ClassObjectType(Cheese.class));
final Declaration declaration = new Declaration("typeOfCheese", extractor, pattern);
final FieldIndex fieldIndex = new FieldIndex(extractor, declaration, MvelConstraint.INDEX_EVALUATOR);
final TupleIndexHashTable map = new TupleIndexHashTable(new FieldIndex[] { fieldIndex }, false);
assertEquals(0, map.size());
final Cheese stilton1 = new Cheese("stilton", 35);
final InternalFactHandle stiltonHandle1 = new DefaultFactHandle(1, stilton1);
RightTuple stiltonRightTuple1 = new RightTupleImpl(stiltonHandle1, null);
map.add(stiltonRightTuple1);
final Cheese cheddar1 = new Cheese("cheddar", 35);
final InternalFactHandle cheddarHandle1 = new DefaultFactHandle(2, cheddar1);
RightTuple cheddarRightTuple1 = new RightTupleImpl(cheddarHandle1, null);
map.add(cheddarRightTuple1);
final Cheese stilton2 = new Cheese("stilton", 81);
final InternalFactHandle stiltonHandle2 = new DefaultFactHandle(3, stilton2);
RightTuple stiltonRightTuple2 = new RightTupleImpl(stiltonHandle2, null);
map.add(stiltonRightTuple2);
assertEquals(3, map.size());
assertEquals(2, tablePopulationSize(map));
// cheddar is in its own bucket, which should be removed once empty. We cannot have
// empty FieldIndexEntries in the Map, as they get their value from the first FactEntry.
map.remove(cheddarRightTuple1);
assertEquals(2, map.size());
assertEquals(1, tablePopulationSize(map));
// We remove t he stiltonHandle2, but there is still one more stilton, so size should be the same
map.remove(stiltonRightTuple2);
assertEquals(1, map.size());
assertEquals(1, tablePopulationSize(map));
// No more stiltons, so the table should be empty
map.remove(stiltonRightTuple1);
assertEquals(0, map.size());
assertEquals(0, tablePopulationSize(map));
}
use of org.drools.core.rule.Declaration in project drools by kiegroup.
the class RightTupleIndexHashTableTest method testEmptyIterator.
@Test
public void testEmptyIterator() {
final InternalReadAccessor extractor = store.getReader(Cheese.class, "type");
final Pattern pattern = new Pattern(0, new ClassObjectType(Cheese.class));
final Declaration declaration = new Declaration("typeOfCheese", extractor, pattern);
final FieldIndex fieldIndex = new FieldIndex(extractor, declaration, MvelConstraint.INDEX_EVALUATOR);
final TupleIndexHashTable map = new TupleIndexHashTable(new FieldIndex[] { fieldIndex }, false);
final Cheese stilton = new Cheese("stilton", 55);
final InternalFactHandle stiltonHandle = new DefaultFactHandle(2, stilton);
assertNull(map.getFirst(new LeftTupleImpl(stiltonHandle, null, true)));
}
Aggregations