Examples with AstBeParameterTag com.developmentontheedge.sql.model.AstBeParameterTag used on opensource projects

Example 1 with AstBeParameterTag

use of com.developmentontheedge.sql.model.AstBeParameterTag in project be5 by DevelopmentOnTheEdge.

the class ContextApplier method applyParameterTag.

private void applyParameterTag(SimpleNode newTree, int i, String key, Function<String, String> varResolver) {
    String value;
    SimpleNode replacement;
    AstBeParameterTag paramNode = (AstBeParameterTag) newTree.child(i);
    String multiple = paramNode.getMultiple();
    boolean tableRefAddend = newTree instanceof AstTableRef && i == 1;
    if (multiple == null) {
        value = context.getParameter(paramNode.getName());
        Map<String, String> propertyMap = getSubQueryPropertyMap(key, varResolver);
        if (value == null && propertyMap.containsKey(paramNode.getName())) {
            value = propertyMap.get(paramNode.getName());
        }
        replacement = applyParameters(paramNode, value, tableRefAddend);
    } else {
        if (newTree instanceof AstInValueList) {
            List<String> values = context.getListParameter(paramNode.getName());
            paramNode.replaceWith(StreamEx.of(values).map(val -> applyParameters(paramNode, val, tableRefAddend)).toArray(SimpleNode[]::new));
            return;
        }
        if (!(newTree instanceof AstInPredicate))
            throw new IllegalArgumentException("Parameter Multiple can only be put inside InPredicate");
        AstInValueList list = new AstInValueList(SqlParserTreeConstants.JJTINVALUELIST);
        List<String> values = context.getListParameter(paramNode.getName());
        if (values != null) {
            for (String val : values) {
                list.addChild(applyParameters(paramNode, val, tableRefAddend));
            }
        }
        replacement = list;
    }
    replacement.inheritFrom(paramNode);
    if (tableRefAddend) {
        AstTableRef tableRef = (AstTableRef) newTree;
        tableRef.setTable(tableRef.getTable() + replacement.format());
        tableRef.removeChild(i);
    } else {
        newTree.jjtAddChild(replacement, i);
    }
}

Example 2 with AstBeParameterTag

use of com.developmentontheedge.sql.model.AstBeParameterTag in project be5 by DevelopmentOnTheEdge.

the class ContextApplier method checkExpression.

private void checkExpression(SimpleNode newTree, String key, Function<String, String> varResolver) {
    for (int i = 0; i < newTree.jjtGetNumChildren(); i++) {
        SimpleNode child = newTree.child(i);
        checkExpression(child, key, varResolver);
        if (child instanceof AstBeCondition)
            i = applyConditions(newTree, i);
        if (child instanceof AstBeParameterTag)
            applyParameterTag(newTree, i, key, varResolver);
        else if (child instanceof AstBePlaceHolder)
            applyPlaceHolder((AstBePlaceHolder) child);
        else if (child instanceof AstBeSessionTag)
            applySessionTag((AstBeSessionTag) child);
        else if (child instanceof AstBeSqlSubQuery)
            applySubQuery((AstBeSqlSubQuery) child);
        else if (child instanceof AstBeSqlAuto)
            applyAutoQuery((AstBeSqlAuto) child);
        else if (child instanceof AstBeConditionChain) {
            applyConditionChain((AstBeConditionChain) child);
            i--;
        } else if (child instanceof AstBeSql)
            applySqlTag((AstBeSql) child);
        else if (child instanceof AstBeDictionary)
            applyDictionary((AstBeDictionary) child);
    }
}