com.groupbyinc.common.jackson.jq.internal.functions.IndicesFunction Maven / Gradle / Ivy
package net.thisptr.jackson.jq.internal.functions;
import java.util.ArrayList;
import java.util.List;
import net.thisptr.jackson.jq.Function;
import net.thisptr.jackson.jq.JsonQuery;
import net.thisptr.jackson.jq.Scope;
import net.thisptr.jackson.jq.exception.JsonQueryException;
import net.thisptr.jackson.jq.internal.BuiltinFunction;
import net.thisptr.jackson.jq.internal.misc.JsonNodeComparator;
import net.thisptr.jackson.jq.internal.misc.Preconditions;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.JsonNodeType;
@BuiltinFunction("indices/1")
public class IndicesFunction implements Function {
private static final JsonNodeComparator comparator = JsonNodeComparator.getInstance();
@Override
public List apply(final Scope scope, final List args, final JsonNode in) throws JsonQueryException {
Preconditions.checkInputType("indices", in, JsonNodeType.STRING, JsonNodeType.ARRAY);
final List out = new ArrayList<>();
for (final JsonNode needle : args.get(0).apply(scope, in)) {
final ArrayNode indices = scope.getObjectMapper().createArrayNode();
for (final int index : indices(needle, in))
indices.add(index);
out.add(indices);
}
return out;
}
public static List indices(final JsonNode needle, final JsonNode haystack) throws JsonQueryException {
final List result = new ArrayList<>();
if (needle.isTextual() && haystack.isTextual()) {
final String haystackText = haystack.asText();
final String needleText = needle.asText();
for (int index = haystackText.indexOf(needleText); index >= 0; index = haystackText.indexOf(needleText, index + 1))
result.add(index);
} else if (needle.isArray() && haystack.isArray()) {
if (needle.size() != 0) {
for (int i = 0; i < haystack.size(); ++i) {
boolean match = true;
for (int j = 0; j < needle.size(); ++j) {
if (i + j >= haystack.size() || comparator.compare(haystack.get(i + j), needle.get(j)) != 0) {
match = false;
break;
}
}
if (match)
result.add(i);
}
}
} else if (haystack.isArray()) {
for (int i = 0; i < haystack.size(); ++i)
if (comparator.compare(haystack.get(i), needle) == 0)
result.add(i);
} else {
throw new JsonQueryException("indices() is not applicable to " + haystack.getNodeType());
}
return result;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy