de.javakaffee.kryoserializers.guava.TreeBasedTableSerializer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kryo-serializers Show documentation

Additional kryo (http://kryo.googlecode.com) serializers for standard jdk types (e.g. currency, jdk proxies) and some for external libs (e.g. joda time, cglib proxies, wicket).

There is a newer version: 0.45

Show newest version

package de.javakaffee.kryoserializers.guava;

import com.esotericsoftware.kryo.Kryo;
import com.esotericsoftware.kryo.Serializer;
import com.esotericsoftware.kryo.io.Input;
import com.esotericsoftware.kryo.io.Output;
import com.google.common.collect.HashBasedTable;
import com.google.common.collect.ImmutableTable;
import com.google.common.collect.Table;
import com.google.common.collect.TreeBasedTable;

import java.util.Comparator;
import java.util.Set;

/**
 * A kryo {@link Serializer} for guava-libraries {@link TreeBasedTable}.
 */
public class TreeBasedTableSerializer extends TableSerializerBase> {

    private static final boolean HANDLES_NULL = false;
    private static final boolean IMMUTABLE = false;

    public TreeBasedTableSerializer() {
        super(HANDLES_NULL, IMMUTABLE);
    }

    @Override
    public void write(Kryo kryo, Output output, TreeBasedTable table) {
        kryo.writeClassAndObject(output, table.rowComparator());
        kryo.writeClassAndObject(output, table.columnComparator());
        super.writeTable(kryo, output, table);
    }

    @Override
    public TreeBasedTable read(Kryo kryo, Input input, Class> type) {
        Comparator rowComparator = (Comparator) kryo.readClassAndObject(input);
        Comparator columnComparator = (Comparator) kryo.readClassAndObject(input);
        TreeBasedTable table = TreeBasedTable.create(rowComparator, columnComparator);
        super.readTable(kryo, input, table);
        return table;
    }

    @Override
    public TreeBasedTable copy(final Kryo kryo, final TreeBasedTable original) {
        return TreeBasedTable.create(original);
    }

    /**
     * Creates a new {@link TreeBasedTableSerializer} and registers its serializer.
     *
     * @param kryo the {@link Kryo} instance to set the serializer on
     */
    public static void registerSerializers(final Kryo kryo) {
        final TreeBasedTableSerializer serializer = new TreeBasedTableSerializer();
        kryo.register(TreeBasedTable.class, serializer);
    }
}