io.redisearch.querybuilder.QueryNode Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jredisearch Show documentation
Official client for Redis Search
The newest version!
package io.redisearch.querybuilder;

import java.util.*;

public abstract class QueryNode implements Node {
    private final List children = new ArrayList<>();
    protected abstract String getJoinString();

    /**
     * Add a match criteria to this node
     * @param field The field to check. If null or empty, then any field is checked
     * @param values Values to check for.
     * @return The current node, for chaining.
     */
    public QueryNode add(String field, Value ...values) {
        children.add(new ValueNode(field, getJoinString(), values));
        return this;
    }

    /**
     * Convenience method to add a list of string values
     * @param field Field to check for
     * @param values One or more string values.
     * @return The current node, for chaining.
     */
    public QueryNode add(String field, String ...values) {
        children.add(new ValueNode(field, getJoinString(), values));
        return this;
    }

    /**
     * Add a list of values from a collection
     * @param field The field to check
     * @param values Collection of values to match
     * @return The current node for chaining.
     */
    public QueryNode add(String field, Collection values) {
        return add(field, values.toArray(new Value[0]));
    }

    /**
     * Add children nodes to this node.
     * @param nodes Children nodes to add
     * @return The current node, for chaining.
     */
    public QueryNode add(Node ...nodes) {
        children.addAll(Arrays.asList(nodes));
        return this;
    }

    protected boolean shouldUseParens(ParenMode mode) {
        if (mode == ParenMode.ALWAYS) {
            return true;
        } 
        if (mode == ParenMode.NEVER) {
            return false;
        }
        return children.size() > 1;
    }

    @Override
    public String toString(ParenMode parenMode) {
        StringBuilder sb = new StringBuilder();
        StringJoiner sj = new StringJoiner(getJoinString());
        if (shouldUseParens(parenMode)) {
            sb.append('(');
        }
        for (Node n : children) {
            sj.add(n.toString(parenMode));
        }
        sb.append(sj.toString());
        if (shouldUseParens(parenMode)) {
            sb.append(')');
        }
        return sb.toString();
    }

    @Override
    public String toString() {
        return toString(ParenMode.DEFAULT);
    }
}