Examples with QueryDescrBuilder org.drools.compiler.lang.api.QueryDescrBuilder used on opensource projects

Example 1 with QueryDescrBuilder

use of org.drools.compiler.lang.api.QueryDescrBuilder in project drools by kiegroup.

the class DRL6StrictParser method query.

/* ------------------------------------------------------------------------------------------------
     *                         QUERY STATEMENT
     * ------------------------------------------------------------------------------------------------ */
/**
 * query := annotation* QUERY stringId parameters? lhsExpression END
 *
 * @return
 * @throws org.antlr.runtime.RecognitionException
 */
public RuleDescr query(PackageDescrBuilder pkg) throws RecognitionException {
    QueryDescrBuilder query = null;
    try {
        query = helper.start(pkg, QueryDescrBuilder.class, null);
        setAnnotationsOn(query);
        // 'query'
        match(input, DRL6Lexer.ID, DroolsSoftKeywords.QUERY, null, DroolsEditorType.KEYWORD);
        if (state.failed)
            return null;
        if (helper.validateIdentifierKey(DroolsSoftKeywords.WHEN) || helper.validateIdentifierKey(DroolsSoftKeywords.THEN) || helper.validateIdentifierKey(DroolsSoftKeywords.END)) {
            failMissingTokenException();
            // in case it is backtracking
            return null;
        }
        String name = stringId();
        if (state.backtracking == 0)
            query.name(name);
        if (state.failed)
            return null;
        if (state.backtracking == 0) {
            helper.emit(Location.LOCATION_RULE_HEADER);
        }
        if (speculateParameters(true)) {
            // parameters
            parameters(query, true);
            if (state.failed)
                return null;
            if (state.backtracking == 0) {
                helper.emit(Location.LOCATION_LHS_BEGIN_OF_CONDITION);
            }
        } else if (speculateParameters(false)) {
            // parameters
            parameters(query, false);
            if (state.failed)
                return null;
            if (state.backtracking == 0) {
                helper.emit(Location.LOCATION_LHS_BEGIN_OF_CONDITION);
            }
        }
        if (state.backtracking == 0 && input.LA(1) != DRL6Lexer.EOF) {
            helper.emit(Location.LOCATION_LHS_BEGIN_OF_CONDITION);
        }
        if (input.LA(1) != DRL6Lexer.EOF) {
            lhsExpression(query != null ? query.lhs() : null);
        }
        match(input, DRL6Lexer.ID, DroolsSoftKeywords.END, null, DroolsEditorType.KEYWORD);
        if (state.failed)
            return null;
        helper.emit(Location.LOCATION_RHS);
    } catch (RecognitionException re) {
        reportError(re);
    } finally {
        helper.end(QueryDescrBuilder.class, query);
    }
    return (query != null) ? query.getDescr() : null;
}

Example 2 with QueryDescrBuilder

use of org.drools.compiler.lang.api.QueryDescrBuilder in project drools by kiegroup.

the class ParserHelper method start.

