data-generator/data-generator-runtime/src/main/java/it/cavallium/data/generator/nativedata/NullableStringSerializer.java

package it.cavallium.data.generator.nativedata;

import it.cavallium.data.generator.DataSerializer;
import it.cavallium.stream.SafeDataInput;
import it.cavallium.stream.SafeDataOutput;
import org.jetbrains.annotations.NotNull;

import java.nio.charset.StandardCharsets;

public class NullableStringSerializer implements DataSerializer<NullableString> {

	public static final NullableStringSerializer INSTANCE = new NullableStringSerializer();

	@Override
	public void serialize(SafeDataOutput dataOutput, @NotNull NullableString data) {
		if (data.isEmpty()) {
			dataOutput.writeBoolean(false);
		} else {
			dataOutput.writeBoolean(true);
			String dataContent = data.get();
			dataOutput.writeUTF(dataContent);
		}
	}

	@NotNull
	@Override
	public NullableString deserialize(SafeDataInput dataInput) {
		var isPresent = dataInput.readBoolean();
		if (!isPresent) {
			return NullableString.empty();
		} else {
			return NullableString.of(dataInput.readShortText(StandardCharsets.UTF_8));
		}
	}
}