DX-100503: missed classes

dremio · Feb 13, 2025 · 02bd636 · 02bd636
1 parent e87f922
commit 02bd636
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/java/c/src/main/java/org/apache/arrow/c/BufferImportTypeVisitor.java b/java/c/src/main/java/org/apache/arrow/c/BufferImportTypeVisitor.java
@@ -381,6 +381,12 @@ public List<ArrowBuf> visit(ArrowType.Timestamp type) {
         maybeImportBitmap(type), importFixedBytes(type, 1, TimeStampVector.TYPE_WIDTH));
   }
 
+  @Override
+  public List<ArrowBuf> visit(ArrowType.TimestampWithPrecision type) {
+    return Arrays.asList(
+        maybeImportBitmap(type), importFixedBytes(type, 1, TimeStampVector.TYPE_WIDTH));
+  }
+
   @Override
   public List<ArrowBuf> visit(ArrowType.Interval type) {
     switch (type.getUnit()) {

diff --git a/java/c/src/main/java/org/apache/arrow/c/Format.java b/java/c/src/main/java/org/apache/arrow/c/Format.java
@@ -208,6 +208,12 @@ static String asString(ArrowType arrowType) {
           String timezone = type.getTimezone();
           return String.format("%s:%s", format, timezone == null ? "" : timezone);
         }
+      case TimestampWithPrecision:
+      {
+        ArrowType.TimestampWithPrecision type = (ArrowType.TimestampWithPrecision) arrowType;
+        String timezone = type.getTimezone();
+        return String.format("tsp:%s,%d", timezone == null ? "" : timezone, type.getPrecision());
+      }
       case Union:
         ArrowType.Union type = (ArrowType.Union) arrowType;
         String typeIDs =
@@ -360,6 +366,9 @@ private static ArrowType parseComplexFormat(String format, String payload)
         return new ArrowType.Timestamp(TimeUnit.MICROSECOND, payloadToTimezone(payload));
       case "tsn":
         return new ArrowType.Timestamp(TimeUnit.NANOSECOND, payloadToTimezone(payload));
+      case "tsp":
+        String[] parts = payload.split(",");
+        return new ArrowType.TimestampWithPrecision(Integer.parseInt(parts[1]), parts[0]);
       default:
         throw new UnsupportedOperationException(
             String.format("Format %s:%s is not supported", format, payload));