add merge sql stat filter

2023-03-04 19:17:16 +00:00 · 2012-03-31 08:40:29 +08:00
parent 04caf0e50f
commit a8bd3b249a
7 changed files with 398 additions and 238 deletions
@@ -0,0 +1,72 @@
+package com.alibaba.druid.sql.dialect.postgresql.visitor;
+
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
+import com.alibaba.druid.sql.ast.expr.SQLCharExpr;
+import com.alibaba.druid.sql.ast.expr.SQLInListExpr;
+import com.alibaba.druid.sql.ast.expr.SQLIntegerExpr;
+import com.alibaba.druid.sql.ast.expr.SQLNCharExpr;
+import com.alibaba.druid.sql.ast.expr.SQLNullExpr;
+import com.alibaba.druid.sql.ast.expr.SQLNumberExpr;
+import com.alibaba.druid.sql.visitor.ParameterizedOutputVisitorUtils;
+
+public class PGParameterizedOutputVisitor extends PGOutputVisitor {
+    public PGParameterizedOutputVisitor() {
+        this (new StringBuilder());
+    }
+    
+    public PGParameterizedOutputVisitor(Appendable appender){
+        super(appender);
+    }
+
+    public boolean visit(SQLInListExpr x) {
+        return ParameterizedOutputVisitorUtils.visit(this, x);
+    }
+
+    public boolean visit(SQLBinaryOpExpr x) {
+        x = ParameterizedOutputVisitorUtils.merge(x);
+
+        return super.visit(x);
+    }
+
+    public boolean visit(SQLNullExpr x) {
+        print('?');
+        return false;
+    }
+
+    public boolean visit(SQLIntegerExpr x) {
+        if (Boolean.TRUE.equals(x.getAttribute(ParameterizedOutputVisitorUtils.ATTR_PARAMS_SKIP))) {
+            return super.visit(x);
+        }
+
+        print('?');
+        return false;
+    }
+
+    public boolean visit(SQLNumberExpr x) {
+        if (Boolean.TRUE.equals(x.getAttribute(ParameterizedOutputVisitorUtils.ATTR_PARAMS_SKIP))) {
+            return super.visit(x);
+        }
+
+        print('?');
+        return false;
+    }
+
+    public boolean visit(SQLCharExpr x) {
+        if (Boolean.TRUE.equals(x.getAttribute(ParameterizedOutputVisitorUtils.ATTR_PARAMS_SKIP))) {
+            return super.visit(x);
+        }
+
+        print('?');
+        return false;
+    }
+
+    public boolean visit(SQLNCharExpr x) {
+        if (Boolean.TRUE.equals(x.getAttribute(ParameterizedOutputVisitorUtils.ATTR_PARAMS_SKIP))) {
+            return super.visit(x);
+        }
+
+        print('?');
+        return false;
+    }
+
+}