net.thisptr.jackson.jq.internal.functions.LengthFunction 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.functions;

import java.util.List;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.IntNode;
import com.google.auto.service.AutoService;

import net.thisptr.jackson.jq.BuiltinFunction;
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.exception.JsonQueryTypeException;
import net.thisptr.jackson.jq.internal.misc.JsonNodeUtils;
import net.thisptr.jackson.jq.internal.misc.UnicodeUtils;
import net.thisptr.jackson.jq.path.Path;

@AutoService(Function.class)
@BuiltinFunction("length/0")
public class LengthFunction implements Function {
	@Override
	public void apply(final Scope scope, final List args, final JsonNode in, final Path ipath, final PathOutput output, final Version version) throws JsonQueryException {
		output.emit(length(in), null);
	}

	public JsonNode length(final JsonNode in) throws JsonQueryException {
		if (in.isTextual()) {
			return IntNode.valueOf(UnicodeUtils.lengthUtf32(in.asText()));
		} else if (in.isArray() || in.isObject()) {
			return new IntNode(in.size());
		} else if (in.isNull()) {
			return new IntNode(0);
		} else if (in.isNumber()) {
			return JsonNodeUtils.asNumericNode(Math.abs(in.asDouble()));
		} else {
			throw new JsonQueryTypeException("%s has no length", in);
		}
	}
}