Support automatic type coercion in Delta table creation

trinodb · Mar 21, 2024 · e9adf94 · e9adf94
1 parent f7db654
commit e9adf94
Show file tree

Hide file tree

Showing 2 changed files with 306 additions and 1 deletion.
diff --git a/plugin/trino-delta-lake/src/main/java/io/trino/plugin/deltalake/DeltaLakeMetadata.java b/plugin/trino-delta-lake/src/main/java/io/trino/plugin/deltalake/DeltaLakeMetadata.java
@@ -300,6 +300,7 @@
 import static io.trino.spi.type.IntegerType.INTEGER;
 import static io.trino.spi.type.RealType.REAL;
 import static io.trino.spi.type.SmallintType.SMALLINT;
+import static io.trino.spi.type.TimestampType.TIMESTAMP_MICROS;
 import static io.trino.spi.type.TinyintType.TINYINT;
 import static io.trino.spi.type.TypeUtils.isFloatingPointNaN;
 import static io.trino.spi.type.VarbinaryType.VARBINARY;
@@ -607,6 +608,15 @@ public ConnectorTableProperties getTableProperties(ConnectorSession session, Con
                 ImmutableList.of());
     }
 
+    @Override
+    public Optional<Type> getSupportedType(ConnectorSession session, Map<String, Object> tableProperties, Type type)
+    {
+        if (type instanceof TimestampType) {
+            return Optional.of(TIMESTAMP_MICROS);
+        }
+        return Optional.empty();
+    }
+
     @Override
     public SchemaTableName getTableName(ConnectorSession session, ConnectorTableHandle table)
     {