org.opentripplanner.routing.graph.kryosupport.JavaImmutableSetSerializer Maven / Gradle / Ivy

Go to download

package org.opentripplanner.routing.graph.kryosupport;

import com.esotericsoftware.kryo.Kryo;
import com.esotericsoftware.kryo.Serializer;
import com.esotericsoftware.kryo.io.Input;
import com.esotericsoftware.kryo.io.Output;
import java.io.Serializable;
import java.util.Set;

@SuppressWarnings("rawtypes")
class JavaImmutableSetSerializer extends Serializer {
    @Override
    public void write(Kryo kryo, Output output, Set set) {
        kryo.writeObject(output, new ImmSerList(set.toArray()));
    }

    @Override
    public Set read(Kryo kryo, Input input, Class type) {
        return kryo.readObject(input, ImmSerList.class).toSet();
    }

    private static class ImmSerList implements Serializable  {
        private static final long serialVersionUID = 1L;
        private final Object[] array;
        private ImmSerList(Object[] array) { this.array = array; }
        private Set toSet() { return Set.of(array); }
    }
}