修复SQLUtils的toSQLString在VisitorFeature.OutputNameQuote模式下,会将Mysql的TIME…

…STAMPDIFF和TIMESTAMPADD第一个unit参数当作字段添加``问题 #5309
alibaba · Jul 9, 2023 · bf4e78e · bf4e78e
1 parent e766f53
commit bf4e78e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/core/src/main/java/com/alibaba/druid/sql/visitor/SQLASTOutputVisitor.java b/core/src/main/java/com/alibaba/druid/sql/visitor/SQLASTOutputVisitor.java
@@ -1923,6 +1923,7 @@ public boolean visit(SQLMethodInvokeExpr x) {
 
     protected void printMethodParameters(SQLMethodInvokeExpr x) {
         String function = x.getMethodName();
+        long nameHashCode64 = x.methodNameHashCode64();
         List<SQLExpr> parameters = x.getArguments();
 
         print('(');
@@ -1971,6 +1972,14 @@ protected void printMethodParameters(SQLMethodInvokeExpr x) {
                 }
             }
 
+            if (i == 0 && (nameHashCode64 == FnvHash.Constants.TIMESTAMPDIFF || nameHashCode64 == FnvHash.Constants.TIMESTAMPADD)
+                    && param instanceof SQLIdentifierExpr) {
+                if (DbType.mysql == dbType) {
+                    print(((SQLIdentifierExpr) param).getName());
+                    continue;
+                }
+            }
+
             printExpr(param, parameterized);
         }