refactor

src/main/java/com/alibaba/druid/mapping/MappingEngine.java

@@ -2,6 +2,9 @@ package com.alibaba.druid.mapping;
 
 import java.util.LinkedHashMap;
 
+import com.alibaba.druid.mapping.spi.MappingProvider;
+import com.alibaba.druid.mapping.spi.MappingVisitor;
+import com.alibaba.druid.mapping.spi.MySqlMappingProvider;
 import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
 import com.alibaba.druid.sql.visitor.SQLASTOutputVisitor;
 
@@ -9,7 +12,15 @@ public class MappingEngine {
 
     private LinkedHashMap<String, Entity> entities = new LinkedHashMap<String, Entity>();
     private Integer                       maxLimit;
-    private MappingProvider               provider;
+    private final MappingProvider         provider;
+
+    public MappingEngine(){
+        this(new MySqlMappingProvider());
+    }
+
+    public MappingEngine(MappingProvider provider){
+        this.provider = provider;
+    }
 
     public Integer getMaxLimit() {
         return maxLimit;
@@ -34,7 +45,7 @@ public class MappingEngine {
     public SQLASTOutputVisitor createOutputVisitor(Appendable out) {
         return provider.createOutputVisitor(this, out);
     }
-    
+
     public SQLSelectQueryBlock explainToSQLObject(String sql) {
         return provider.explainToSQLObject(this, sql);
     }

src/main/java/com/alibaba/druid/mapping/OracleMappingVisitor.java

@@ -1,93 +0,0 @@
-package com.alibaba.druid.mapping;
-
-import java.util.LinkedHashMap;
-import java.util.Map;
-
-import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
-import com.alibaba.druid.sql.ast.statement.SQLExprTableSource;
-import com.alibaba.druid.sql.ast.statement.SQLJoinTableSource;
-import com.alibaba.druid.sql.ast.statement.SQLSelectItem;
-import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
-import com.alibaba.druid.sql.ast.statement.SQLSubqueryTableSource;
-import com.alibaba.druid.sql.ast.statement.SQLTableSource;
-import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleSelectQueryBlock;
-import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleSelectTableReference;
-import com.alibaba.druid.sql.dialect.oracle.visitor.OracleASTVIsitorAdapter;
-
-public class OracleMappingVisitor extends OracleASTVIsitorAdapter implements MappingVisitor {
-	private final LinkedHashMap<String, Entity> entities;
-	private final Map<String, SQLTableSource> tableSources = new LinkedHashMap<String, SQLTableSource>();
-
-	public OracleMappingVisitor(LinkedHashMap<String, Entity> entities) {
-		super();
-		this.entities = entities;
-	}
-
-	public LinkedHashMap<String, Entity> getEntities() {
-		return entities;
-	}
-
-	public Map<String, SQLTableSource> getTableSources() {
-		return tableSources;
-	}
-
-	public Entity getFirstEntity() {
-		for (Map.Entry<String, Entity> entry : entities.entrySet()) {
-			return entry.getValue();
-		}
-
-		return null;
-	}
-	
-	public Entity getEntity(String name) {
-		Entity entity = this.entities.get(name);
-
-		if (entity == null) {
-			for (Map.Entry<String, Entity> entry : entities.entrySet()) {
-				if (entry.getKey().equalsIgnoreCase(name)) {
-					entity = entry.getValue();
-					break;
-				}
-			}
-		}
-
-		return entity;
-	}
-	
-	   public boolean visit(SQLSelectItem x) {
-	        x.getExpr().setParent(x);
-	        return true;
-	    }
-
-	    public boolean visit(OracleSelectQueryBlock x) {
-	        return MappingVisitorUtils.visit(this, x);
-	    }
-
-	    public boolean visit(SQLSelectQueryBlock x) {
-	        return MappingVisitorUtils.visit(this, x);
-	    }
-
-	    protected void fillSelectList(SQLSelectQueryBlock x) {
-	        MappingVisitorUtils.fillSelectList(this, x);
-	    }
-
-	    public boolean visit(SQLIdentifierExpr x) {
-	        return MappingVisitorUtils.visit(this, x);
-	    }
-
-	    public boolean visit(SQLSubqueryTableSource x) {
-	        return MappingVisitorUtils.visit(this, x);
-	    }
-
-	    public boolean visit(SQLJoinTableSource x) {
-	        return MappingVisitorUtils.visit(this, x);
-	    }
-
-	    public boolean visit(SQLExprTableSource x) {
-	        return MappingVisitorUtils.visit(this, x);
-	    }
-	    
-	    public boolean visit(OracleSelectTableReference x) {
-	        return MappingVisitorUtils.visit(this, x);
-	    }
-}

src/main/java/com/alibaba/druid/mapping/spi/MSSQLServerMappingVisitor.java

@@ -0,0 +1,84 @@
+package com.alibaba.druid.mapping.spi;
+
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+import com.alibaba.druid.mapping.Entity;
+import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
+import com.alibaba.druid.sql.ast.statement.SQLExprTableSource;
+import com.alibaba.druid.sql.ast.statement.SQLJoinTableSource;
+import com.alibaba.druid.sql.ast.statement.SQLSelectItem;
+import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
+import com.alibaba.druid.sql.ast.statement.SQLSubqueryTableSource;
+import com.alibaba.druid.sql.ast.statement.SQLTableSource;
+import com.alibaba.druid.sql.dialect.sqlserver.ast.SQLServerSelectQueryBlock;
+import com.alibaba.druid.sql.dialect.sqlserver.visitor.SQLServerASTVisitorAdapter;
+
+public class MSSQLServerMappingVisitor extends SQLServerASTVisitorAdapter implements MappingVisitor {
+
+    private final LinkedHashMap<String, Entity> entities;
+    private final Map<String, SQLTableSource>   tableSources = new LinkedHashMap<String, SQLTableSource>();
+
+    public MSSQLServerMappingVisitor(LinkedHashMap<String, Entity> entities){
+        super();
+        this.entities = entities;
+    }
+
+    public LinkedHashMap<String, Entity> getEntities() {
+        return entities;
+    }
+
+    public Map<String, SQLTableSource> getTableSources() {
+        return tableSources;
+    }
+
+    @Override
+    public Entity getFirstEntity() {
+        for (Map.Entry<String, Entity> entry : entities.entrySet()) {
+            return entry.getValue();
+        }
+
+        return null;
+    }
+
+    public Entity getEntity(String name) {
+        return MappingVisitorUtils.getEntity(this, name);
+    }
+
+    @Override
+    public boolean visit(SQLSelectItem x) {
+        x.getExpr().setParent(x);
+        return true;
+    }
+
+    @Override
+    public boolean visit(SQLServerSelectQueryBlock x) {
+        return MappingVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLSelectQueryBlock x) {
+        return MappingVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLIdentifierExpr x) {
+        return MappingVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLSubqueryTableSource x) {
+        return MappingVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLJoinTableSource x) {
+        return MappingVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLExprTableSource x) {
+        return MappingVisitorUtils.visit(this, x);
+    }
+
+}

src/main/java/com/alibaba/druid/mapping/spi/MappingProvider.java

@@ -1,5 +1,6 @@
-package com.alibaba.druid.mapping;
+package com.alibaba.druid.mapping.spi;
 
+import com.alibaba.druid.mapping.MappingEngine;
 import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
 import com.alibaba.druid.sql.visitor.SQLASTOutputVisitor;
 

src/main/java/com/alibaba/druid/mapping/spi/MappingVisitor.java

@@ -1,7 +1,8 @@
-package com.alibaba.druid.mapping;
+package com.alibaba.druid.mapping.spi;
 
 import java.util.Map;
 
+import com.alibaba.druid.mapping.Entity;
 import com.alibaba.druid.sql.ast.statement.SQLTableSource;
 import com.alibaba.druid.sql.visitor.SQLASTVisitor;
 

src/main/java/com/alibaba/druid/mapping/spi/MappingVisitorUtils.java

@@ -1,5 +1,10 @@
-package com.alibaba.druid.mapping;
+package com.alibaba.druid.mapping.spi;
 
+import java.util.Map;
+
+import com.alibaba.druid.mapping.DruidMappingException;
+import com.alibaba.druid.mapping.Entity;
+import com.alibaba.druid.mapping.Property;
 import com.alibaba.druid.sql.ast.SQLExpr;
 import com.alibaba.druid.sql.ast.expr.SQLAllColumnExpr;
 import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
@@ -9,6 +14,7 @@ import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
 import com.alibaba.druid.sql.ast.statement.SQLTableSource;
 
 public class MappingVisitorUtils {
+
     public static boolean visit(MappingVisitor visitor, SQLExprTableSource x) {
         SQLExpr expr = x.getExpr();
 
@@ -31,12 +37,12 @@ public class MappingVisitorUtils {
 
         return false;
     }
-    
+
     public static boolean visit(MappingVisitor visitor, SQLTableSource x) {
         if (x.getAlias() != null) {
             visitor.getTableSources().put(x.getAlias(), x);
         }
-        
+
         return true;
     }
 
@@ -100,4 +106,19 @@ public class MappingVisitorUtils {
 
         return true;
     }
+
+    public static Entity getEntity(MappingVisitor visitor, String name) {
+        Entity entity = visitor.getEntities().get(name);
+
+        if (entity == null) {
+            for (Map.Entry<String, Entity> entry : visitor.getEntities().entrySet()) {
+                if (entry.getKey().equalsIgnoreCase(name)) {
+                    entity = entry.getValue();
+                    break;
+                }
+            }
+        }
+
+        return entity;
+    }
 }

src/main/java/com/alibaba/druid/mapping/spi/MySqlMappingProvider.java

@@ -1,5 +1,6 @@
-package com.alibaba.druid.mapping;
+package com.alibaba.druid.mapping.spi;
 
+import com.alibaba.druid.mapping.MappingEngine;
 import com.alibaba.druid.sql.ast.expr.SQLNumberExpr;
 import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
 import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlSelectQueryBlock;

src/main/java/com/alibaba/druid/mapping/spi/MySqlMappingVisitor.java

@@ -1,8 +1,9 @@
-package com.alibaba.druid.mapping;
+package com.alibaba.druid.mapping.spi;
 
 import java.util.LinkedHashMap;
 import java.util.Map;
 
+import com.alibaba.druid.mapping.Entity;
 import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
 import com.alibaba.druid.sql.ast.statement.SQLExprTableSource;
 import com.alibaba.druid.sql.ast.statement.SQLJoinTableSource;
@@ -39,49 +40,41 @@ public class MySqlMappingVisitor extends MySqlASTVisitorAdapter implements Mappi
     }
 
     public Entity getEntity(String name) {
-        Entity entity = this.entities.get(name);
-
-        if (entity == null) {
-            for (Map.Entry<String, Entity> entry : entities.entrySet()) {
-                if (entry.getKey().equalsIgnoreCase(name)) {
-                    entity = entry.getValue();
-                    break;
-                }
-            }
-        }
-
-        return entity;
+        return MappingVisitorUtils.getEntity(this, name);
     }
 
+    @Override
     public boolean visit(SQLSelectItem x) {
         x.getExpr().setParent(x);
         return true;
     }
 
+    @Override
     public boolean visit(MySqlSelectQueryBlock x) {
         return MappingVisitorUtils.visit(this, x);
     }
 
+    @Override
     public boolean visit(SQLSelectQueryBlock x) {
         return MappingVisitorUtils.visit(this, x);
     }
 
-    protected void fillSelectList(SQLSelectQueryBlock x) {
-        MappingVisitorUtils.fillSelectList(this, x);
-    }
-
+    @Override
     public boolean visit(SQLIdentifierExpr x) {
         return MappingVisitorUtils.visit(this, x);
     }
 
+    @Override
     public boolean visit(SQLSubqueryTableSource x) {
         return MappingVisitorUtils.visit(this, x);
     }
 
+    @Override
     public boolean visit(SQLJoinTableSource x) {
         return MappingVisitorUtils.visit(this, x);
     }
 
+    @Override
     public boolean visit(SQLExprTableSource x) {
         return MappingVisitorUtils.visit(this, x);
     }

src/main/java/com/alibaba/druid/mapping/spi/OracleMappingProvider.java

@@ -1,5 +1,6 @@
-package com.alibaba.druid.mapping;
+package com.alibaba.druid.mapping.spi;
 
+import com.alibaba.druid.mapping.MappingEngine;
 import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
 import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleSelectQueryBlock;
 import com.alibaba.druid.sql.dialect.oracle.parser.OracleSelectParser;

src/main/java/com/alibaba/druid/mapping/spi/OracleMappingVisitor.java

@@ -0,0 +1,88 @@
+package com.alibaba.druid.mapping.spi;
+
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+import com.alibaba.druid.mapping.Entity;
+import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
+import com.alibaba.druid.sql.ast.statement.SQLExprTableSource;
+import com.alibaba.druid.sql.ast.statement.SQLJoinTableSource;
+import com.alibaba.druid.sql.ast.statement.SQLSelectItem;
+import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
+import com.alibaba.druid.sql.ast.statement.SQLSubqueryTableSource;
+import com.alibaba.druid.sql.ast.statement.SQLTableSource;
+import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleSelectQueryBlock;
+import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleSelectTableReference;
+import com.alibaba.druid.sql.dialect.oracle.visitor.OracleASTVIsitorAdapter;
+
+public class OracleMappingVisitor extends OracleASTVIsitorAdapter implements MappingVisitor {
+
+    private final LinkedHashMap<String, Entity> entities;
+    private final Map<String, SQLTableSource>   tableSources = new LinkedHashMap<String, SQLTableSource>();
+
+    public OracleMappingVisitor(LinkedHashMap<String, Entity> entities){
+        super();
+        this.entities = entities;
+    }
+
+    public LinkedHashMap<String, Entity> getEntities() {
+        return entities;
+    }
+
+    public Map<String, SQLTableSource> getTableSources() {
+        return tableSources;
+    }
+
+    public Entity getFirstEntity() {
+        for (Map.Entry<String, Entity> entry : entities.entrySet()) {
+            return entry.getValue();
+        }
+
+        return null;
+    }
+
+    public Entity getEntity(String name) {
+        return MappingVisitorUtils.getEntity(this, name);
+    }
+
+    @Override
+    public boolean visit(SQLSelectItem x) {
+        x.getExpr().setParent(x);
+        return true;
+    }
+
+    @Override
+    public boolean visit(OracleSelectQueryBlock x) {
+        return MappingVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLSelectQueryBlock x) {
+        return MappingVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLIdentifierExpr x) {
+        return MappingVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLSubqueryTableSource x) {
+        return MappingVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLJoinTableSource x) {
+        return MappingVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLExprTableSource x) {
+        return MappingVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(OracleSelectTableReference x) {
+        return MappingVisitorUtils.visit(this, x);
+    }
+}

src/main/java/com/alibaba/druid/sql/dialect/sqlserver/visitor/SQLServerASTVisitorAdapter.java

@@ -0,0 +1,29 @@
+package com.alibaba.druid.sql.dialect.sqlserver.visitor;
+
+import com.alibaba.druid.sql.dialect.sqlserver.ast.SQLServerSelectQueryBlock;
+import com.alibaba.druid.sql.dialect.sqlserver.ast.Top;
+import com.alibaba.druid.sql.visitor.SQLASTVisitorAdapter;
+
+public class SQLServerASTVisitorAdapter extends SQLASTVisitorAdapter implements SQLServerASTVisitor {
+
+    @Override
+    public boolean visit(SQLServerSelectQueryBlock x) {
+        return true;
+    }
+
+    @Override
+    public void endVisit(SQLServerSelectQueryBlock x) {
+
+    }
+
+    @Override
+    public boolean visit(Top x) {
+        return true;
+    }
+
+    @Override
+    public void endVisit(Top x) {
+
+    }
+
+}