refactor: remove setTimestamp function

Author: NguyenHoangSon96

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

@@ -185,52 +185,48 @@ public Stream<Object[]> query(@Nonnull final String query,
         return queryData(query, parameters, options)
                 .flatMap(vector -> {
                     List<FieldVector> fieldVectors = vector.getFieldVectors();
+                    var fields = vector.getSchema().getFields();
                     return IntStream.range(0, vector.getRowCount())
                            .mapToObj(rowNumber -> {
                                ArrayList<Object> row = new ArrayList<>();
                                for (int i = 0; i < fieldVectors.size(); i++) {
-                                   var schema = vector.getSchema().getFields().get(i);
+                                   var schema = fields.get(i);
                                    var metaType = schema.getMetadata().get("iox::column::type");
                                    String valueType = metaType != null ? metaType.split("::")[2] : null;
 
+                                   Object value = fieldVectors.get(i).getObject(rowNumber);
                                    if ("field".equals(valueType)) {
                                        switch (metaType) {
                                            case "iox::column_type::field::integer":
                                            case "iox::column_type::field::uinteger":
-                                               var intValue = (Long) fieldVectors.get(i)
-                                                                                 .getObject(rowNumber);
+                                               var intValue = (Long) value;
                                                row.add(intValue);
                                                break;
                                            case "iox::column_type::field::float":
-                                               var doubleValue = (Double) fieldVectors.get(i)
-                                                                                      .getObject(rowNumber);
+                                               var doubleValue = (Double) value;
                                                row.add(doubleValue);
                                                break;
                                            case "iox::column_type::field::string":
-                                               var textValue = (Text) fieldVectors.get(i)
-                                                                                  .getObject(rowNumber);
+                                               var textValue = (Text) value;
                                                row.add(textValue.toString());
                                                break;
                                            case "iox::column_type::field::boolean":
-                                               var boolValue = (Boolean) fieldVectors.get(i)
-                                                                                     .getObject(rowNumber);
+                                               var boolValue = (Boolean) value;
                                                row.add(boolValue);
                                                break;
                                            default:
                                        }
                                    } else if ("timestamp".equals(valueType)
                                            || Objects.equals(schema.getName(), "time")) {
-                                       var timestamp = fieldVectors.get(i).getObject(rowNumber);
-                                       BigInteger time = NanosecondConverter.getTimestampNano(timestamp, schema);
+                                       BigInteger time = NanosecondConverter.getTimestampNano(value, schema);
                                        row.add(time);
                                    } else {
-                                       Object value = fieldVectors.get(i).getObject(rowNumber);
                                        row.add(value);
                                    }
                                }
 
                                return row.toArray();
-                                    });
+                           });
                 });
     }
 

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

@@ -21,22 +21,18 @@
  */
 package com.influxdb.v3.client.internal;
 
-import java.time.Instant;
 import java.time.LocalDateTime;
-import java.time.ZoneOffset;
 import java.util.List;
 import java.util.Objects;
-import java.util.concurrent.TimeUnit;
 import javax.annotation.Nonnull;
 import javax.annotation.concurrent.ThreadSafe;
 
 import org.apache.arrow.vector.FieldVector;
 import org.apache.arrow.vector.VectorSchemaRoot;
-import org.apache.arrow.vector.types.pojo.ArrowType;
-import org.apache.arrow.vector.types.pojo.Field;
 import org.apache.arrow.vector.util.Text;
 
 import com.influxdb.v3.client.PointValues;
+import com.influxdb.v3.client.write.WritePrecision;
 
 
 /**
@@ -82,7 +78,8 @@ PointValues toPointValues(final int rowNumber,
 
             if (metaType == null) {
                 if (Objects.equals(name, "time") && (value instanceof Long || value instanceof LocalDateTime)) {
-                    setTimestamp(value, schema, p);
+                    var timeNano = NanosecondConverter.getTimestampNano(value, schema);
+                    p.setTimestamp(timeNano, WritePrecision.NS);
                 } else {
                     // just push as field If you don't know what type is it
                     p.setField(name, value);
@@ -99,7 +96,8 @@ PointValues toPointValues(final int rowNumber,
             } else if ("tag".equals(valueType) && value instanceof String) {
                 p.setTag(name, (String) value);
             } else if ("timestamp".equals(valueType)) {
-                setTimestamp(value, schema, p);
+                var timeNano = NanosecondConverter.getTimestampNano(value, schema);
+                p.setTimestamp(timeNano, WritePrecision.NS);
             }
         }
         return p;
@@ -128,38 +126,8 @@ private void setFieldWithMetaType(final PointValues p,
                 p.setBooleanField(name, (Boolean) value);
                 break;
             default:
-        }
-    }
-
-    private void setTimestamp(@Nonnull final Object value,
-                              @Nonnull final Field schema,
-                              @Nonnull final PointValues pointValues) {
-        if (value instanceof Long) {
-            if (schema.getFieldType().getType() instanceof ArrowType.Timestamp) {
-                ArrowType.Timestamp type = (ArrowType.Timestamp) schema.getFieldType().getType();
-                TimeUnit timeUnit;
-                switch (type.getUnit()) {
-                    case SECOND:
-                        timeUnit = TimeUnit.SECONDS;
-                        break;
-                    case MILLISECOND:
-                        timeUnit = TimeUnit.MILLISECONDS;
-                        break;
-                    case MICROSECOND:
-                        timeUnit = TimeUnit.MICROSECONDS;
-                        break;
-                    default:
-                    case NANOSECOND:
-                        timeUnit = TimeUnit.NANOSECONDS;
-                        break;
-                }
-                long nanoseconds = TimeUnit.NANOSECONDS.convert((Long) value, timeUnit);
-                pointValues.setTimestamp(Instant.ofEpochSecond(0, nanoseconds));
-            } else {
-                pointValues.setTimestamp(Instant.ofEpochMilli((Long) value));
-            }
-        } else if (value instanceof LocalDateTime) {
-            pointValues.setTimestamp(((LocalDateTime) value).toInstant(ZoneOffset.UTC));
+                p.setField(name, value);
+                break;
         }
     }
 }