@SuppressWarnings("unchecked")
public <T extends DescrBuilder<?, ?>> T start(DescrBuilder<?, ?> ctxBuilder, Class<T> clazz, String param) {
    if (state.backtracking == 0) {
        if (PackageDescrBuilder.class.isAssignableFrom(clazz)) {
            pushParaphrases(DroolsParaphraseTypes.PACKAGE);
            beginSentence(DroolsSentenceType.PACKAGE);
            setStart(ctxBuilder);
        } else if (ImportDescrBuilder.class.isAssignableFrom(clazz)) {
            ImportDescrBuilder imp;
            if (validateLT(2, DroolsSoftKeywords.FUNCTION) || validateLT(2, DroolsSoftKeywords.STATIC)) {
                imp = ctxBuilder == null ? DescrFactory.newPackage().newFunctionImport() : ((PackageDescrBuilder) ctxBuilder).newFunctionImport();
            } else {
                imp = ctxBuilder == null ? DescrFactory.newPackage().newImport() : ((PackageDescrBuilder) ctxBuilder).newImport();
            }
            pushParaphrases(DroolsParaphraseTypes.IMPORT);
            beginSentence(DroolsSentenceType.IMPORT_STATEMENT);
            setStart(imp);
            return (T) imp;
        } else if (UnitDescrBuilder.class.isAssignableFrom(clazz)) {
            UnitDescrBuilder imp = ctxBuilder == null ? DescrFactory.newPackage().newUnit() : ((PackageDescrBuilder) ctxBuilder).newUnit();
            pushParaphrases(DroolsParaphraseTypes.UNIT);
            beginSentence(DroolsSentenceType.UNIT);
            setStart(imp);
            return (T) imp;
        } else if (AccumulateImportDescrBuilder.class.isAssignableFrom(clazz)) {
            AccumulateImportDescrBuilder imp = ctxBuilder == null ? DescrFactory.newPackage().newAccumulateImport() : ((PackageDescrBuilder) ctxBuilder).newAccumulateImport();
            pushParaphrases(DroolsParaphraseTypes.ACCUMULATE_IMPORT);
            beginSentence(DroolsSentenceType.ACCUMULATE_IMPORT_STATEMENT);
            setStart(imp);
            return (T) imp;
        } else if (GlobalDescrBuilder.class.isAssignableFrom(clazz)) {
            GlobalDescrBuilder global = ctxBuilder == null ? DescrFactory.newPackage().newGlobal() : ((PackageDescrBuilder) ctxBuilder).newGlobal();
            pushParaphrases(DroolsParaphraseTypes.GLOBAL);
            beginSentence(DroolsSentenceType.GLOBAL);
            setStart(global);
            return (T) global;
        } else if (DeclareDescrBuilder.class.isAssignableFrom(clazz)) {
            DeclareDescrBuilder declare = ctxBuilder == null ? DescrFactory.newPackage().newDeclare() : ((PackageDescrBuilder) ctxBuilder).newDeclare();
            return (T) declare;
        } else if (TypeDeclarationDescrBuilder.class.isAssignableFrom(clazz)) {
            TypeDeclarationDescrBuilder declare = ctxBuilder == null ? DescrFactory.newPackage().newDeclare().type() : ((DeclareDescrBuilder) ctxBuilder).type();
            pushParaphrases(DroolsParaphraseTypes.TYPE_DECLARE);
            beginSentence(DroolsSentenceType.TYPE_DECLARATION);
            setStart(declare);
            return (T) declare;
        } else if (EnumDeclarationDescrBuilder.class.isAssignableFrom(clazz)) {
            EnumDeclarationDescrBuilder declare = ctxBuilder == null ? DescrFactory.newPackage().newDeclare().enumerative() : ((DeclareDescrBuilder) ctxBuilder).enumerative();
            pushParaphrases(DroolsParaphraseTypes.ENUM_DECLARE);
            beginSentence(DroolsSentenceType.ENUM_DECLARATION);
            setStart(declare);
            return (T) declare;
        } else if (EntryPointDeclarationDescrBuilder.class.isAssignableFrom(clazz)) {
            EntryPointDeclarationDescrBuilder declare = ctxBuilder == null ? DescrFactory.newPackage().newDeclare().entryPoint() : ((DeclareDescrBuilder) ctxBuilder).entryPoint();
            pushParaphrases(DroolsParaphraseTypes.ENTRYPOINT_DECLARE);
            beginSentence(DroolsSentenceType.ENTRYPOINT_DECLARATION);
            setStart(declare);
            return (T) declare;
        } else if (WindowDeclarationDescrBuilder.class.isAssignableFrom(clazz)) {
            WindowDeclarationDescrBuilder declare = ctxBuilder == null ? DescrFactory.newPackage().newDeclare().window() : ((DeclareDescrBuilder) ctxBuilder).window();
            pushParaphrases(DroolsParaphraseTypes.WINDOW_DECLARE);
            beginSentence(DroolsSentenceType.WINDOW_DECLARATION);
            setStart(declare);
            return (T) declare;
        } else if (FieldDescrBuilder.class.isAssignableFrom(clazz)) {
            FieldDescrBuilder field = ((AbstractClassTypeDeclarationBuilder) ctxBuilder).newField(param);
            setStart(field);
            return (T) field;
        } else if (EnumLiteralDescrBuilder.class.isAssignableFrom(clazz)) {
            EnumLiteralDescrBuilder literal = ((EnumDeclarationDescrBuilder) ctxBuilder).newEnumLiteral(param);
            setStart(literal);
            return (T) literal;
        } else if (FunctionDescrBuilder.class.isAssignableFrom(clazz)) {
            FunctionDescrBuilder function;
            if (ctxBuilder == null) {
                function = DescrFactory.newPackage().newFunction();
            } else {
                PackageDescrBuilder pkg = (PackageDescrBuilder) ctxBuilder;
                function = pkg.newFunction().namespace(pkg.getDescr().getName());
                AttributeDescr attribute = pkg.getDescr().getAttribute("dialect");
                if (attribute != null) {
                    function.dialect(attribute.getValue());
                }
            }
            pushParaphrases(DroolsParaphraseTypes.FUNCTION);
            beginSentence(DroolsSentenceType.FUNCTION);
            setStart(function);
            return (T) function;
        } else if (RuleDescrBuilder.class.isAssignableFrom(clazz)) {
            RuleDescrBuilder rule = ctxBuilder == null ? DescrFactory.newPackage().newRule() : ((PackageDescrBuilder) ctxBuilder).newRule();
            pushParaphrases(DroolsParaphraseTypes.RULE);
            beginSentence(DroolsSentenceType.RULE);
            setStart(rule);
            return (T) rule;
        } else if (QueryDescrBuilder.class.isAssignableFrom(clazz)) {
            QueryDescrBuilder query = ctxBuilder == null ? DescrFactory.newPackage().newQuery() : ((PackageDescrBuilder) ctxBuilder).newQuery();
            pushParaphrases(DroolsParaphraseTypes.QUERY);
            beginSentence(DroolsSentenceType.QUERY);
            setStart(query);
            return (T) query;
        } else if (AttributeDescrBuilder.class.isAssignableFrom(clazz)) {
            AttributeDescrBuilder<?> attribute = ((AttributeSupportBuilder<?>) ctxBuilder).attribute(param);
            setStart(attribute);
            return (T) attribute;
        } else if (EvalDescrBuilder.class.isAssignableFrom(clazz)) {
            EvalDescrBuilder<?> eval = ((CEDescrBuilder<?, ?>) ctxBuilder).eval();
            pushParaphrases(DroolsParaphraseTypes.EVAL);
            beginSentence(DroolsSentenceType.EVAL);
            setStart(eval);
            return (T) eval;
        } else if (ForallDescrBuilder.class.isAssignableFrom(clazz)) {
            ForallDescrBuilder<?> forall = ((CEDescrBuilder<?, ?>) ctxBuilder).forall();
            setStart(forall);
            return (T) forall;
        } else if (CEDescrBuilder.class.isAssignableFrom(clazz)) {
            setStart(ctxBuilder);
            return (T) ctxBuilder;
        } else if (PatternDescrBuilder.class.isAssignableFrom(clazz)) {
            PatternDescrBuilder<?> pattern = ((PatternContainerDescrBuilder<?, ?>) ctxBuilder).pattern();
            pushParaphrases(DroolsParaphraseTypes.PATTERN);
            setStart(pattern);
            return (T) pattern;
        } else if (CollectDescrBuilder.class.isAssignableFrom(clazz)) {
            CollectDescrBuilder<?> collect = ((PatternDescrBuilder<?>) ctxBuilder).from().collect();
            setStart(collect);
            return (T) collect;
        } else if (AccumulateDescrBuilder.class.isAssignableFrom(clazz)) {
            AccumulateDescrBuilder<?> accumulate = ((PatternDescrBuilder<?>) ctxBuilder).from().accumulate();
            setStart(accumulate);
            return (T) accumulate;
        } else if (BehaviorDescrBuilder.class.isAssignableFrom(clazz)) {
            BehaviorDescrBuilder<?> behavior = ((PatternDescrBuilder<?>) ctxBuilder).behavior();
            setStart(behavior);
            return (T) behavior;
        } else if (NamedConsequenceDescrBuilder.class.isAssignableFrom(clazz)) {
            NamedConsequenceDescrBuilder<?> namedConsequence = ((CEDescrBuilder<?, ?>) ctxBuilder).namedConsequence();
            setStart(namedConsequence);
            return (T) namedConsequence;
        } else if (ConditionalBranchDescrBuilder.class.isAssignableFrom(clazz)) {
            ConditionalBranchDescrBuilder<?> conditionalBranch = ((CEDescrBuilder<?, ?>) ctxBuilder).conditionalBranch();
            setStart(conditionalBranch);
            return (T) conditionalBranch;
        }
    }
    return null;
}

