sql parser schema stat

2023-03-04 19:17:16 +00:00 · 2012-05-24 13:54:44 +08:00
parent ae2cdb6069 a7ed24b3e9
commit 4b826f2d3e
13 changed files with 302 additions and 16 deletions
@@ -4,9 +4,15 @@ import com.alibaba.druid.sql.ast.statement.SQLCreateTableStatement;
 import com.alibaba.druid.sql.dialect.hive.ast.stmt.HiveCreateTableStatement;
 import com.alibaba.druid.sql.dialect.hive.ast.stmt.HiveShowTablesStatement;
 import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
+import com.alibaba.druid.util.JdbcUtils;

 public class HiveSchemaStatVisitor extends SchemaStatVisitor implements HiveASTVisitor {

+    @Override
+    public String getDbType() {
+        return JdbcUtils.HIVE;
+    }
+    
    @Override
    public void endVisit(HiveCreateTableStatement x) {
        
@@ -113,6 +113,7 @@ import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlUpdateStatement;
 import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
 import com.alibaba.druid.stat.TableStat;
 import com.alibaba.druid.stat.TableStat.Mode;
+import com.alibaba.druid.util.JdbcUtils;

 public class MySqlSchemaStatVisitor extends SchemaStatVisitor implements MySqlASTVisitor {

@@ -122,6 +123,11 @@ public class MySqlSchemaStatVisitor extends SchemaStatVisitor implements MySqlAS

        return true;
    }
+    
+    @Override
+    public String getDbType() {
+        return JdbcUtils.MYSQL;
+    }

    // DUAL
    public boolean visit(MySqlDeleteStatement x) {
@@ -0,0 +1,64 @@
+package com.alibaba.druid.sql.dialect.oracle.visitor;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
+import com.alibaba.druid.sql.ast.expr.SQLCharExpr;
+import com.alibaba.druid.sql.ast.expr.SQLVariantRefExpr;
+import com.alibaba.druid.sql.visitor.SQLEvalVisitor;
+import com.alibaba.druid.sql.visitor.SQLEvalVisitorUtils;
+
+
+public class OracleEvalVisitor extends OracleASTVisitorAdapter implements SQLEvalVisitor {
+    private List<Object> parameters       = new ArrayList<Object>();
+
+    private int          variantIndex     = -1;
+
+    private boolean      markVariantIndex = true;
+
+    public OracleEvalVisitor(){
+        this(new ArrayList<Object>(1));
+    }
+
+    public OracleEvalVisitor(List<Object> parameters){
+        this.parameters = parameters;
+    }
+
+    public List<Object> getParameters() {
+        return parameters;
+    }
+
+    public void setParameters(List<Object> parameters) {
+        this.parameters = parameters;
+    }
+
+    public boolean visit(SQLCharExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    public int incrementAndGetVariantIndex() {
+        return ++variantIndex;
+    }
+
+    public int getVariantIndex() {
+        return variantIndex;
+    }
+
+    public boolean visit(SQLVariantRefExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    public boolean visit(SQLBinaryOpExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    public boolean isMarkVariantIndex() {
+        return markVariantIndex;
+    }
+
+    public void setMarkVariantIndex(boolean markVariantIndex) {
+        this.markVariantIndex = markVariantIndex;
+    }
+
+}
@@ -1,5 +1,7 @@
 package com.alibaba.druid.sql.dialect.oracle.visitor;

+import java.util.ArrayList;
+import java.util.List;
 import java.util.Map;

 import com.alibaba.druid.sql.ast.SQLExpr;
@@ -140,15 +142,26 @@ import com.alibaba.druid.stat.TableStat;
 import com.alibaba.druid.stat.TableStat.Column;
 import com.alibaba.druid.stat.TableStat.Mode;
 import com.alibaba.druid.stat.TableStat.Relationship;
+import com.alibaba.druid.util.JdbcUtils;

 public class OracleSchemaStatVisitor extends SchemaStatVisitor implements OracleASTVisitor {

    public OracleSchemaStatVisitor(){
+        this (new ArrayList<Object>());
+    }
+    
+    public OracleSchemaStatVisitor(List<Object> parameters) {
+        super (parameters);
        this.variants.put("DUAL", null);
        this.variants.put("NOTFOUND", null);
        this.variants.put("TRUE", null);
        this.variants.put("FALSE", null);
    }
+    
+    @Override
+    public String getDbType() {
+        return JdbcUtils.ORACLE;
+    }

    protected Column getColumn(SQLExpr expr) {
        if (expr instanceof OracleOuterExpr) {
@@ -0,0 +1,64 @@
+package com.alibaba.druid.sql.dialect.postgresql.visitor;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
+import com.alibaba.druid.sql.ast.expr.SQLCharExpr;
+import com.alibaba.druid.sql.ast.expr.SQLVariantRefExpr;
+import com.alibaba.druid.sql.visitor.SQLEvalVisitor;
+import com.alibaba.druid.sql.visitor.SQLEvalVisitorUtils;
+
+
+public class PGEvalVisitor extends PGASTVisitorAdapter implements SQLEvalVisitor {
+    private List<Object> parameters       = new ArrayList<Object>();
+
+    private int          variantIndex     = -1;
+
+    private boolean      markVariantIndex = true;
+
+    public PGEvalVisitor(){
+        this(new ArrayList<Object>(1));
+    }
+
+    public PGEvalVisitor(List<Object> parameters){
+        this.parameters = parameters;
+    }
+
+    public List<Object> getParameters() {
+        return parameters;
+    }
+
+    public void setParameters(List<Object> parameters) {
+        this.parameters = parameters;
+    }
+
+    public boolean visit(SQLCharExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    public int incrementAndGetVariantIndex() {
+        return ++variantIndex;
+    }
+
+    public int getVariantIndex() {
+        return variantIndex;
+    }
+
+    public boolean visit(SQLVariantRefExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    public boolean visit(SQLBinaryOpExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    public boolean isMarkVariantIndex() {
+        return markVariantIndex;
+    }
+
+    public void setMarkVariantIndex(boolean markVariantIndex) {
+        this.markVariantIndex = markVariantIndex;
+    }
+
+}
@@ -23,9 +23,15 @@ import com.alibaba.druid.sql.dialect.postgresql.ast.stmt.PGUpdateStatement;
 import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
 import com.alibaba.druid.stat.TableStat;
 import com.alibaba.druid.stat.TableStat.Mode;
+import com.alibaba.druid.util.JdbcUtils;

 public class PGSchemaStatVisitor extends SchemaStatVisitor implements PGASTVisitor {

+    @Override
+    public String getDbType() {
+        return JdbcUtils.POSTGRESQL;
+    }
+    
    @Override
    public void endVisit(WindowClause x) {

@@ -5,10 +5,16 @@ import com.alibaba.druid.sql.dialect.sqlserver.ast.SQLServerSelectQueryBlock;
 import com.alibaba.druid.sql.dialect.sqlserver.ast.Top;
 import com.alibaba.druid.sql.dialect.sqlserver.ast.expr.SQLServerObjectReferenceExpr;
 import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
+import com.alibaba.druid.util.JdbcUtils;


 public class SQLServerSchemaStatVisitor extends SchemaStatVisitor implements SQLServerASTVisitor {

+    @Override
+    public String getDbType() {
+        return JdbcUtils.SQL_SERVER;
+    }
+    
    @Override
    public boolean visit(SQLServerSelectQueryBlock x) {
        return visit((SQLSelectQueryBlock) x);