Examples with StringLiteralEvaluator org.apache.nifi.attribute.expression.language.evaluation.literals.StringLiteralEvaluator used on opensource projects

Example 1 with StringLiteralEvaluator

use of org.apache.nifi.attribute.expression.language.evaluation.literals.StringLiteralEvaluator in project nifi by apache.

the class ExpressionCompiler method newStringLiteralEvaluator.

private Evaluator<String> newStringLiteralEvaluator(final String literalValue) {
    if (literalValue == null || literalValue.length() < 2) {
        return addToken(new StringLiteralEvaluator(literalValue), literalValue);
    }
    final List<Range> ranges = Query.extractExpressionRanges(literalValue);
    if (ranges.isEmpty()) {
        return addToken(new StringLiteralEvaluator(literalValue), literalValue);
    }
    final List<Evaluator<?>> evaluators = new ArrayList<>();
    int lastIndex = 0;
    for (final Range range : ranges) {
        if (range.getStart() > lastIndex) {
            evaluators.add(newStringLiteralEvaluator(literalValue.substring(lastIndex, range.getStart())));
        }
        final String treeText = literalValue.substring(range.getStart(), range.getEnd() + 1);
        evaluators.add(buildEvaluator(compileTree(treeText)));
        lastIndex = range.getEnd() + 1;
    }
    final Range lastRange = ranges.get(ranges.size() - 1);
    if (lastRange.getEnd() + 1 < literalValue.length()) {
        final String treeText = literalValue.substring(lastRange.getEnd() + 1);
        evaluators.add(newStringLiteralEvaluator(treeText));
    }
    if (evaluators.size() == 1) {
        return toStringEvaluator(evaluators.get(0));
    }
    Evaluator<String> lastEvaluator = toStringEvaluator(evaluators.get(0));
    for (int i = 1; i < evaluators.size(); i++) {
        lastEvaluator = new AppendEvaluator(lastEvaluator, toStringEvaluator(evaluators.get(i)));
    }
    this.evaluators.addAll(evaluators);
    return lastEvaluator;
}

Example 2 with StringLiteralEvaluator

use of org.apache.nifi.attribute.expression.language.evaluation.literals.StringLiteralEvaluator in project nifi by apache.

the class StandardPreparedQuery method getVariableImpact.

@Override
public VariableImpact getVariableImpact() {
    final VariableImpact existing = this.variableImpact;
    if (existing != null) {
        return existing;
    }
    final Set<String> variables = new HashSet<>();
    for (final Expression expression : expressions) {
        if (!(expression instanceof CompiledExpression)) {
            continue;
        }
        final CompiledExpression compiled = (CompiledExpression) expression;
        for (final Evaluator<?> evaluator : compiled.getAllEvaluators()) {
            if (evaluator instanceof AttributeEvaluator) {
                final AttributeEvaluator attributeEval = (AttributeEvaluator) evaluator;
                final Evaluator<String> nameEval = attributeEval.getNameEvaluator();
                if (nameEval instanceof StringLiteralEvaluator) {
                    final String referencedVar = nameEval.evaluate(Collections.emptyMap()).getValue();
                    variables.add(referencedVar);
                }
            } else if (evaluator instanceof AllAttributesEvaluator) {
                final AllAttributesEvaluator allAttrsEval = (AllAttributesEvaluator) evaluator;
                final MultiAttributeEvaluator iteratingEval = allAttrsEval.getVariableIteratingEvaluator();
                if (iteratingEval instanceof MultiNamedAttributeEvaluator) {
                    variables.addAll(((MultiNamedAttributeEvaluator) iteratingEval).getAttributeNames());
                } else if (iteratingEval instanceof MultiMatchAttributeEvaluator) {
                    return VariableImpact.ALWAYS_IMPACTED;
                }
            } else if (evaluator instanceof AnyAttributeEvaluator) {
                final AnyAttributeEvaluator allAttrsEval = (AnyAttributeEvaluator) evaluator;
                final MultiAttributeEvaluator iteratingEval = allAttrsEval.getVariableIteratingEvaluator();
                if (iteratingEval instanceof MultiNamedAttributeEvaluator) {
                    variables.addAll(((MultiNamedAttributeEvaluator) iteratingEval).getAttributeNames());
                } else if (iteratingEval instanceof MultiMatchAttributeEvaluator) {
                    return VariableImpact.ALWAYS_IMPACTED;
                }
            } else if (evaluator instanceof MappingEvaluator) {
                final MappingEvaluator<?> allAttrsEval = (MappingEvaluator<?>) evaluator;
                final MultiAttributeEvaluator iteratingEval = allAttrsEval.getVariableIteratingEvaluator();
                if (iteratingEval instanceof MultiNamedAttributeEvaluator) {
                    variables.addAll(((MultiNamedAttributeEvaluator) iteratingEval).getAttributeNames());
                }
            }
        }
    }
    final VariableImpact impact = new NamedVariableImpact(variables);
    this.variableImpact = impact;
    return impact;
}