add mapping support

2023-03-04 19:17:16 +00:00 · 2012-03-13 23:41:59 +08:00
parent fdea9fd669
commit 5f1570a642
11 changed files with 326 additions and 0 deletions
@@ -0,0 +1,8 @@
+package com.alibaba.druid.mapping.spi;
+
+import com.alibaba.druid.sql.visitor.SQLASTVisitor;
+
+
+public interface ExportParameterVisitor extends SQLASTVisitor {
+    
+}
@@ -0,0 +1,87 @@
+package com.alibaba.druid.mapping.spi;
+
+import java.util.List;
+
+import com.alibaba.druid.sql.ast.SQLExpr;
+import com.alibaba.druid.sql.ast.expr.SQLBetweenExpr;
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
+import com.alibaba.druid.sql.ast.expr.SQLCharExpr;
+import com.alibaba.druid.sql.ast.expr.SQLLiteralExpr;
+import com.alibaba.druid.sql.ast.expr.SQLNumericLiteralExpr;
+import com.alibaba.druid.sql.ast.expr.SQLVariantRefExpr;
+import com.alibaba.druid.sql.dialect.mysql.ast.expr.MySqlBooleanExpr;
+
+public class ExportParameterVisitorUtils {
+
+    public static boolean exportParamterAndAccept(final List<Object> parameters, List<SQLExpr> list) {
+        for (int i = 0, size = list.size(); i < size; ++i) {
+            SQLExpr param = list.get(i);
+
+            SQLExpr result = exportParameter(parameters, param);
+            if (result != param) {
+                list.set(i, result);
+            }
+        }
+
+        return false;
+    }
+
+    public static SQLExpr exportParameter(final List<Object> parameters, final SQLExpr param) {
+        if (param instanceof SQLCharExpr) {
+            Object value = ((SQLCharExpr) param).getText();
+            parameters.add(value);
+            return new SQLVariantRefExpr("?");
+        }
+
+        if (param instanceof MySqlBooleanExpr) {
+            Object value = ((MySqlBooleanExpr) param).getValue();
+            parameters.add(value);
+            return new SQLVariantRefExpr("?");
+        }
+
+        if (param instanceof SQLNumericLiteralExpr) {
+            Object value = ((SQLNumericLiteralExpr) param).getNumber();
+            parameters.add(value);
+            return new SQLVariantRefExpr("?");
+        }
+
+        return param;
+    }
+
+    public static void exportParameter(final List<Object> parameters, SQLBinaryOpExpr x) {
+        if (x.getLeft() instanceof SQLLiteralExpr && x.getRight() instanceof SQLLiteralExpr) {
+            return;
+        }
+
+        {
+            SQLExpr leftResult = ExportParameterVisitorUtils.exportParameter(parameters, x.getLeft());
+            if (leftResult != x.getLeft()) {
+                x.setLeft(leftResult);
+            }
+        }
+
+        {
+            SQLExpr rightResult = exportParameter(parameters, x.getRight());
+            if (rightResult != x.getRight()) {
+                x.setRight(rightResult);
+            }
+        }
+    }
+
+    public static void exportParameter(final List<Object> parameters, SQLBetweenExpr x) {
+        {
+            SQLExpr result = exportParameter(parameters, x.getBeginExpr());
+            if (result != x.getBeginExpr()) {
+                x.setBeginExpr(result);
+            }
+        }
+
+        {
+            SQLExpr result = exportParameter(parameters, x.getEndExpr());
+            if (result != x.getBeginExpr()) {
+                x.setEndExpr(result);
+            }
+        }
+
+    }
+}
@@ -0,0 +1,66 @@
+package com.alibaba.druid.mapping.spi;
+
+import java.util.List;
+
+import com.alibaba.druid.sql.ast.SQLOrderBy;
+import com.alibaba.druid.sql.ast.expr.SQLBetweenExpr;
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
+import com.alibaba.druid.sql.ast.expr.SQLInListExpr;
+import com.alibaba.druid.sql.ast.expr.SQLMethodInvokeExpr;
+import com.alibaba.druid.sql.ast.statement.SQLSelectGroupByClause;
+import com.alibaba.druid.sql.ast.statement.SQLSelectItem;
+import com.alibaba.druid.sql.dialect.sqlserver.visitor.SQLServerASTVisitorAdapter;
+
+public class MSSQLServerExportParameterVisitor extends SQLServerASTVisitorAdapter implements ExportParameterVisitor {
+
+    private final List<Object> parameters;
+
+    public MSSQLServerExportParameterVisitor(List<Object> parameters){
+        this.parameters = parameters;
+    }
+
+    public List<Object> getParameters() {
+        return parameters;
+    }
+
+    @Override
+    public boolean visit(SQLSelectItem x) {
+        return false;
+    }
+
+    @Override
+    public boolean visit(SQLOrderBy x) {
+        return false;
+    }
+
+    @Override
+    public boolean visit(SQLSelectGroupByClause x) {
+        return false;
+    }
+
+    @Override
+    public boolean visit(SQLMethodInvokeExpr x) {
+        ExportParameterVisitorUtils.exportParamterAndAccept(this.parameters, x.getParameters());
+
+        return true;
+    }
+
+    @Override
+    public boolean visit(SQLInListExpr x) {
+        ExportParameterVisitorUtils.exportParamterAndAccept(this.parameters, x.getTargetList());
+
+        return true;
+    }
+
+    @Override
+    public boolean visit(SQLBetweenExpr x) {
+        ExportParameterVisitorUtils.exportParameter(this.parameters, x);
+        return true;
+    }
+
+    public boolean visit(SQLBinaryOpExpr x) {
+        ExportParameterVisitorUtils.exportParameter(this.parameters, x);
+        return true;
+    }
+
+}
@@ -1,5 +1,7 @@
 package com.alibaba.druid.mapping.spi;

