com.dslplatform.json.runtime.MapEncoder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dsl-json-java8 Show documentation
DSL Platform compatible Java JSON library (https://dsl-platform.com)
There is a newer version: 1.10.0
package com.dslplatform.json.runtime;

import com.dslplatform.json.DslJson;
import com.dslplatform.json.JsonWriter;
import com.dslplatform.json.NumberConverter;
import com.dslplatform.json.SerializationException;

import java.util.Map;

public final class MapEncoder> implements JsonWriter.WriteObject {

	private final DslJson json;
	private final boolean checkForConversionToString;
	private final JsonWriter.WriteObject keyEncoder;
	private final JsonWriter.WriteObject valueEncoder;

	public MapEncoder(
			final DslJson json,
			final boolean checkForConversionToString,
			final JsonWriter.WriteObject keyEncoder,
			final JsonWriter.WriteObject valueEncoder) {
		if (json == null) throw new IllegalArgumentException("json can't be null");
		this.json = json;
		this.checkForConversionToString = checkForConversionToString;
		this.keyEncoder = keyEncoder;
		this.valueEncoder = valueEncoder;
	}

	private static final byte[] EMPTY = {'{', '}'};

	@Override
	public void write(JsonWriter writer, T value) {
		if (value == null) writer.writeNull();
		else if (value.isEmpty()) writer.writeAscii(EMPTY);
		else if (keyEncoder != null && valueEncoder != null) {
			boolean pastFirst = false;
			writer.writeByte(JsonWriter.OBJECT_START);
			for (final Map.Entry e : value.entrySet()) {
				if (pastFirst) {
					writer.writeByte(JsonWriter.COMMA);
				} else {
					pastFirst = true;
				}
				if (checkForConversionToString) {
					writeQuoted(writer, keyEncoder, e.getKey());
				} else keyEncoder.write(writer, e.getKey());
				writer.writeByte(JsonWriter.SEMI);
				valueEncoder.write(writer, e.getValue());
			}
			writer.writeByte(JsonWriter.OBJECT_END);
		} else {
			boolean pastFirst = false;
			writer.writeByte(JsonWriter.OBJECT_START);
			Class lastKeyClass = null;
			Class lastValueClass = null;
			JsonWriter.WriteObject lastKeyEncoder = keyEncoder;
			JsonWriter.WriteObject lastValueEncoder = null;
			for (final Map.Entry e : value.entrySet()) {
				if (pastFirst) {
					writer.writeByte(JsonWriter.COMMA);
				} else {
					pastFirst = true;
				}
				final Class currentKeyClass = e.getKey().getClass();
				if (keyEncoder == null && currentKeyClass != lastKeyClass) {
					lastKeyClass = currentKeyClass;
					lastKeyEncoder = json.tryFindWriter(lastKeyClass);
					if (lastKeyEncoder == null) {
						throw new SerializationException("Unable to find writer for " + lastKeyClass);
					}
				}
				writeQuoted(writer, lastKeyEncoder, e.getKey());
				writer.writeByte(JsonWriter.SEMI);
				if (valueEncoder != null) {
					valueEncoder.write(writer, e.getValue());
				} else {
					if (e.getValue() == null) {
						writer.writeNull();
					} else {
						final Class currentValueClass = e.getValue().getClass();
						if (currentValueClass != lastValueClass) {
							lastValueClass = currentValueClass;
							lastValueEncoder = json.tryFindWriter(lastValueClass);
							if (lastValueEncoder == null)
								throw new SerializationException("Unable to find writer for " + lastValueClass);
						}
						lastValueEncoder.write(writer, e.getValue());
					}
				}
			}
			writer.writeByte(JsonWriter.OBJECT_END);
		}
	}

	private void writeQuoted(final JsonWriter writer, final JsonWriter.WriteObject keyWriter, final K key) {
		if (key instanceof Double) {
			final double value = (Double) key;
			if (value == Double.NaN) writer.writeAscii("NaN");
			else if (value == Double.POSITIVE_INFINITY) writer.writeAscii("Infinity");
			else if (value == Double.NEGATIVE_INFINITY) writer.writeAscii("-Infinity");
			else {
				writer.writeByte(JsonWriter.QUOTE);
				NumberConverter.serialize(value, writer);
				writer.writeByte(JsonWriter.QUOTE);
			}
		} else if (key instanceof Float) {
			final float value = (Float) key;
			if (value == Float.NaN) writer.writeAscii("NaN");
			else if (value == Float.POSITIVE_INFINITY) writer.writeAscii("Infinity");
			else if (value == Float.NEGATIVE_INFINITY) writer.writeAscii("-Infinity");
			else {
				writer.writeByte(JsonWriter.QUOTE);
				NumberConverter.serialize(value, writer);
				writer.writeByte(JsonWriter.QUOTE);
			}
		} else if (key instanceof Number) {
			writer.writeByte(JsonWriter.QUOTE);
			keyWriter.write(writer, key);
			writer.writeByte(JsonWriter.QUOTE);
		} else {
			keyWriter.write(writer, key);
		}
	}
}