Examples with FromCollectCompositeFactPattern org.drools.workbench.models.datamodel.rule.FromCollectCompositeFactPattern used on opensource projects

Example 6 with FromCollectCompositeFactPattern

use of org.drools.workbench.models.datamodel.rule.FromCollectCompositeFactPattern in project drools by kiegroup.

the class RuleModelDRLPersistenceUnmarshallingTest method testCollectWithFreeFormDRL_MethodsDefined.

@Test
public void testCollectWithFreeFormDRL_MethodsDefined() throws Exception {
    // https://bugzilla.redhat.com/show_bug.cgi?id=1060816
    String drl = "package org.sample.resourceassignment;\n" + "rule \"r1\"\n" + "dialect \"mvel\"\n" + "when\n" + "$trans : Transactions()\n" + "$transactions : java.util.List( eval( size > 0 ) ) from collect ( Transaction() from $trans.getRecCategorization().get(\"APES-01\") )\n" + "then\n" + "end";
    addModelField("Transactions", "recCategorization", "java.util.Map", DataType.TYPE_COLLECTION);
    addMethodInformation("Transactions", "getRecCategorization()", Collections.emptyList(), "java.util.Map", null, DataType.TYPE_COLLECTION);
    addMethodInformation("java.util.Map", "get", new ArrayList<String>() {

        {
            add("p0");
        }
    }, "java.lang.String", null, DataType.TYPE_STRING);
    final RuleModel m = RuleModelDRLPersistenceImpl.getInstance().unmarshal(drl, Collections.emptyList(), dmo);
    assertNotNull(m);
    assertEquals(2, m.lhs.length);
    IPattern p0 = m.lhs[0];
    assertTrue(p0 instanceof FactPattern);
    FactPattern fp0 = (FactPattern) p0;
    assertEquals("Transactions", fp0.getFactType());
    assertEquals("$trans", fp0.getBoundName());
    assertEquals(0, fp0.getNumberOfConstraints());
    IPattern p1 = m.lhs[1];
    assertTrue(p1 instanceof FromCollectCompositeFactPattern);
    FromCollectCompositeFactPattern fp1 = (FromCollectCompositeFactPattern) p1;
    assertEquals("java.util.List", fp1.getFactPattern().getFactType());
    assertEquals("$transactions", fp1.getFactPattern().getBoundName());
    assertEquals(1, fp1.getFactPattern().getNumberOfConstraints());
    assertTrue(fp1.getFactPattern().getConstraint(0) instanceof SingleFieldConstraint);
    SingleFieldConstraint fp1sfc = (SingleFieldConstraint) fp1.getFactPattern().getConstraint(0);
    assertEquals("size > 0", fp1sfc.getValue());
    assertEquals(BaseSingleFieldConstraint.TYPE_PREDICATE, fp1sfc.getConstraintValueType());
    assertTrue(fp1.getRightPattern() instanceof FromCompositeFactPattern);
    FromCompositeFactPattern fp2 = (FromCompositeFactPattern) fp1.getRightPattern();
    assertNotNull(fp2.getFactPattern());
    FactPattern fp3 = fp2.getFactPattern();
    assertEquals("Transaction", fp3.getFactType());
    assertEquals(0, fp3.getNumberOfConstraints());
    assertNotNull(fp2.getExpression());
    ExpressionFormLine efl = fp2.getExpression();
    assertEquals(3, efl.getParts().size());
    assertTrue(efl.getParts().get(0) instanceof ExpressionVariable);
    ExpressionVariable ev = (ExpressionVariable) efl.getParts().get(0);
    assertEquals("$trans", ev.getName());
    assertEquals("Transactions", ev.getClassType());
    assertTrue(efl.getParts().get(1) instanceof ExpressionMethod);
    ExpressionMethod em = (ExpressionMethod) efl.getParts().get(1);
    assertEquals("getRecCategorization()", em.getName());
    assertEquals("java.util.Map", em.getClassType());
    assertEquals(DataType.TYPE_COLLECTION, em.getGenericType());
    assertTrue(efl.getParts().get(2) instanceof ExpressionText);
    ExpressionText et = (ExpressionText) efl.getParts().get(2);
    assertEquals("get(\"APES-01\")", et.getName());
    assertEquals("java.lang.String", et.getClassType());
    assertEquals(DataType.TYPE_STRING, et.getGenericType());
    assertEquals(0, m.rhs.length);
}

Example 7 with FromCollectCompositeFactPattern

use of org.drools.workbench.models.datamodel.rule.FromCollectCompositeFactPattern in project drools by kiegroup.

the class RuleModelDRLPersistenceUnmarshallingTest method testLHSFromCollectWithoutListDeclared.

