Search in sources :

Example 1 with DashNode

use of org.kie.dmn.feel.lang.ast.DashNode in project drools by kiegroup.

the class FEELImpl method evaluateUnaryTests.

@Override
public List<UnaryTest> evaluateUnaryTests(String expression, Map<String, Type> variableTypes) {
    // DMN defines a special case where, unless the expressions are unary tests
    // or ranges, they need to be converted into an equality test unary expression.
    // This way, we have to compile and check the low level AST nodes to properly
    // deal with this case
    CompilerContext ctx = newCompilerContext();
    for (Map.Entry<String, Type> e : variableTypes.entrySet()) {
        ctx.addInputVariableType(e.getKey(), e.getValue());
    }
    CompiledExpressionImpl compiledExpression = (CompiledExpressionImpl) compileExpressionList(expression, ctx);
    if (compiledExpression != null) {
        ListNode listNode = (ListNode) compiledExpression.getExpression();
        List<BaseNode> tests = new ArrayList<>();
        for (BaseNode o : listNode.getElements()) {
            if (o == null) {
                // not much we can do, so just skip it. Error was reported somewhere else
                continue;
            } else if (o instanceof UnaryTestNode || o instanceof DashNode) {
                tests.add(o);
            } else if (o instanceof RangeNode || o instanceof ListNode) {
                tests.add(new UnaryTestNode("in", o));
            } else {
                tests.add(new UnaryTestNode("=", o));
            }
        }
        listNode.setElements(tests);
        compiledExpression.setExpression(listNode);
        // now we can evaluate the expression to build the list of unary tests
        List<UnaryTest> uts = (List<UnaryTest>) evaluate(compiledExpression, FEELImpl.EMPTY_INPUT);
        return uts;
    }
    return Collections.emptyList();
}
Also used : UnaryTestNode(org.kie.dmn.feel.lang.ast.UnaryTestNode) CompilerContext(org.kie.dmn.feel.lang.CompilerContext) BaseNode(org.kie.dmn.feel.lang.ast.BaseNode) ArrayList(java.util.ArrayList) ListNode(org.kie.dmn.feel.lang.ast.ListNode) UnaryTest(org.kie.dmn.feel.runtime.UnaryTest) Type(org.kie.dmn.feel.lang.Type) ArrayList(java.util.ArrayList) List(java.util.List) DashNode(org.kie.dmn.feel.lang.ast.DashNode) HashMap(java.util.HashMap) Map(java.util.Map) RangeNode(org.kie.dmn.feel.lang.ast.RangeNode)

Aggregations

ArrayList (java.util.ArrayList)1 HashMap (java.util.HashMap)1 List (java.util.List)1 Map (java.util.Map)1 CompilerContext (org.kie.dmn.feel.lang.CompilerContext)1 Type (org.kie.dmn.feel.lang.Type)1 BaseNode (org.kie.dmn.feel.lang.ast.BaseNode)1 DashNode (org.kie.dmn.feel.lang.ast.DashNode)1 ListNode (org.kie.dmn.feel.lang.ast.ListNode)1 RangeNode (org.kie.dmn.feel.lang.ast.RangeNode)1 UnaryTestNode (org.kie.dmn.feel.lang.ast.UnaryTestNode)1 UnaryTest (org.kie.dmn.feel.runtime.UnaryTest)1