A Protobuffer implementation of com.esotericsoftware.kryo.Serializer

PBSerialize.java

public abstract class PBSerialize<T extends Message> extends
        Serializer {

    protected abstract T parseForm(CodedInputStream in);

    private final ThreadLocal<byte[]> thread_local_buffer = new ThreadLocal<byte[]>() {
        @Override
        protected byte[] initialValue() {
            return new byte[1024 * 10]; // 10 KB
        }
    };

    @SuppressWarnings("unchecked")
    @Override
    public <U> U readObjectData(ByteBuffer bb, Class<U> type) {
        int len = bb.getInt();
        byte[] tempBuf = thread_local_buffer.get();
        if (tempBuf.length < len) {
            tempBuf = new byte[len];
            thread_local_buffer.set(tempBuf);
        }
        bb.get(tempBuf, 0, len);
        return (U) parseForm(CodedInputStream.newInstance(tempBuf, 0, len));
    }

    @Override
    public void writeObjectData(ByteBuffer bb, Object obj) {
        Message message = (Message) obj;
        byte[] tempBuf = thread_local_buffer.get();
        int len = message.getSerializedSize();
        if (tempBuf.length < len) {
            tempBuf = new byte[len];
            thread_local_buffer.set(tempBuf);
        }
        CodedOutputStream codedOut = CodedOutputStream.newInstance(tempBuf);
        try {
            message.writeTo(codedOut);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        bb.putInt(message.getSerializedSize());
        bb.put(tempBuf, 0, len);
    }

}