Impl methods

Author: ayushtkn

common/src/java/org/apache/hadoop/hive/common/type/TimestampNano.java

@@ -19,4 +19,8 @@
 package org.apache.hadoop.hive.common.type;
 
 public class TimestampNano extends Timestamp {
+
+  public TimestampNano(Timestamp ts) {
+    super(ts);
+  }
 }

common/src/java/org/apache/hadoop/hive/common/type/TimestampNanoTZ.java

@@ -18,5 +18,10 @@
 
 package org.apache.hadoop.hive.common.type;
 
+import java.time.ZonedDateTime;
+
 public class TimestampNanoTZ extends TimestampTZ {
+  public TimestampNanoTZ(ZonedDateTime zonedDateTime) {
+    super(zonedDateTime);
+  }
 }

iceberg/iceberg-handler/src/main/java/org/apache/iceberg/mr/hive/serde/objectinspector/IcebergNanosecondTimestampObjectInspectorHive3.java

@@ -19,6 +19,9 @@
 
 package org.apache.iceberg.mr.hive.serde.objectinspector;
 
+import org.apache.hadoop.hive.common.type.Timestamp;
+import org.apache.hadoop.hive.common.type.TimestampNano;
+import org.apache.hadoop.hive.serde2.io.TimestampNanoWritable;
 import org.apache.hadoop.hive.serde2.typeinfo.TypeInfoFactory;
 
 public class IcebergNanosecondTimestampObjectInspectorHive3 extends IcebergTimestampObjectInspectorHive3 {
@@ -34,4 +37,18 @@ private IcebergNanosecondTimestampObjectInspectorHive3() {
     super(TypeInfoFactory.timestampNanoTypeInfo);
   }
 
+  @Override
+  public TimestampNano getPrimitiveJavaObject(Object o) {
+    Timestamp timestamp = super.getPrimitiveJavaObject(o);
+    if (timestamp == null) {
+      return null;
+    }
+    return new TimestampNano(timestamp);
+  }
+
+  @Override
+  public TimestampNanoWritable getPrimitiveWritableObject(Object o) {
+    TimestampNano ts = getPrimitiveJavaObject(o);
+    return ts == null ? null : new TimestampNanoWritable(ts);
+  }
 }

iceberg/iceberg-handler/src/main/java/org/apache/iceberg/mr/hive/serde/objectinspector/IcebergNanosecondTimestampWithZoneObjectInspectorHive3.java

@@ -19,6 +19,9 @@
 
 package org.apache.iceberg.mr.hive.serde.objectinspector;
 
+import org.apache.hadoop.hive.common.type.TimestampNanoTZ;
+import org.apache.hadoop.hive.common.type.TimestampTZ;
+import org.apache.hadoop.hive.serde2.io.TimestampNanoTZWritable;
 import org.apache.hadoop.hive.serde2.typeinfo.TypeInfoFactory;
 
 public class IcebergNanosecondTimestampWithZoneObjectInspectorHive3
@@ -34,4 +37,19 @@ public static IcebergNanosecondTimestampWithZoneObjectInspectorHive3 get() {
   private IcebergNanosecondTimestampWithZoneObjectInspectorHive3() {
     super(TypeInfoFactory.timestampNanosTZTypeInfo);
   }
+
+  @Override
+  public TimestampNanoTZ getPrimitiveJavaObject(Object o) {
+    TimestampTZ ttz = super.getPrimitiveJavaObject(o);
+    if (ttz == null) {
+      return null;
+    }
+    return new TimestampNanoTZ(ttz.getZonedDateTime());
+  }
+
+  @Override
+  public TimestampNanoTZWritable getPrimitiveWritableObject(Object o) {
+    TimestampNanoTZ tsTz = getPrimitiveJavaObject(o);
+    return tsTz == null ? null : new TimestampNanoTZWritable(tsTz);
+  }
 }

serde/src/java/org/apache/hadoop/hive/serde2/io/TimestampNanoTZWritable.java

@@ -18,5 +18,10 @@
 
 package org.apache.hadoop.hive.serde2.io;
 
+import org.apache.hadoop.hive.common.type.TimestampNanoTZ;
+
 public class TimestampNanoTZWritable extends TimestampLocalTZWritable {
+  public TimestampNanoTZWritable(TimestampNanoTZ tsTz) {
+    super(tsTz);
+  }
 }

serde/src/java/org/apache/hadoop/hive/serde2/io/TimestampNanoWritable.java

@@ -18,5 +18,10 @@
 
 package org.apache.hadoop.hive.serde2.io;
 
+import org.apache.hadoop.hive.common.type.TimestampNano;
+
 public class TimestampNanoWritable extends TimestampWritableV2 {
+  public TimestampNanoWritable(TimestampNano ts) {
+    super(ts);
+  }
 }