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

Go to download
/*
 * Copyright 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.core.Version;
import com.fasterxml.jackson.databind.DeserializationContext;
import com.fasterxml.jackson.databind.JsonDeserializer;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.fasterxml.jackson.databind.deser.std.StdDeserializer;
import com.fasterxml.jackson.databind.module.SimpleModule;
import com.fasterxml.jackson.databind.ser.std.StdSerializer;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import java.io.IOException;
import java.util.List;
import java.util.Map;

/** Serializers and deserializers for JSR 310 {@code java.time.*}. */
@IgnoreJDK6Requirement
@SuppressWarnings("Since15")
class GraphSON1JavaTimeModule extends SimpleModule {

  GraphSON1JavaTimeModule(String name, Version version) {
    super(name, version, createDeserializers(), createSerializers());
  }

  private static Map, JsonDeserializer> createDeserializers() {

    return ImmutableMap., JsonDeserializer>builder()
        .put(java.time.Duration.class, new DurationJacksonDeserializer())
        .put(java.time.Instant.class, new InstantJacksonDeserializer())
        .put(java.time.LocalDate.class, new LocalDateJacksonDeserializer())
        .put(java.time.LocalTime.class, new LocalTimeJacksonDeserializer())
        .build();
  }

  private static List> createSerializers() {
    return ImmutableList.>builder()
        .add(new DurationJacksonSerializer())
        .add(new InstantJacksonSerializer())
        .add(new LocalDateJacksonSerializer())
        .add(new LocalTimeJacksonSerializer())
        .build();
  }

  /** Base class for serializing the {@code java.time.*} types to ISO-8061 formats. */
  @IgnoreJDK6Requirement
  abstract static 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());
    }
  }

  /** Base class for deserializing the {@code java.time.*} types from ISO-8061 formats. */
  @IgnoreJDK6Requirement
  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());
    }
  }

  @IgnoreJDK6Requirement
  static final class DurationJacksonSerializer
      extends AbstractJavaTimeSerializer {

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

  @IgnoreJDK6Requirement
  static final 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
  static final class InstantJacksonSerializer
      extends AbstractJavaTimeSerializer {

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

  @IgnoreJDK6Requirement
  static final 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
  static final class LocalDateJacksonSerializer
      extends AbstractJavaTimeSerializer {

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

  @IgnoreJDK6Requirement
  static final 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
  static final class LocalTimeJacksonSerializer
      extends AbstractJavaTimeSerializer {

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

  @IgnoreJDK6Requirement
  static final 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);
    }
  }
}