+import java.util.List;
+
 import com.alibaba.druid.mapping.MappingEngine;
 import com.alibaba.druid.sql.ast.expr.SQLNumberExpr;
 import com.alibaba.druid.sql.ast.expr.SQLNumericLiteralExpr;
@@ -65,4 +67,9 @@ public class MSSQLServerMappingProvider implements MappingProvider {
        PGSQLStatementParser parser = new PGSQLStatementParser(sql);
        return (SQLInsertStatement) parser.parseInsert();
    }
+
+    @Override
+    public ExportParameterVisitor createExportParameterVisitor(List<Object> parameters) {
+        return new MSSQLServerExportParameterVisitor(parameters);
+    }
 }
@@ -1,5 +1,7 @@
 package com.alibaba.druid.mapping.spi;

+import java.util.List;
+
 import com.alibaba.druid.mapping.MappingEngine;
 import com.alibaba.druid.sql.ast.statement.SQLDeleteStatement;
 import com.alibaba.druid.sql.ast.statement.SQLInsertStatement;
@@ -10,6 +12,8 @@ import com.alibaba.druid.sql.visitor.SQLASTOutputVisitor;
 public interface MappingProvider {

    MappingVisitor createMappingVisitor(MappingEngine engine);
+    
+    ExportParameterVisitor createExportParameterVisitor(List<Object> parameters);

    SQLASTOutputVisitor createOutputVisitor(MappingEngine engine, Appendable out);

@@ -121,4 +121,5 @@ public class MappingVisitorUtils {

        return entity;
    }
+
 }
@@ -0,0 +1,72 @@
+package com.alibaba.druid.mapping.spi;
+
+import java.util.List;
+
+import com.alibaba.druid.sql.ast.SQLOrderBy;
+import com.alibaba.druid.sql.ast.expr.SQLBetweenExpr;
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
+import com.alibaba.druid.sql.ast.expr.SQLInListExpr;
+import com.alibaba.druid.sql.ast.expr.SQLMethodInvokeExpr;
+import com.alibaba.druid.sql.ast.statement.SQLSelectGroupByClause;
+import com.alibaba.druid.sql.ast.statement.SQLSelectItem;
+import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlSelectQueryBlock.Limit;
+import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlASTVisitorAdapter;
+
+public class MySqlExportParameterVisitor extends MySqlASTVisitorAdapter implements ExportParameterVisitor {
+
+    private final List<Object> parameters;
+
+    public MySqlExportParameterVisitor(List<Object> parameters){
+        this.parameters = parameters;
+    }
+
+    public List<Object> getParameters() {
+        return parameters;
+    }
+
+    @Override
+    public boolean visit(SQLSelectItem x) {
+        return false;
+    }
+
+    @Override
+    public boolean visit(Limit x) {
+        return false;
+    }
+
+    @Override
+    public boolean visit(SQLOrderBy x) {
+        return false;
+    }
+
+    @Override
+    public boolean visit(SQLSelectGroupByClause x) {
+        return false;
+    }
+
+    @Override
+    public boolean visit(SQLMethodInvokeExpr x) {
+        ExportParameterVisitorUtils.exportParamterAndAccept(this.parameters, x.getParameters());
+
+        return true;
+    }
+
+    @Override
+    public boolean visit(SQLInListExpr x) {
+        ExportParameterVisitorUtils.exportParamterAndAccept(this.parameters, x.getTargetList());
+
+        return true;
+    }
+
+    @Override
+    public boolean visit(SQLBetweenExpr x) {
+        ExportParameterVisitorUtils.exportParameter(this.parameters, x);
+        return true;
+    }
+
+    public boolean visit(SQLBinaryOpExpr x) {
+        ExportParameterVisitorUtils.exportParameter(this.parameters, x);
+        return true;
+    }
+
+}
@@ -1,5 +1,7 @@
 package com.alibaba.druid.mapping.spi;

