io.datakernel.codec.json.JsonStructuredOutput Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of datakernel-codec Show documentation
Tools for encoding/decoding of primitives and objects.
There is a newer version: 3.1.0
package io.datakernel.codec.json;

import com.google.gson.stream.JsonWriter;
import io.datakernel.codec.StructuredCodecs;
import io.datakernel.codec.StructuredEncoder;
import io.datakernel.codec.StructuredOutput;

import java.io.IOException;
import java.lang.reflect.Type;
import java.util.Base64;
import java.util.List;
import java.util.Map;

public class JsonStructuredOutput implements StructuredOutput {
	final JsonWriter writer;

	public JsonStructuredOutput(JsonWriter writer) {
		this.writer = writer;
	}

	@Override
	public void writeBoolean(boolean value) {
		try {
			writer.value(value);
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public void writeByte(byte value) {
		try {
			writer.value(value & 0xFF);
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public void writeInt(int value) {
		try {
			writer.value(value);
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public void writeLong(long value) {
		try {
			writer.value(value);
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public void writeInt32(int value) {
		try {
			writer.value(value);
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public void writeLong64(long value) {
		try {
			writer.value(value);
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public void writeFloat(float value) {
		try {
			writer.value(value);
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public void writeDouble(double value) {
		try {
			writer.value(value);
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public void writeBytes(byte[] bytes, int off, int len) {
		// TODO
	}

	@Override
	public void writeBytes(byte[] bytes) {
		try {
			writer.value(Base64.getEncoder().encodeToString(bytes));
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public void writeString(String value) {
		try {
			writer.value(value);
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public void writeNull() {
		try {
			writer.nullValue();
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public  void writeNullable(StructuredEncoder encoder, T value) {
		if (value != null) {
			encoder.encode(this, value);
		} else {
			writeNull();
		}
	}

	@Override
	public  void writeList(StructuredEncoder encoder, List list) {
		try {
			writer.beginArray();
			for (T item : list) {
				encoder.encode(this, item);
			}
			writer.endArray();
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public  void writeMap(StructuredEncoder keyEncoder, StructuredEncoder valueEncoder, Map map) {
		try {
			if (keyEncoder == StructuredCodecs.STRING_CODEC) {
				writer.beginObject();
				for (Map.Entry entry : map.entrySet()) {
					writer.name((String) entry.getKey());
					valueEncoder.encode(this, entry.getValue());
				}
				writer.endObject();
			} else {
				writer.beginArray();
				for (Map.Entry entry : map.entrySet()) {
					writer.beginArray();
					keyEncoder.encode(this, entry.getKey());
					valueEncoder.encode(this, entry.getValue());
					writer.endArray();
				}
				writer.endArray();
			}
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public  void writeTuple(StructuredEncoder encoder, T value) {
		try {
			writer.beginArray();
			encoder.encode(this, value);
			writer.endArray();
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public  void writeObject(StructuredEncoder encoder, T value) {
		try {
			writer.beginObject();
			encoder.encode(this, value);
			writer.endObject();
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public void writeKey(String field) {
		try {
			writer.name(field);
		} catch (IOException ignored) {
			throw new AssertionError();
		}
	}

	@Override
	public  void writeCustom(Type type, T value) {
		throw new UnsupportedOperationException("No custom type writers");
	}
}