@Test
public // https://bugzilla.redhat.com/show_bug.cgi?id=1158813
void testLHSFromCollectWithoutListDeclared() throws Exception {
    String drl = "package org.test;\n" + "rule \"rule1\"\n" + "dialect \"mvel\"\n" + "when\n" + "  c : Customer( )\n" + "  items : java.util.List( eval( size == c.items.size )) \n" + "  from collect ( var : Item( price > 10 )) \n" + "then \n" + "end";
    addModelField("org.test.Customer", "this", "org.test.Customer", DataType.TYPE_THIS);
    addModelField("org.test.Item", "this", "org.test.Item", DataType.TYPE_THIS);
    addModelField("org.test.Item", "price", "java.lang.Double", DataType.TYPE_NUMERIC_DOUBLE);
    addMethodInformation("org.test.Customer", "items", new ArrayList<String>() {

        {
        }
    }, DataType.TYPE_NUMERIC_INTEGER, null, DataType.TYPE_NUMERIC_INTEGER);
    when(dmo.getPackageName()).thenReturn("org.test");
    final RuleModel m = RuleModelDRLPersistenceImpl.getInstance().unmarshal(drl, Collections.emptyList(), dmo);
    assertNotNull(m);
    assertEquals(2, m.lhs.length);
    assertEquals(0, m.rhs.length);
    assertTrue(m.lhs[0] instanceof FactPattern);
    final FactPattern fp = (FactPattern) m.lhs[0];
    assertEquals("Customer", fp.getFactType());
    assertEquals("c", fp.getBoundName());
    assertEquals(0, fp.getNumberOfConstraints());
    assertTrue(m.lhs[1] instanceof FromCollectCompositeFactPattern);
    final FromCollectCompositeFactPattern fcfp = (FromCollectCompositeFactPattern) m.lhs[1];
    assertNotNull(fcfp.getFactPattern());
    assertEquals("java.util.List", fcfp.getFactPattern().getFactType());
    assertEquals("items", fcfp.getFactPattern().getBoundName());
    assertEquals(1, fcfp.getFactPattern().getNumberOfConstraints());
    assertTrue(fcfp.getFactPattern().getConstraint(0) instanceof SingleFieldConstraint);
    final SingleFieldConstraint sfc0 = (SingleFieldConstraint) fcfp.getFactPattern().getConstraint(0);
    assertNull(sfc0.getFactType());
    assertNull(sfc0.getFieldName());
    assertEquals("size == c.items.size", sfc0.getValue());
    assertEquals(BaseSingleFieldConstraint.TYPE_PREDICATE, sfc0.getConstraintValueType());
    assertNotNull(fcfp.getRightPattern());
    assertTrue(fcfp.getRightPattern() instanceof FactPattern);
    final FactPattern rfp = (FactPattern) fcfp.getRightPattern();
    assertEquals("Item", rfp.getFactType());
    assertEquals("var", rfp.getBoundName());
    assertEquals(1, rfp.getNumberOfConstraints());
    assertTrue(rfp.getConstraint(0) instanceof SingleFieldConstraint);
    final SingleFieldConstraint sfc1 = (SingleFieldConstraint) rfp.getConstraint(0);
    assertEquals("Item", sfc1.getFactType());
    assertEquals("price", sfc1.getFieldName());
    assertEquals("10", sfc1.getValue());
    assertEquals(">", sfc1.getOperator());
    assertEquals(BaseSingleFieldConstraint.TYPE_LITERAL, sfc1.getConstraintValueType());
    assertEqualsIgnoreWhitespace(drl, RuleModelDRLPersistenceImpl.getInstance().marshal(m));
}

Example 8 with FromCollectCompositeFactPattern

use of org.drools.workbench.models.datamodel.rule.FromCollectCompositeFactPattern in project drools by kiegroup.

the class RuleModelDRLPersistenceImpl method parsePatternSource.