Example 3 with QueryDescrBuilder

use of org.drools.compiler.lang.api.QueryDescrBuilder in project drools by kiegroup.

the class PackageDescrBuilderImpl method newQuery.

public QueryDescrBuilder newQuery() {
    QueryDescrBuilder query = new QueryDescrBuilderImpl(this);
    descr.addRule(initDescr(query));
    query.getDescr().setUnit(descr.getUnit());
    return query;
}

Example 4 with QueryDescrBuilder

use of org.drools.compiler.lang.api.QueryDescrBuilder in project drools by kiegroup.

the class DRL6Parser method query.

/* ------------------------------------------------------------------------------------------------
     *                         QUERY STATEMENT
     * ------------------------------------------------------------------------------------------------ */
/**
 * query := QUERY stringId parameters? annotation* lhsExpression END
 *
 * @return
 * @throws org.antlr.runtime.RecognitionException
 */
public RuleDescr query(PackageDescrBuilder pkg) throws RecognitionException {
    QueryDescrBuilder query = null;
    try {
        query = helper.start(pkg, QueryDescrBuilder.class, null);
        // 'query'
        match(input, DRL6Lexer.ID, DroolsSoftKeywords.QUERY, null, DroolsEditorType.KEYWORD);
        if (state.failed)
            return null;
        if (helper.validateIdentifierKey(DroolsSoftKeywords.WHEN) || helper.validateIdentifierKey(DroolsSoftKeywords.THEN) || helper.validateIdentifierKey(DroolsSoftKeywords.END)) {
            failMissingTokenException();
            // in case it is backtracking
            return null;
        }
        String name = stringId();
        if (state.backtracking == 0)
            query.name(name);
        if (state.failed)
            return null;
        if (state.backtracking == 0) {
            helper.emit(Location.LOCATION_RULE_HEADER);
        }
        if (speculateParameters(true)) {
            // parameters
            parameters(query, true);
            if (state.failed)
                return null;
            if (state.backtracking == 0) {
                helper.emit(Location.LOCATION_LHS_BEGIN_OF_CONDITION);
            }
        } else if (speculateParameters(false)) {
            // parameters
            parameters(query, false);
            if (state.failed)
                return null;
            if (state.backtracking == 0) {
                helper.emit(Location.LOCATION_LHS_BEGIN_OF_CONDITION);
            }
        }
        while (input.LA(1) == DRL6Lexer.AT) {
            // annotation*
            annotation(query);
            if (state.failed)
                return null;
        }
        if (state.backtracking == 0 && input.LA(1) != DRL6Lexer.EOF) {
            helper.emit(Location.LOCATION_LHS_BEGIN_OF_CONDITION);
        }
        if (input.LA(1) != DRL6Lexer.EOF) {
            lhsExpression(query != null ? query.lhs() : null);
        }
        match(input, DRL6Lexer.ID, DroolsSoftKeywords.END, null, DroolsEditorType.KEYWORD);
        if (state.failed)
            return null;
        helper.emit(Location.LOCATION_RHS);
    } catch (RecognitionException re) {
        reportError(re);
    } finally {
        helper.end(QueryDescrBuilder.class, query);
    }
    return (query != null) ? query.getDescr() : null;
}

