wip

Author: NguyenHoangSon96

src/main/java/com/influxdb/v3/client/internal/InfluxDBClientImpl.java

@@ -221,7 +221,7 @@ public Stream<Object[]> query(@Nonnull final String query,
                                             } else if ("timestamp".equals(valueType)
                                                     || Objects.equals(schema.getName(), "time")) {
                                                 var timestamp = fieldVectors.get(i).getObject(rowNumber);
-                                                BigInteger time = NanosecondConverter.getTimestamp(timestamp, schema);
+                                                BigInteger time = NanosecondConverter.getTimestampNanoSecond(timestamp, schema);
                                                 row.add(time);
                                             } else {
                                                 Object value = fieldVectors.get(i).getObject(rowNumber);

src/main/java/com/influxdb/v3/client/internal/NanosecondConverter.java

@@ -119,7 +119,7 @@ public static BigInteger convert(final Instant instant, final WritePrecision pre
         return FROM_NANOS.get(precision).apply(nanos);
     }
 
-    public static BigInteger getTimestamp(@Nonnull final Object value, @Nonnull final Field schema) {
+    public static BigInteger getTimestampNanoSecond(@Nonnull final Object value, @Nonnull final Field schema) {
         BigInteger result = null;
 
         if (value instanceof Long) {

src/test/java/com/influxdb/v3/client/internal/NanosecondConverterTest.java

@@ -0,0 +1,60 @@
+package com.influxdb.v3.client.internal;
+
+import java.math.BigInteger;
+
+import org.apache.arrow.vector.types.TimeUnit;
+import org.apache.arrow.vector.types.pojo.ArrowType;
+import org.apache.arrow.vector.types.pojo.Field;
+import org.apache.arrow.vector.types.pojo.FieldType;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+
+public class NanosecondConverterTest {
+
+    @Test
+    void testGetTimestampNanosecond() {
+        BigInteger timestampNanoSecond = null;
+
+        // Second
+        FieldType timeTypeSecond = new FieldType(true,
+                                                 new ArrowType.Timestamp(TimeUnit.SECOND, "UTC"),
+                                                 null);
+        Field timeFieldSecond = new Field("time", timeTypeSecond, null);
+        timestampNanoSecond = NanosecondConverter.getTimestampNanoSecond(123_456L, timeFieldSecond);
+        Assertions.assertEquals(
+                BigInteger.valueOf(123_456L)
+                          .multiply(BigInteger.valueOf(1_000_000_000)), timestampNanoSecond
+        );
+
+        // MilliSecond
+        FieldType timeTypeMilliSecond = new FieldType(true,
+                                                      new ArrowType.Timestamp(TimeUnit.MILLISECOND, "UTC"),
+                                                      null);
+        Field timeFieldMilliSecond = new Field("time", timeTypeMilliSecond, null);
+        timestampNanoSecond = NanosecondConverter.getTimestampNanoSecond(123_456L, timeFieldMilliSecond);
+        Assertions.assertEquals(
+                BigInteger.valueOf(123_456L)
+                          .multiply(BigInteger.valueOf(1_000_000)), timestampNanoSecond
+        );
+
+        // MicroSecond
+        FieldType timeTypeMicroSecond = new FieldType(true,
+                                                      new ArrowType.Timestamp(TimeUnit.MICROSECOND, "UTC"),
+                                                      null);
+        Field timeFieldMicroSecond = new Field("time", timeTypeMicroSecond, null);
+        timestampNanoSecond = NanosecondConverter.getTimestampNanoSecond(123_456L, timeFieldMicroSecond);
+        Assertions.assertEquals(
+                BigInteger.valueOf(123_456L)
+                          .multiply(BigInteger.valueOf(1_000)), timestampNanoSecond
+        );
+
+        // Nano Second
+        FieldType timeTypeNanoSecond = new FieldType(true,
+                                                      new ArrowType.Timestamp(TimeUnit.NANOSECOND, "UTC"),
+                                                      null);
+        Field timeFieldNanoSecond = new Field("time", timeTypeNanoSecond, null);
+        timestampNanoSecond = NanosecondConverter.getTimestampNanoSecond(123_456L, timeFieldNanoSecond);
+        Assertions.assertEquals(BigInteger.valueOf(123_456L), timestampNanoSecond
+        );
+    }
+}