streams.codec.Kryo Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of streams-core Show documentation

The newest version!

/**
 * 
 */
package streams.codec;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.Serializable;
import java.util.HashMap;
import java.util.Map;

import com.esotericsoftware.kryo.io.Input;
import com.esotericsoftware.kryo.io.Output;

import stream.Data;
import stream.data.DataFactory;

/**
 * This class implements a data item codec using an internal Kryo instance. The
 * kryo instance is shared, so the calls to decode and encode are not
 * thread-safe.
 * 
 * @author Christian Bockermann
 *
 */
public class Kryo implements Codec {

    final HashMap template = new HashMap();
    final com.esotericsoftware.kryo.Kryo codec = new com.esotericsoftware.kryo.Kryo();

    public Kryo() {
    }

    /**
     * @see stream.io.Codec#decode(byte[])
     */
    @Override
    public Data decode(byte[] rawBytes) throws Exception {
        @SuppressWarnings("unchecked")
        Map values = codec.readObject(new Input(new ByteArrayInputStream(rawBytes)),
                template.getClass());
        Data item = DataFactory.create(values);
        return item;
    }

    /**
     * @see stream.io.Codec#encode(java.lang.Object)
     */
    @Override
    public byte[] encode(Data object) throws Exception {
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        Output out = new Output(baos);
        codec.writeObject(out, object);
        out.flush();
        out.close();
        return baos.toByteArray();
    }
}