net.thisptr.jackson.jq.internal.functions.SortFunction Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of common-test Show documentation

${project.name}

There is a newer version: 198

package net.thisptr.jackson.jq.internal.functions;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
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.JsonNodeUtils;
import net.thisptr.jackson.jq.internal.misc.Pair;
import net.thisptr.jackson.jq.internal.misc.Preconditions;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.JsonNodeType;

@BuiltinFunction("sort/1")
public class SortFunction implements Function {
	private static final JsonNodeComparator comparator = JsonNodeComparator.getInstance();

	@Override
	public List apply(final Scope scope, final List args, final JsonNode items) throws JsonQueryException {
		Preconditions.checkInputType("sort", items, JsonNodeType.ARRAY);

		final List> zipped = new ArrayList<>(items.size());
		for (final JsonNode item : items) {
			final JsonNode value = JsonNodeUtils.asArrayNode(scope.getObjectMapper(), args.get(0).apply(scope, item));
			zipped.add(Pair.of(item, value));
		}

		Collections.sort(zipped, new Comparator>() {
			@Override
			public int compare(final Pair o1, final Pair o2) {
				return comparator.compare(o1._2, o2._2);
			}
		});

		return Collections. singletonList(JsonNodeUtils.asArrayNode(scope.getObjectMapper(), Pair._1(zipped)));
	}
}