+import java.util.List;
+
 import com.alibaba.druid.mapping.MappingEngine;
 import com.alibaba.druid.sql.ast.expr.SQLIntegerExpr;
 import com.alibaba.druid.sql.ast.expr.SQLNumericLiteralExpr;
@@ -63,4 +65,9 @@ public class MySqlMappingProvider implements MappingProvider {
        MySqlStatementParser parser = new MySqlStatementParser(sql);
        return (SQLInsertStatement) parser.parseInsert();
    }
+    
+    @Override
+    public ExportParameterVisitor createExportParameterVisitor(List<Object> parameters) {
+        return new MySqlExportParameterVisitor(parameters);
+    }
 }
@@ -0,0 +1,66 @@
+package com.alibaba.druid.mapping.spi;
+
+import java.util.List;
+
+import com.alibaba.druid.sql.ast.SQLOrderBy;
+import com.alibaba.druid.sql.ast.expr.SQLBetweenExpr;
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
+import com.alibaba.druid.sql.ast.expr.SQLInListExpr;
+import com.alibaba.druid.sql.ast.expr.SQLMethodInvokeExpr;
+import com.alibaba.druid.sql.ast.statement.SQLSelectGroupByClause;
+import com.alibaba.druid.sql.ast.statement.SQLSelectItem;
+import com.alibaba.druid.sql.dialect.oracle.visitor.OracleASTVIsitorAdapter;
+
+public class OracleExportParameterVisitor extends OracleASTVIsitorAdapter implements ExportParameterVisitor {
+
+    private final List<Object> parameters;
+
+    public OracleExportParameterVisitor(List<Object> parameters){
+        this.parameters = parameters;
+    }
+
+    public List<Object> getParameters() {
+        return parameters;
+    }
+
+    @Override
+    public boolean visit(SQLSelectItem x) {
+        return false;
+    }
+
+    @Override
+    public boolean visit(SQLOrderBy x) {
+        return false;
+    }
+
+    @Override
+    public boolean visit(SQLSelectGroupByClause x) {
+        return false;
+    }
+
+    @Override
+    public boolean visit(SQLMethodInvokeExpr x) {
+        ExportParameterVisitorUtils.exportParamterAndAccept(this.parameters, x.getParameters());
+
+        return true;
+    }
+
+    @Override
+    public boolean visit(SQLInListExpr x) {
+        ExportParameterVisitorUtils.exportParamterAndAccept(this.parameters, x.getTargetList());
+
+        return true;
+    }
+
+    @Override
+    public boolean visit(SQLBetweenExpr x) {
+        ExportParameterVisitorUtils.exportParameter(this.parameters, x);
+        return true;
+    }
+
+    public boolean visit(SQLBinaryOpExpr x) {
+        ExportParameterVisitorUtils.exportParameter(this.parameters, x);
+        return true;
+    }
+
+}
@@ -1,5 +1,7 @@
 package com.alibaba.druid.mapping.spi;

+import java.util.List;
+
 import com.alibaba.druid.mapping.MappingEngine;
 import com.alibaba.druid.sql.ast.statement.SQLInsertStatement;
 import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
@@ -45,4 +47,9 @@ public class OracleMappingProvider implements MappingProvider {
        OracleStatementParser parser = new OracleStatementParser(sql);
        return (SQLInsertStatement) parser.parseInsert();
    }
+    
+    @Override
+    public ExportParameterVisitor createExportParameterVisitor(List<Object> parameters) {
+        return new OracleExportParameterVisitor(parameters);
+    }
 }
@@ -24,4 +24,5 @@ public abstract class SQLLiteralExpr extends SQLExprImpl {
    public SQLLiteralExpr(){

    }
+
 }