net.thisptr.jackson.jq.internal.JsonArgumentFunction Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jackson-jq Show documentation

There is a newer version: 1.2.0

package net.thisptr.jackson.jq.internal;

import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

import com.fasterxml.jackson.databind.JsonNode;

import net.thisptr.jackson.jq.Expression;
import net.thisptr.jackson.jq.Function;
import net.thisptr.jackson.jq.PathOutput;
import net.thisptr.jackson.jq.Scope;
import net.thisptr.jackson.jq.Version;
import net.thisptr.jackson.jq.exception.JsonQueryException;
import net.thisptr.jackson.jq.path.Path;

public abstract class JsonArgumentFunction implements Function {
	protected abstract JsonNode fn(final List args, final JsonNode in) throws JsonQueryException;

	private void combinations(final PathOutput output, final Stack args, final int index, final List> argmat, final JsonNode in) throws JsonQueryException {
		if (index >= argmat.size()) {
			output.emit(fn(args, in), null);
			return;
		}

		for (final JsonNode arg : argmat.get(index)) {
			args.push(arg);
			combinations(output, args, index + 1, argmat, in);
			args.pop();
		}
	}

	@Override
	public void apply(final Scope scope, final List args, final JsonNode in, final Path ipath, final PathOutput output, final Version version) throws JsonQueryException {
		final List> _args = new ArrayList<>(args.size());
		for (final Expression arg : args) {
			final List out = new ArrayList<>();
			arg.apply(scope, in, out::add);
			_args.add(out);
		}

		combinations(output, new Stack<>(), 0, _args, in);
	}
}