com.datastax.driver.dse.graph.GraphSON2JavaTimeModule Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dse-java-driver-core Show documentation
A driver for DataStax Enterprise (DSE) and Apache Cassandra 1.2+ clusters that works exclusively with the Cassandra Query Language version 3 (CQL3) and Cassandra's binary protocol, supporting DSE-specific features such as geospatial types, DSE Graph and DSE authentication.
There is a newer version: 2.4.0
Show newest version
/*
 * Copyright (C) 2012-2017 DataStax Inc.
 *
 * This software can be used solely with DataStax Enterprise. Please consult the license at
 * http://www.datastax.com/terms/datastax-dse-driver-license-terms
 */
package com.datastax.driver.dse.graph;

import com.datastax.driver.dse.IgnoreJDK6Requirement;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.databind.DeserializationContext;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.deser.std.StdDeserializer;
import com.fasterxml.jackson.databind.jsontype.TypeSerializer;
import com.fasterxml.jackson.databind.ser.std.StdSerializer;
import com.google.common.collect.ImmutableMap;

import java.io.IOException;
import java.util.Map;

@IgnoreJDK6Requirement
@SuppressWarnings("Since15")
class GraphSON2JavaTimeModule extends GraphSON2JacksonModule {

    GraphSON2JavaTimeModule() {
        super("graph-graphson2time");

        addSerializer(java.time.Duration.class, new DurationJacksonSerializer());
        addSerializer(java.time.Instant.class, new InstantJacksonSerializer());
        addSerializer(java.time.LocalDate.class, new LocalDateJacksonSerializer());
        addSerializer(java.time.LocalTime.class, new LocalTimeJacksonSerializer());

        addDeserializer(java.time.Duration.class, new DurationJacksonDeserializer());
        addDeserializer(java.time.Instant.class, new InstantJacksonDeserializer());
        addDeserializer(java.time.LocalDate.class, new LocalDateJacksonDeserializer());
        addDeserializer(java.time.LocalTime.class, new LocalTimeJacksonDeserializer());
    }

    @Override
    public Map, String> getTypeDefinitions() {
        final ImmutableMap.Builder, String> builder = ImmutableMap.builder();

        builder.put(java.time.Instant.class, "Instant");
        builder.put(java.time.Duration.class, "Duration");
        builder.put(java.time.LocalDate.class, "LocalDate");
        builder.put(java.time.LocalTime.class, "LocalTime");

        return builder.build();
    }

    @Override
    public String getTypeNamespace() {
        return "gx";
    }

    /**
     * Base class for serializing the {@code java.time.*} from ISO-8061 formats.
     */
    abstract static class AbstractJavaTimeJacksonDeserializer extends StdDeserializer {
        AbstractJavaTimeJacksonDeserializer(final Class clazz) {
            super(clazz);
        }

        abstract T parse(final String val);

        @Override
        public T deserialize(final JsonParser jsonParser, final DeserializationContext deserializationContext) throws IOException {
            return parse(jsonParser.getText());
        }
    }

    /**
     * Base class for serializing the {@code java.time.*} to ISO-8061 formats.
     */
    static abstract class AbstractJavaTimeSerializer extends StdSerializer {

        AbstractJavaTimeSerializer(final Class clazz) {
            super(clazz);
        }

        @Override
        public void serialize(final T value, final JsonGenerator gen,
                              final SerializerProvider serializerProvider) throws IOException {
            gen.writeString(value.toString());
        }

        @Override
        public void serializeWithType(final T value, final JsonGenerator gen,
                                      final SerializerProvider serializers, final TypeSerializer typeSer) throws IOException {
            typeSer.writeTypePrefixForScalar(value, gen);
            gen.writeString(value.toString());
            typeSer.writeTypeSuffixForScalar(value, gen);
        }
    }

    @IgnoreJDK6Requirement
    final static class DurationJacksonSerializer extends AbstractJavaTimeSerializer {

        DurationJacksonSerializer() {
            super(java.time.Duration.class);
        }
    }

    @IgnoreJDK6Requirement
    final static class DurationJacksonDeserializer extends AbstractJavaTimeJacksonDeserializer {

        DurationJacksonDeserializer() {
            super(java.time.Duration.class);
        }

        @Override
        public java.time.Duration parse(final String val) {
            return java.time.Duration.parse(val);
        }
    }

    @IgnoreJDK6Requirement
    final static class InstantJacksonSerializer extends AbstractJavaTimeSerializer {

        InstantJacksonSerializer() {
            super(java.time.Instant.class);
        }
    }

    @IgnoreJDK6Requirement
    final static class InstantJacksonDeserializer extends AbstractJavaTimeJacksonDeserializer {

        InstantJacksonDeserializer() {
            super(java.time.Instant.class);
        }

        @Override
        public java.time.Instant parse(final String val) {
            return java.time.Instant.parse(val);
        }
    }

    @IgnoreJDK6Requirement
    final static class LocalDateJacksonSerializer extends AbstractJavaTimeSerializer {

        LocalDateJacksonSerializer() {
            super(java.time.LocalDate.class);
        }
    }

    @IgnoreJDK6Requirement
    final static class LocalDateJacksonDeserializer extends AbstractJavaTimeJacksonDeserializer {

        LocalDateJacksonDeserializer() {
            super(java.time.LocalDate.class);
        }

        @Override
        public java.time.LocalDate parse(final String val) {
            return java.time.LocalDate.parse(val);
        }
    }

    @IgnoreJDK6Requirement
    final static class LocalTimeJacksonSerializer extends AbstractJavaTimeSerializer {

        LocalTimeJacksonSerializer() {
            super(java.time.LocalTime.class);
        }
    }

    @IgnoreJDK6Requirement
    final static class LocalTimeJacksonDeserializer extends AbstractJavaTimeJacksonDeserializer {

        LocalTimeJacksonDeserializer() {
            super(java.time.LocalTime.class);
        }

        @Override
        public java.time.LocalTime parse(final String val) {
            return java.time.LocalTime.parse(val);
        }
    }




}