private IFactPattern parsePatternSource(final RuleModel m, final PatternDescr pattern, final PatternSourceDescr patternSource, final boolean isJavaDialect, final Map<String, String> boundParams, final PackageDataModelOracle dmo) {
    if (pattern.getIdentifier() != null) {
        boundParams.put(pattern.getIdentifier(), pattern.getObjectType());
    }
    if (patternSource instanceof AccumulateDescr) {
        AccumulateDescr accumulate = (AccumulateDescr) patternSource;
        FromAccumulateCompositeFactPattern fac = new FromAccumulateCompositeFactPattern();
        fac.setSourcePattern(parseBaseDescr(m, accumulate.getInput(), isJavaDialect, boundParams, dmo));
        fac.setInitCode(accumulate.getInitCode());
        fac.setActionCode(accumulate.getActionCode());
        fac.setReverseCode(accumulate.getReverseCode());
        fac.setResultCode(accumulate.getResultCode());
        FactPattern factPattern = new FactPattern(pattern.getObjectType());
        factPattern.setBoundName(pattern.getIdentifier());
        parseConstraint(m, factPattern, pattern.getConstraint(), isJavaDialect, boundParams, dmo);
        fac.setFactPattern(factPattern);
        for (AccumulateDescr.AccumulateFunctionCallDescr func : accumulate.getFunctions()) {
            String funcName = func.getFunction();
            boolean first = true;
            StringBuilder sb = new StringBuilder();
            for (String param : func.getParams()) {
                if (first) {
                    first = false;
                } else {
                    sb.append(", ");
                }
                sb.append(param);
            }
            fac.setFunction(funcName + "(" + sb + ")");
            break;
        }
        return fac;
    } else if (patternSource instanceof CollectDescr) {
        CollectDescr collect = (CollectDescr) patternSource;
        FromCollectCompositeFactPattern fac = new FromCollectCompositeFactPattern();
        fac.setRightPattern(parseBaseDescr(m, collect.getInputPattern(), isJavaDialect, boundParams, dmo));
        fac.setFactPattern(getFactPattern(m, pattern, isJavaDialect, boundParams, dmo));
        return fac;
    } else if (patternSource instanceof EntryPointDescr) {
        EntryPointDescr entryPoint = (EntryPointDescr) patternSource;
        FromEntryPointFactPattern fep = new FromEntryPointFactPattern();
        fep.setEntryPointName(entryPoint.getText());
        fep.setFactPattern(getFactPattern(m, pattern, isJavaDialect, boundParams, dmo));
        return fep;
    } else if (patternSource instanceof FromDescr) {
        FromDescr from = (FromDescr) patternSource;
        FromCompositeFactPattern fcfp = new FromCompositeFactPattern();
        FactPattern factPattern = new FactPattern(pattern.getObjectType());
        factPattern.setBoundName(pattern.getIdentifier());
        parseConstraint(m, factPattern, pattern.getConstraint(), isJavaDialect, boundParams, dmo);
        fcfp.setFactPattern(factPattern);
        ExpressionFormLine expression = new ExpressionFormLine();
        fcfp.setExpression(expression);
        String dataSource = from.getDataSource().toString();
        String[] splitSource = dataSource.split("\\.");
        ModelField[] fields = null;
        for (int i = 0; i < splitSource.length; i++) {
            String sourcePart = splitSource[i];
            if (i == 0) {
                String type = boundParams.get(sourcePart);
                expression.appendPart(new ExpressionVariable(sourcePart, type, DataType.TYPE_NUMERIC));
                fields = findFields(m, dmo, type);
            } else {
                ModelField modelField = null;
                if (fields != null) {
                    for (ModelField field : fields) {
                        if (field.getName().equals(sourcePart)) {
                            modelField = field;
                            break;
                        }
                    }
                }
                if (modelField == null) {
                    final String previousClassName = expression.getClassType();
                    final List<MethodInfo> mis = dmo.getModuleMethodInformation().get(previousClassName);
                    boolean isMethod = false;
                    if (mis != null) {
                        for (MethodInfo mi : mis) {
                            if (mi.getName().equals(sourcePart)) {
                                expression.appendPart(new ExpressionMethod(mi.getName(), mi.getReturnClassType(), mi.getGenericType(), mi.getParametricReturnType()));
                                isMethod = true;
                                break;
                            }
                        }
                    }
                    if (isMethod == false) {
                        expression.appendPart(new ExpressionText(sourcePart));
                    }
                } else {
                    expression.appendPart(new ExpressionField(sourcePart, modelField.getClassName(), modelField.getType()));
                    fields = findFields(m, dmo, modelField.getClassName());
                }
            }
        }
        return fcfp;
    }
    throw new RuntimeException("Unknown pattern source " + patternSource);
}

Example 9 with FromCollectCompositeFactPattern

use of org.drools.workbench.models.datamodel.rule.FromCollectCompositeFactPattern in project drools-wb by kiegroup.

the class RuleModelCloneVisitor method visitFromCollectCompositeFactPattern.

private FromCollectCompositeFactPattern visitFromCollectCompositeFactPattern(FromCollectCompositeFactPattern pattern) {
    FromCollectCompositeFactPattern clone = new FromCollectCompositeFactPattern();
    clone.setExpression((ExpressionFormLine) visit(pattern.getExpression()));
    clone.setFactPattern((FactPattern) visit(pattern.getFactPattern()));
    clone.setRightPattern((IPattern) visit(pattern.getRightPattern()));
    return clone;
}

Example 10 with FromCollectCompositeFactPattern

use of org.drools.workbench.models.datamodel.rule.FromCollectCompositeFactPattern in project drools-wb by kiegroup.

the class RuleModelCloneVisitorTest method buildFromCollectCompositeFactPattern.

private static FromCollectCompositeFactPattern buildFromCollectCompositeFactPattern() {
    FromCollectCompositeFactPattern fcoll = new FromCollectCompositeFactPattern();
    fcoll.setExpression(buildExpressionFormLine());
    fcoll.setFactPattern(buildFactPattern());
    fcoll.setRightPattern(buildExpressionFormLine());
    return fcoll;
}