org.modelmapper.module.jsr310.FromTemporalConverter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of modelmapper-module-jsr310 Show documentation
ModelMapper Module for JSR310
The newest version!
package org.modelmapper.module.jsr310;

import java.math.BigDecimal;
import java.math.BigInteger;
import java.time.*;
import java.time.temporal.Temporal;
import java.util.Calendar;
import java.util.Date;
import org.modelmapper.Converter;
import org.modelmapper.internal.Errors;
import org.modelmapper.spi.ConditionalConverter;
import org.modelmapper.spi.MappingContext;

/**
 * Converts  {@link Temporal} to {@link Object}
 *
 * @author Chun Han Hsiao
 */
public class FromTemporalConverter implements ConditionalConverter {
  private Jsr310ModuleConfig config;
  private final LocalDateTimeConverter localDateTimeConverter = new LocalDateTimeConverter();
  private final ZonedDateTimeConverter zonedDateTimeConverter = new ZonedDateTimeConverter();
  private final LocalDateConverter localDateConverter = new LocalDateConverter();
  private final OffsetDateTimeConverter offsetDateTimeConverter = new OffsetDateTimeConverter();
  private final InstantConverter instantConverter = new InstantConverter();

  public FromTemporalConverter(Jsr310ModuleConfig config) {
    this.config = config;
  }

  @Override
  public MatchResult match(Class sourceType, Class destinationType) {
    return Temporal.class.isAssignableFrom(sourceType)
        ? MatchResult.FULL : MatchResult.NONE;
  }

  @Override
  public Object convert(MappingContext mappingContext) {
    if (mappingContext.getSource() == null)
      return null;

    Class sourceType = mappingContext.getSourceType();
    if (LocalDateTime.class.equals(sourceType))
      return localDateTimeConverter.convert(mappingContext);
    else if (LocalDate.class.equals(sourceType))
      return localDateConverter.convert(mappingContext);
    else if (OffsetDateTime.class.equals(sourceType))
      return offsetDateTimeConverter.convert(mappingContext);
    else if (Instant.class.equals(sourceType))
      return instantConverter.convert(mappingContext);
    else if (ZonedDateTime.class.equals(sourceType))
      return zonedDateTimeConverter.convert(mappingContext);
    else
      throw new Errors().addMessage("Unsupported mapping types[%s->%s]",
          LocalDateTime.class.getName(), mappingContext.getDestinationType().getName())
          .toMappingException();
  }

  private class LocalDateTimeConverter implements Converter {
    @Override
    public Object convert(MappingContext mappingContext) {
      LocalDateTime source = (LocalDateTime) mappingContext.getSource();
      return convertLocalDateTime(source, mappingContext);
    }
  }

  private class ZonedDateTimeConverter implements Converter {
    @Override
    public Object convert(MappingContext mappingContext) {
      ZonedDateTime source = (ZonedDateTime) mappingContext.getSource();
      return convertZonedDateTime(source, mappingContext);
    }
  }

  private class LocalDateConverter implements Converter {
    @Override
    public Object convert(MappingContext mappingContext) {
      LocalDate source = (LocalDate) mappingContext.getSource();
      Class destinationType = mappingContext.getDestinationType();
      if (destinationType.equals(String.class))
        return config.getDateFormatter()
            .format(source);

      LocalDateTime localDateTime = source.atStartOfDay();
      return convertLocalDateTime(localDateTime, mappingContext);
    }
  }

  private class OffsetDateTimeConverter implements Converter {
    @Override
    public Object convert(MappingContext mappingContext) {
      OffsetDateTime source = (OffsetDateTime) mappingContext.getSource();
      return convertOffsetDateTime(source, mappingContext);
    }
  }

  private class InstantConverter implements Converter {
    @Override
    public Object convert(MappingContext mappingContext) {
      Instant source = (Instant) mappingContext.getSource();
      return convertInstant(source, mappingContext);
    }
  }

  private Object convertLocalDateTime(LocalDateTime source, MappingContext mappingContext) {
    Class destinationType = mappingContext.getDestinationType();
    if (destinationType.equals(String.class))
      return config.getDateTimeFormatter()
          .format(source);

    Instant instant = source.atZone(config.getZoneId()).toInstant();
    return convertInstant(instant, mappingContext);
  }

  private Object convertZonedDateTime(ZonedDateTime source, MappingContext mappingContext) {
    Class destinationType = mappingContext.getDestinationType();
    if (destinationType.equals(String.class))
      return config.getDateTimeOffsetFormatter()
              .format(source);

    Instant instant = source.toInstant();
    return convertInstant(instant, mappingContext);
  }


  private Object convertOffsetDateTime(OffsetDateTime source, MappingContext mappingContext) {
    Class destinationType = mappingContext.getDestinationType();
    if (destinationType.equals(String.class))
      return config.getDateTimeOffsetFormatter()
              .format(source);

    Instant instant = source.toInstant();
    return convertInstant(instant, mappingContext);
  }

  private Object convertInstant(Instant source, MappingContext mappingContext) {
    Class destinationType = mappingContext.getDestinationType();
    if (destinationType.equals(String.class))
      return config.getDateTimeFormatter()
          .withZone(config.getZoneId())
          .format(source);
    else if (Date.class.isAssignableFrom(destinationType))
      return new Date(epochMilliOf(source));
    else if (Calendar.class.isAssignableFrom(destinationType))
      return calendarOf(source);
    else if (Long.class.equals(destinationType) || Long.TYPE.equals(destinationType))
      return epochMilliOf(source);
    else if (BigDecimal.class.equals(destinationType))
      return new BigDecimal(epochMilliOf(source));
    else if (BigInteger.class.equals(destinationType))
      return BigInteger.valueOf(epochMilliOf(source));
    else
      throw new Errors().addMessage("Unsupported mapping types[%s->%s]",
          mappingContext.getSourceType().getName(), destinationType.getName())
          .toMappingException();
  }

  private long epochMilliOf(Instant instant) {
    return instant.toEpochMilli();
  }

  private Calendar calendarOf(Instant instant) {
    Calendar calendar = Calendar.getInstance();
    calendar.setTimeInMillis(epochMilliOf(instant));
    return calendar;
  }
}