org.apache.tinkerpop.shaded.kryo.Serializer Java Examples

/**
 * Creates a {@code GryoMapper}.
 */
public GryoMapper create() {
    // consult the registry if provided and inject registry entries as custom classes.
    registries.forEach(registry -> {
        final List<Pair<Class, Object>> serializers = registry.find(GryoIo.class);
        serializers.forEach(p -> {
            if (null == p.getValue1())
                addCustom(p.getValue0());
            else if (p.getValue1() instanceof SerializerShim)
                addCustom(p.getValue0(), new ShadedSerializerAdapter((SerializerShim) p.getValue1()));
            else if (p.getValue1() instanceof Serializer)
                addCustom(p.getValue0(), (Serializer) p.getValue1());
            else if (p.getValue1() instanceof Function)
                addCustom(p.getValue0(), (Function<Kryo, Serializer>) p.getValue1());
            else
                throw new IllegalStateException(String.format(
                        "Unexpected value provided by %s for serializable class %s - expected a parameter in [null, %s implementation or Function<%s, %s>], but received %s",
                        registry.getClass().getSimpleName(), p.getValue0().getClass().getCanonicalName(),
                        Serializer.class.getName(), Kryo.class.getSimpleName(),
                        Serializer.class.getSimpleName(), p.getValue1()));
        });
    });

    return new GryoMapper(this);
}
 

private GryoTypeReg(final Class<T> clazz,
                    final Serializer<T> shadedSerializer,
                    final SerializerShim<T> serializerShim,
                    final Function<Kryo, Serializer> functionOfShadedKryo,
                    final int id) {
    if (null == clazz) throw new IllegalArgumentException("clazz cannot be null");

    this.clazz = clazz;
    this.shadedSerializer = shadedSerializer;
    this.serializerShim = serializerShim;
    this.functionOfShadedKryo = functionOfShadedKryo;
    this.id = id;

    int serializerCount = 0;
    if (null != this.shadedSerializer)
        serializerCount++;
    if (null != this.serializerShim)
        serializerCount++;
    if (null != this.functionOfShadedKryo)
        serializerCount++;

    if (1 < serializerCount) {
        final String msg = String.format(
                "GryoTypeReg accepts at most one kind of serializer, but multiple " +
                        "serializers were supplied for class %s (id %s).  " +
                        "Shaded serializer: %s.  Shim serializer: %s.  Shaded serializer function: %s.",
                this.clazz.getCanonicalName(), id,
                this.shadedSerializer, this.serializerShim, this.functionOfShadedKryo);
        throw new IllegalArgumentException(msg);
    }
}
 

private GryoTypeReg(final Class<T> clazz,
                    final Serializer<T> shadedSerializer,
                    final SerializerShim<T> serializerShim,
                    final Function<Kryo, Serializer> functionOfShadedKryo,
                    final int id) {
    this.clazz = clazz;
    this.shadedSerializer = shadedSerializer;
    this.serializerShim = serializerShim;
    this.functionOfShadedKryo = functionOfShadedKryo;
    this.id = id;

    int serializerCount = 0;
    if (null != this.shadedSerializer)
        serializerCount++;
    if (null != this.serializerShim)
        serializerCount++;
    if (null != this.functionOfShadedKryo)
        serializerCount++;

    if (1 < serializerCount) {
        final String msg = String.format(
                "GryoTypeReg accepts at most one kind of serializer, but multiple " +
                        "serializers were supplied for class %s (id %s).  " +
                        "Shaded serializer: %s.  Shim serializer: %s.  Shaded serializer function: %s.",
                this.clazz.getCanonicalName(), id,
                this.shadedSerializer, this.serializerShim, this.functionOfShadedKryo);
        throw new IllegalArgumentException(msg);
    }
}
 

static <T> GryoTypeReg<T> of(final Class<T> clazz, final int id, final Serializer<T> shadedSerializer) {
    return new GryoTypeReg<>(clazz, shadedSerializer, null, null, id);
}
 

static <T> GryoTypeReg<T> of(final Class clazz, final int id, final Function<Kryo, Serializer> fct) {
    return new GryoTypeReg<>(clazz, null, null, fct, id);
}
 

@Override
public Serializer<T> getShadedSerializer() {
    return shadedSerializer;
}
 

@Override
public Function<Kryo, Serializer> getFunctionOfShadedKryo() {
    return functionOfShadedKryo;
}
 

/**
 * Register custom class to serialize with a custom serialization class. Note that calling this method for
 * a class that is already registered will override that registration.
 */
public Builder addCustom(final Class clazz, final Serializer serializer) {
    addOrOverrideRegistration(clazz, id -> GryoTypeReg.of(clazz, id, serializer));
    return this;
}
 

/**
 * Register a custom class to serialize with a custom serializer as returned from a {@link Function}. Note
 * that calling this method for a class that is already registered will override that registration.
 */
public Builder addCustom(final Class clazz, final Function<Kryo, Serializer> functionOfKryo) {
    addOrOverrideRegistration(clazz, id -> GryoTypeReg.of(clazz, id, functionOfKryo));
    return this;
}
 

private static <T> GryoTypeReg<T> of(final Class<T> clazz, final int id, final Serializer<T> shadedSerializer) {
    return new GryoTypeReg<>(clazz, shadedSerializer, null, null, id);
}
 

private static <T> GryoTypeReg<T> of(final Class clazz, final int id, final Function<Kryo, Serializer> fct) {
    return new GryoTypeReg<>(clazz, null, null, fct, id);
}
 

@Override
public Serializer<T> getShadedSerializer() {
    return shadedSerializer;
}
 

@Override
public Function<Kryo, Serializer> getFunctionOfShadedKryo() {
    return functionOfShadedKryo;
}
 

/**
 * @return the shaded-Kryo serializer that handles this type, if one is defined
 */
Serializer<T> getShadedSerializer();
 

/**
 * @return a function that accepts a shaded-Kryo instance and returns a serializer, if such a function is defined
 */
java.util.function.Function<Kryo, Serializer> getFunctionOfShadedKryo();