com.github.flysium.io.yew.common.serialize.kryo.KryoObjectOutput Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yew-serialization-kryo Show documentation
There is a newer version: 1.0.4
Show newest version
/*
 * Copyright 2018-2025 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.github.flysium.io.yew.common.serialize.kryo;

import com.esotericsoftware.kryo.Kryo;
import com.esotericsoftware.kryo.KryoException;
import com.esotericsoftware.kryo.io.Output;
import com.github.flysium.io.yew.common.serialize.Cleanable;
import com.github.flysium.io.yew.common.serialize.ObjectOutput;
import com.github.flysium.io.yew.common.serialize.kryo.utils.KryoUtils;
import java.io.IOException;
import java.io.OutputStream;

/**
 * Kryo Object Output
 *
 * @author Sven Augustus
 * @version 1.0
 */
public class KryoObjectOutput implements ObjectOutput, Cleanable {

	private final Output output;
	private Kryo kryo;

	public KryoObjectOutput(OutputStream outputStream) {
		this.output = new Output(outputStream);
		this.kryo = KryoUtils.get();
	}

	@Override
	public void writeObject(Object obj) throws IOException {
		try {
			this.kryo.writeClassAndObject(output, obj);
		} catch (KryoException e) {
			throw new IOException(e);
		}
	}

	@Override
	public void writeBoolean(boolean v) throws IOException {
		output.writeBoolean(v);
	}

	@Override
	public void writeByte(byte v) throws IOException {
		output.writeByte(v);
	}

	@Override
	public void writeShort(short v) throws IOException {
		output.writeShort(v);
	}

	@Override
	public void writeInt(int v) throws IOException {
		output.writeInt(v);
	}

	@Override
	public void writeLong(long v) throws IOException {
		output.writeLong(v);
	}

	@Override
	public void writeFloat(float v) throws IOException {
		output.writeFloat(v);
	}

	@Override
	public void writeDouble(double v) throws IOException {
		output.writeDouble(v);
	}

	@Override
	public void writeUTF(String v) throws IOException {
		output.writeString(v);
	}


	@Override
	public void writeBytes(byte[] v) throws IOException {
		if (v == null) {
			output.writeInt(-1);
		} else {
			writeBytes(v, 0, v.length);
		}
	}

	@Override
	public void writeBytes(byte[] v, int off, int len) throws IOException {
		if (v == null) {
			output.writeInt(-1);
		} else {
			output.writeInt(len);
			output.write(v, off, len);
		}
	}

	@Override
	public void flushBuffer() throws IOException {
		output.flush();
	}

	@Override
	public void cleanup() {
		KryoUtils.release(this.kryo);
		kryo = null;
	}
}