use coralsqldialect in trino reltosql

linkedin · Jul 30, 2024 · bd35f9a · bd35f9a
1 parent af0832c
commit bd35f9a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/coral-trino/src/main/java/com/linkedin/coral/trino/rel2trino/RelToTrinoConverter.java b/coral-trino/src/main/java/com/linkedin/coral/trino/rel2trino/RelToTrinoConverter.java
@@ -5,6 +5,7 @@
  */
 package com.linkedin.coral.trino.rel2trino;
 
+import com.linkedin.coral.transformers.CoralRelToSqlNodeConverter;
 import java.math.BigDecimal;
 import java.util.ArrayList;
 import java.util.Collections;
@@ -34,6 +35,7 @@
 import com.linkedin.coral.common.functions.FunctionFieldReferenceOperator;
 
 import static com.google.common.base.Preconditions.*;
+import static com.linkedin.coral.transformers.CoralRelToSqlNodeConverter.INSTANCE;
 import static com.linkedin.coral.trino.rel2trino.CoralTrinoConfigKeys.*;
 
 
@@ -60,7 +62,7 @@ public class RelToTrinoConverter extends RelToSqlConverter {
    * @param mscClient client interface used to interact with the Hive Metastore service.
    */
   public RelToTrinoConverter(HiveMetastoreClient mscClient) {
-    super(TrinoSqlDialect.INSTANCE);
+    super(INSTANCE);
     _hiveMetastoreClient = mscClient;
   }
 
@@ -70,7 +72,7 @@ public RelToTrinoConverter(HiveMetastoreClient mscClient) {
    * @param configs configs
    */
   public RelToTrinoConverter(HiveMetastoreClient mscClient, Map<String, Boolean> configs) {
-    super(TrinoSqlDialect.INSTANCE);
+    super(INSTANCE);
     checkNotNull(configs);
     this.configs = configs;
     _hiveMetastoreClient = mscClient;

diff --git a/coral-trino/src/main/java/com/linkedin/coral/trino/rel2trino/TrinoSqlDialect.java b/coral-trino/src/main/java/com/linkedin/coral/trino/rel2trino/TrinoSqlDialect.java
@@ -76,7 +76,11 @@ public void unparseCall(SqlWriter writer, SqlCall call, int leftPrec, int rightP
         unparseMapValueConstructor(writer, call, leftPrec, rightPrec);
         break;
       default:
-        super.unparseCall(writer, call, leftPrec, rightPrec);
+        if (call.getOperator().getName().equals("timestamp_from_unixtime")) {
+          TIMESTAMP_FROM_UNIXTIME.unparse(writer, call, leftPrec, rightPrec);
+        } else {
+          super.unparseCall(writer, call, leftPrec, rightPrec);
+        }
     }
   }