org.modelmapper.module.jsr310.ToTemporalConverter 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
There is a newer version: 1.2.1
Show newest version
package org.modelmapper.module.jsr310;

import java.time.Instant;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
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 Object} to {@link Temporal}
 *
 * @author Chun Han Hsiao
 */
public class ToTemporalConverter implements ConditionalConverter {
  private Jsr310ModuleConfig config;
  private final LocalDateTimeConverter localDateTimeConverter = new LocalDateTimeConverter();
  private final LocalDateConverter localDateConverter = new LocalDateConverter();
  private final InstantConverter instantConverter = new InstantConverter();

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

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

  @Override
  public Temporal convert(MappingContext mappingContext) {
    Class destinationType = mappingContext.getDestinationType();
    if (LocalDateTime.class.equals(destinationType))
      return localDateTimeConverter.convert(mappingContext);
    else if (LocalDate.class.equals(destinationType))
      return localDateConverter.convert(mappingContext);
    else if (Instant.class.equals(destinationType))
      return instantConverter.convert(mappingContext);
    else
      throw new Errors().addMessage("Unsupported mapping types[%s->%s]",
          mappingContext.getSourceType().getName(), destinationType.getName())
          .toMappingException();
  }

  private class LocalDateTimeConverter implements Converter {
    @Override
    public Temporal convert(MappingContext mappingContext) {
      return convertLocalDateTime(mappingContext);
    }
  }

  private class LocalDateConverter implements Converter {
    @Override
    public Temporal convert(MappingContext mappingContext) {
      return convertLocalDate(mappingContext);
    }
  }

  private class InstantConverter implements Converter {
    @Override
    public Temporal convert(MappingContext mappingContext) {
      return convertInstant(mappingContext);
    }
  }

  private LocalDate convertLocalDate(MappingContext mappingContext) {
    Object source = mappingContext.getSource();
    Class sourceType = source.getClass();
    if (sourceType.equals(String.class))
      return LocalDate.parse((String) source,
          DateTimeFormatter.ofPattern(config.getDatePattern()));
    return convertInstant(mappingContext).atZone(config.getZoneId()).toLocalDate();
  }

  private LocalDateTime convertLocalDateTime(MappingContext mappingContext) {
    Object source = mappingContext.getSource();
    Class sourceType = source.getClass();
    if (sourceType.equals(String.class))
      return LocalDateTime.parse((String) source,
          DateTimeFormatter.ofPattern(config.getDateTimePattern()));
    return convertInstant(mappingContext).atZone(config.getZoneId()).toLocalDateTime();
  }

  private Instant convertInstant(MappingContext mappingContext) {
    Object source = mappingContext.getSource();
    Class sourceType = source.getClass();
    if (sourceType.equals(String.class))
      return LocalDateTime.parse((String) source,
          DateTimeFormatter.ofPattern(config.getDateTimePattern()))
          .atZone(config.getZoneId()).toInstant();
    else if (Date.class.isAssignableFrom(sourceType))
      return Instant.ofEpochMilli(((Date) source).getTime());
    else if (Calendar.class.isAssignableFrom(sourceType))
      return Instant.ofEpochMilli(((Calendar) source).getTime().getTime());
    else if (Number.class.isAssignableFrom(sourceType))
      return Instant.ofEpochMilli(((Number) source).longValue());
    else
      throw new Errors().addMessage("Unsupported mapping types[%s->%s]",
          sourceType.getName(), mappingContext.getDestinationType().getName())
          .toMappingException();
  }
}