CavalliumDBEngine/src/main/java/it/cavallium/dbengine/database/serialization/BufferDataOutput.java

package it.cavallium.dbengine.database.serialization;

import io.net5.buffer.api.Buffer;
import java.io.DataOutput;
import java.nio.charset.StandardCharsets;
import org.jetbrains.annotations.NotNull;

public class BufferDataOutput implements DataOutput {

	private final Buffer buf;

	public BufferDataOutput(Buffer bufferSend) {
		this.buf = bufferSend;
	}

	@Override
	public void write(int b) {
		buf.ensureWritable(Integer.BYTES);
		buf.writeUnsignedByte(b);
	}

	@Override
	public void write(byte @NotNull [] b) {
		buf.ensureWritable(Byte.BYTES * b.length);
		buf.writeBytes(b);
	}

	@Override
	public void write(byte @NotNull [] b, int off, int len) {
		buf.ensureWritable(len);
		buf.writeBytes(b, off, len);
	}

	@Override
	public void writeBoolean(boolean v) {
		buf.ensureWritable(Byte.BYTES);
		buf.writeUnsignedByte(v ? 1 : 0);
	}

	@Override
	public void writeByte(int v) {
		buf.ensureWritable(Byte.BYTES);
		buf.writeByte((byte) v);
	}

	@Override
	public void writeShort(int v) {
		buf.ensureWritable(Short.BYTES);
		buf.writeShort((short) v);
	}

	@Override
	public void writeChar(int v) {
		buf.ensureWritable(Character.BYTES);
		buf.writeChar((char) v);
	}

	@Override
	public void writeInt(int v) {
		buf.ensureWritable(Integer.BYTES);
		buf.writeInt(v);
	}

	@Override
	public void writeLong(long v) {
		buf.ensureWritable(Long.BYTES);
		buf.writeLong(v);
	}

	@Override
	public void writeFloat(float v) {
		buf.ensureWritable(Float.BYTES);
		buf.writeFloat(v);
	}

	@Override
	public void writeDouble(double v) {
		buf.ensureWritable(Double.BYTES);
		buf.writeDouble(v);
	}

	@Override
	public void writeBytes(@NotNull String s) {
		var b= s.getBytes(StandardCharsets.UTF_8);
		buf.ensureWritable(Byte.BYTES * b.length);
		buf.writeBytes(b);
	}

	@Override
	public void writeChars(@NotNull String s) {
		var chars = s.toCharArray();
		buf.ensureWritable(Character.BYTES * chars.length);
		for (char c : chars) {
			buf.writeChar(c);
		}
	}

	@Override
	public void writeUTF(@NotNull String s) {
		int sizeShortOffset = buf.writerOffset();
		buf.ensureWritable(Short.BYTES + 1);
		int stringOffset = sizeShortOffset + Short.BYTES;
		buf.writerOffset(stringOffset);
		// todo: replace with writeCharSequence when it will be optimized in netty 5
		{
			byte[] bytes = s.getBytes(StandardCharsets.UTF_8);
			buf.ensureWritable(bytes.length);
			buf.writeBytes(bytes);
		}
		int endOffset = buf.writerOffset();
		int stringSize = endOffset - stringOffset;
		buf.writerOffset(sizeShortOffset);
		buf.writeUnsignedShort(stringSize);
		if (stringSize > (1 << 16) - 1) {
			buf.writerOffset(sizeShortOffset);
			throw new IndexOutOfBoundsException("String too large: " + stringSize);
		} else {
			buf.writerOffset(endOffset);
		}
	}
}