| package net.onrc.onos.core.datastore.utils; |
| |
| import net.onrc.onos.core.datastore.DataStoreClient; |
| |
| import com.esotericsoftware.kryo.io.Input; |
| import com.esotericsoftware.kryo.io.Output; |
| |
| /** |
| * {@link Serializer} implementation using Kryo. |
| */ |
| public final class KryoSerializer |
| implements Serializer { |
| |
| private final ThreadLocalKryo kryo; |
| |
| public KryoSerializer(Class<?>... expectedTypes) { |
| kryo = new ThreadLocalKryo(expectedTypes); |
| } |
| |
| @Override |
| public byte[] serialize(Object obj) { |
| // 1MB RAMCloud limit |
| Output out = new Output(DataStoreClient.MAX_VALUE_BYTES); |
| kryo.get().writeClassAndObject(out, obj); |
| return out.toBytes(); |
| } |
| |
| @Override |
| public <T> T deserialize(byte[] bytes) { |
| Input in = new Input(bytes); |
| @SuppressWarnings("unchecked") |
| T obj = (T) kryo.get().readClassAndObject(in); |
| return obj; |
| } |
| } |