net.thisptr.jackson.jq.internal.operators.MultiplyOperator 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.operators;

import java.util.Iterator;
import java.util.Map.Entry;

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.Strings;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.NullNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.fasterxml.jackson.databind.node.TextNode;

public class MultiplyOperator implements BinaryOperator {
	@Override
	public JsonNode apply(ObjectMapper mapper, JsonNode lhs, JsonNode rhs) throws JsonQueryException {
		if (lhs.isIntegralNumber() && rhs.isIntegralNumber()) {
			final long r = lhs.asLong() * rhs.asLong();
			return JsonNodeUtils.asNumericNode(r);
		} else if (lhs.isNumber() && rhs.isNumber()) {
			final double r = lhs.asDouble() * rhs.asDouble();
			return JsonNodeUtils.asNumericNode(r);
		} else if (lhs.isTextual() && rhs.canConvertToInt()) {
			final int count = rhs.asInt();
			if (count <= 0)
				return NullNode.getInstance();
			return new TextNode(Strings.repeat(lhs.asText(), count));
		} else if (lhs.isObject() && rhs.isObject()) {
			return mergeRecursive(mapper, (ObjectNode) lhs, (ObjectNode) rhs);
		} else {
			throw new JsonQueryTypeException(lhs, rhs, "cannot be multiplied");
		}
	}

	private static ObjectNode mergeRecursive(final ObjectMapper mapper, final ObjectNode lhs, final ObjectNode rhs) {
		final ObjectNode result = mapper.createObjectNode();

		final Iterator> liter = lhs.fields();
		while (liter.hasNext()) {
			final Entry e = liter.next();
			result.set(e.getKey(), e.getValue());
		}

		final Iterator> riter = rhs.fields();
		while (riter.hasNext()) {
			final Entry e = riter.next();
			final JsonNode l = result.get(e.getKey());
			final JsonNode r = e.getValue();

			JsonNode resolved = r;
			if (l != null && l.isObject() && r.isObject())
				resolved = mergeRecursive(mapper, (ObjectNode) l, (ObjectNode) r);
			result.set(e.getKey(), resolved);
		}
		return result;
	}

	@Override
	public String image() {
		return "*";
	}
}