1 /***********************************************************************
2  * Copyright (c) 2013-2024 Commonwealth Computer Research, Inc.
3  * All rights reserved. This program and the accompanying materials
4  * are made available under the terms of the Apache License, Version 2.0
5  * which accompanies this distribution and is available at
6  * http://www.opensource.org/licenses/apache2.0.php.
7  ***********************************************************************/
8 
9 package org.locationtech.geomesa.utils.geotools.converters
10 
11 import org.geotools.util.factory.Hints
12 import org.geotools.util.{Converter, ConverterFactory}
13 import org.locationtech.geomesa.utils.text.DateParsing
14 
15 import java.time._
16 import java.time.temporal.{ChronoField, TemporalAccessor}
17 import java.util.Date
18 
19 class JavaTimeConverterFactory extends ConverterFactory {
20   def createConverter(source: Class[_], target: Class[_], hints: Hints): Converter = {
21     if (classOf[Date].isAssignableFrom(source) && classOf[String].isAssignableFrom(target)) {
22       JavaTimeConverterFactory.DateToStringConverter
23     } else if (classOf[Date].isAssignableFrom(target)) {
24       if (classOf[String].isAssignableFrom(source)) {
25         JavaTimeConverterFactory.StringToDateConverter
26       } else if (classOf[ZonedDateTime].isAssignableFrom(source) || classOf[Instant].isAssignableFrom(source)
27           || classOf[OffsetDateTime].isAssignableFrom(source)) {
28         JavaTimeConverterFactory.TemporalToDateConverter
29       } else if (classOf[LocalDateTime].isAssignableFrom(source)) {
30         JavaTimeConverterFactory.LocalDateTimeToDateConverter
31       } else if (classOf[LocalDate].isAssignableFrom(source)) {
32         JavaTimeConverterFactory.LocalDateToDateConverter
33       } else {
34         null
35       }
36     } else {
37       null
38     }
39   }
40 }
41 
42 object JavaTimeConverterFactory {
43 
44   private val DateToStringConverter = new Converter {
45     override def convert[T](source: AnyRef, target: Class[T]): T =
46       DateParsing.formatDate(source.asInstanceOf[Date]).asInstanceOf[T]
47   }
48 
49   private val StringToDateConverter = new Converter {
50     override def convert[T](source: AnyRef, target: Class[T]): T =
51       DateParsing.parseDate(source.asInstanceOf[String]).asInstanceOf[T]
52   }
53 
54   private val TemporalToDateConverter = new Converter {
55     override def convert[T](source: AnyRef, target: Class[T]): T = {
56       val accessor = source.asInstanceOf[TemporalAccessor]
57       val millis = accessor.getLong(ChronoField.INSTANT_SECONDS) * 1000 +
58           accessor.getLong(ChronoField.MILLI_OF_SECOND)
59       new Date(millis).asInstanceOf[T]
60     }
61   }
62 
63   private val LocalDateToDateConverter = new Converter {
64     override def convert[T](source: AnyRef, target: Class[T]): T =
65       TemporalToDateConverter.convert(source.asInstanceOf[LocalDate].atStartOfDay(ZoneOffset.UTC), target)
66   }
67 
68   private val LocalDateTimeToDateConverter = new Converter {
69     override def convert[T](source: AnyRef, target: Class[T]): T =
70       TemporalToDateConverter.convert(source.asInstanceOf[LocalDateTime].atZone(ZoneOffset.UTC), target)
71   }
72 }