Example 5 with QueryDescrBuilder

use of org.drools.compiler.lang.api.QueryDescrBuilder in project drools by kiegroup.

the class DRL5Parser method query.

/* ------------------------------------------------------------------------------------------------
     *                         QUERY STATEMENT
     * ------------------------------------------------------------------------------------------------ */
/**
 * query := QUERY stringId parameters? annotation* lhsExpression END
 *
 * @return
 * @throws RecognitionException
 */
public RuleDescr query(PackageDescrBuilder pkg) throws RecognitionException {
    QueryDescrBuilder query = null;
    try {
        query = helper.start(pkg, QueryDescrBuilder.class, null);
        // 'query'
        match(input, DRL5Lexer.ID, DroolsSoftKeywords.QUERY, null, DroolsEditorType.KEYWORD);
        if (state.failed)
            return null;
        if (helper.validateIdentifierKey(DroolsSoftKeywords.WHEN) || helper.validateIdentifierKey(DroolsSoftKeywords.THEN) || helper.validateIdentifierKey(DroolsSoftKeywords.END)) {
            failMissingTokenException();
            // in case it is backtracking
            return null;
        }
        String name = stringId();
        if (state.backtracking == 0)
            query.name(name);
        if (state.failed)
            return null;
        if (state.backtracking == 0) {
            helper.emit(Location.LOCATION_RULE_HEADER);
        }
        if (speculateParameters(true)) {
            // parameters
            parameters(query, true);
            if (state.failed)
                return null;
            if (state.backtracking == 0) {
                helper.emit(Location.LOCATION_LHS_BEGIN_OF_CONDITION);
            }
        } else if (speculateParameters(false)) {
            // parameters
            parameters(query, false);
            if (state.failed)
                return null;
            if (state.backtracking == 0) {
                helper.emit(Location.LOCATION_LHS_BEGIN_OF_CONDITION);
            }
        }
        while (input.LA(1) == DRL5Lexer.AT) {
            // annotation*
            annotation(query);
            if (state.failed)
                return null;
        }
        if (state.backtracking == 0 && input.LA(1) != DRL5Lexer.EOF) {
            helper.emit(Location.LOCATION_LHS_BEGIN_OF_CONDITION);
        }
        if (input.LA(1) != DRL5Lexer.EOF) {
            lhsExpression(query != null ? query.lhs() : null);
        }
        match(input, DRL5Lexer.ID, DroolsSoftKeywords.END, null, DroolsEditorType.KEYWORD);
        if (state.failed)
            return null;
        helper.emit(Location.LOCATION_RHS);
    } catch (RecognitionException re) {
        reportError(re);
    } finally {
        helper.end(QueryDescrBuilder.class, query);
    }
    return (query != null) ? query.getDescr() : null;
}