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

Go to download
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.Multimap;
import java.util.Iterator;
import java.util.Map;

public abstract class MultimapSerializerBase> extends Serializer {

    public MultimapSerializerBase(boolean acceptsNull, boolean immutable) {
        super(acceptsNull, immutable);
    }

    protected void writeMultimap(Kryo kryo, Output output, Multimap multimap) {
        output.writeInt(multimap.size(), true);
        for (final Map.Entry entry : multimap.entries()) {
            kryo.writeClassAndObject(output, entry.getKey());
            kryo.writeClassAndObject(output, entry.getValue());
        }
    }

    protected void readMultimap(Kryo kryo, Input input, Multimap multimap) {
        final int size = input.readInt(true);
        for (int i = 0; i < size; ++i) {
            final K key = (K) kryo.readClassAndObject(input);
            final V value = (V) kryo.readClassAndObject(input);
            multimap.put(key, value);
        }
    }

    protected abstract Multimap createCopy(Kryo kryo, Multimap original);

    @Override
    public Multimap copy(final Kryo kryo, final Multimap original) {
        Multimap copy = createCopy(kryo, original);
        Iterator iter = original.entries().iterator();

        while(iter.hasNext()) {
            Map.Entry entry = iter.next();
            copy.put(kryo.copy(entry.getKey()), kryo.copy(entry.getValue()));
        }

        return copy;
    }
}