Add support for kotlinx.datetime.Instant.

Add tests.

Author: eranl

firebase-firestore/firebase-firestore.gradle

@@ -142,6 +142,7 @@ dependencies {
     implementation libs.grpc.stub
     implementation libs.kotlin.stdlib
     implementation libs.kotlinx.coroutines.core
+    implementation libs.kotlinx.datetime
 
     compileOnly libs.autovalue.annotations
     compileOnly libs.javax.annotation.jsr250

firebase-firestore/src/main/java/com/google/firebase/firestore/util/CustomClassMapper.java

@@ -183,6 +183,10 @@ private static <T> Object serialize(T o, ErrorPath path) {
     } else if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O && o instanceof Instant) {
       Instant instant = (Instant) o;
       return new Timestamp(instant.getEpochSecond(), instant.getNano());
+    } else if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O
+        && o instanceof kotlinx.datetime.Instant) {
+      kotlinx.datetime.Instant instant = (kotlinx.datetime.Instant) o;
+      return new Timestamp(instant.getEpochSeconds(), instant.getNanosecondsOfSecond());
     } else if (o instanceof Uri || o instanceof URI || o instanceof URL) {
       return o.toString();
     } else {
@@ -246,6 +250,9 @@ private static <T> T deserializeToClass(Object o, Class<T> clazz, DeserializeCon
     } else if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O
         && Instant.class.isAssignableFrom(clazz)) {
       return (T) convertInstant(o, context);
+    } else if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O
+        && kotlinx.datetime.Instant.class.isAssignableFrom(clazz)) {
+      return (T) new kotlinx.datetime.Instant(convertInstant(o, context));
     } else if (Blob.class.isAssignableFrom(clazz)) {
       return (T) convertBlob(o, context);
     } else if (GeoPoint.class.isAssignableFrom(clazz)) {
@@ -958,7 +965,8 @@ private void applyFieldAnnotations(Field field) {
         Class<?> fieldType = field.getType();
         if (fieldType != Date.class
             && fieldType != Timestamp.class
-            && !(Build.VERSION.SDK_INT >= Build.VERSION_CODES.O && fieldType == Instant.class)) {
+            && !(Build.VERSION.SDK_INT >= Build.VERSION_CODES.O
+                && (fieldType == Instant.class || fieldType == kotlinx.datetime.Instant.class))) {
           throw new IllegalArgumentException(
               "Field "
                   + field.getName()
@@ -981,7 +989,8 @@ private void applyGetterAnnotations(Method method) {
         Class<?> returnType = method.getReturnType();
         if (returnType != Date.class
             && returnType != Timestamp.class
-            && !(Build.VERSION.SDK_INT >= Build.VERSION_CODES.O && returnType == Instant.class)) {
+            && !(Build.VERSION.SDK_INT >= Build.VERSION_CODES.O
+                && (returnType == Instant.class || returnType == kotlinx.datetime.Instant.class))) {
           throw new IllegalArgumentException(
               "Method "
                   + method.getName()

firebase-firestore/src/test/java/com/google/firebase/firestore/util/MapperTest.java

@@ -22,13 +22,15 @@
 import static org.junit.Assert.fail;
 
 import androidx.annotation.Nullable;
+import com.google.firebase.Timestamp;
 import com.google.firebase.firestore.DocumentId;
 import com.google.firebase.firestore.DocumentReference;
 import com.google.firebase.firestore.Exclude;
 import com.google.firebase.firestore.PropertyName;
 import com.google.firebase.firestore.TestUtil;
 import com.google.firebase.firestore.ThrowOnExtraProperties;
 import java.io.Serializable;
+import java.time.Instant;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collection;
@@ -37,6 +39,7 @@
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import java.util.Set;
 import org.junit.Test;
 import org.robolectric.annotation.Config;
@@ -95,6 +98,44 @@ public boolean isValue() {
     }
   }
 
+  private static class TimeBean {
+    public Timestamp timestamp;
+    public Date date;
+    public Instant instant;
+    public kotlinx.datetime.Instant instantKt;
+
+    @Override
+    public boolean equals(Object o) {
+      if (o == null || getClass() != o.getClass()) {
+        return false;
+      }
+      TimeBean timeBean = (TimeBean) o;
+      return Objects.equals(timestamp, timeBean.timestamp)
+          && Objects.equals(date, timeBean.date)
+          && Objects.equals(instant, timeBean.instant)
+          && Objects.equals(instantKt, timeBean.instantKt);
+    }
+
+    @Override
+    public int hashCode() {
+      return Objects.hash(timestamp, date, instant, instantKt);
+    }
+
+    @Override
+    public String toString() {
+      return "TimeBean{"
+          + "_date="
+          + date
+          + ", _timestamp="
+          + timestamp
+          + ", _instant="
+          + instant
+          + ", _instantKt="
+          + instantKt
+          + '}';
+    }
+  }
+
   private static class ShortBean {
     private short value;
 
@@ -1476,6 +1517,48 @@ public void serializeBooleanBean() {
     assertJson("{'value': true}", serialize(bean));
   }
 
+  @Test
+  public void serializeTimeBean() {
+    TimeBean bean = new TimeBean();
+    bean.instant = Instant.ofEpochSecond(1234, 5678);
+    bean.timestamp = new Timestamp(bean.instant);
+    bean.date = new Date(1234);
+    bean.instantKt = new kotlinx.datetime.Instant(bean.instant);
+    assertEquals(
+        Map.of(
+            "timestamp",
+            bean.timestamp,
+            "date",
+            bean.date,
+            "instant",
+            bean.timestamp,
+            "instantKt",
+            bean.timestamp),
+        serialize(bean));
+  }
+
+  @Test
+  public void deserializeTimeBean() {
+    TimeBean bean = new TimeBean();
+    bean.instant = Instant.ofEpochSecond(1234, 5678);
+    bean.timestamp = new Timestamp(bean.instant);
+    bean.date = new Date(1234);
+    bean.instantKt = new kotlinx.datetime.Instant(bean.instant);
+    assertEquals(
+        bean,
+        convertToCustomClass(
+            Map.of(
+                "timestamp",
+                bean.timestamp,
+                "date",
+                bean.date,
+                "instant",
+                bean.timestamp,
+                "instantKt",
+                bean.timestamp),
+            TimeBean.class));
+  }
+
   @Test
   public void serializeFloatBean() {
     FloatBean bean = new FloatBean();