Trino dialect, the extract function supports dow and week parameters

Signed-off-by: zhiminr.ren <[email protected]>
StarRocks · Nov 12, 2024 · 21f42c0 · 21f42c0
1 parent 134f856
commit 21f42c0
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/fe/fe-core/src/main/java/com/starrocks/connector/parser/trino/AstBuilder.java b/fe/fe-core/src/main/java/com/starrocks/connector/parser/trino/AstBuilder.java
@@ -998,14 +998,9 @@ protected ParseNode visitCoalesceExpression(CoalesceExpression node, ParseTreeCo
 
     @Override
     protected ParseNode visitExtract(Extract node, ParseTreeContext context) {
-        String fieldString = node.getField().toString();
-        if (fieldString.equalsIgnoreCase("dow")) {
-            fieldString = "dayofweek_iso";
-        } else if (fieldString.equalsIgnoreCase("week")) {
-            fieldString = "week_iso";
-        }
-        return new FunctionCallExpr(fieldString,
-                new FunctionParams(Lists.newArrayList((Expr) visit(node.getExpression(), context))));
+        String fieldString = node.getField().toString().toLowerCase();
+        return  Trino2SRFunctionCallTransformer.convert(fieldString,
+                Lists.newArrayList((Expr) visit(node.getExpression(), context)));
     }
 
     @Override