Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

add mapping support

Browse Source
This commit is contained in:
温少
2012-03-12 20:44:25 +08:00
parent f893c1c5c9
commit ea50c9919d
5 changed files with 117 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/mapping/MappingEngine.java
+21 -33
View File
@@ -2,17 +2,14 @@ package com.alibaba.druid.mapping;
import java.util.LinkedHashMap;
import com.alibaba.druid.sql.SQLUtils;
import com.alibaba.druid.sql.ast.expr.SQLNumberExpr;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlSelectQueryBlock;
import com.alibaba.druid.sql.dialect.mysql.ast.statement.MySqlSelectQueryBlock.Limit;
import com.alibaba.druid.sql.dialect.mysql.parser.MySqlSelectParser;
import com.alibaba.druid.sql.visitor.SQLASTVisitor;
import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
import com.alibaba.druid.sql.visitor.SQLASTOutputVisitor;
public class MappingEngine {
private LinkedHashMap<String, Entity> entities = new LinkedHashMap<String, Entity>();
private Integer maxLimit;
private MappingProvider provider;
public Integer getMaxLimit() {
return maxLimit;
@@ -30,36 +27,27 @@ public class MappingEngine {
this.entities.put(entity.getName(), entity);
}
public SQLASTVisitor createMySqlMappingVisitor() {
return new MySqlMappingVisitor(entities);
public MappingVisitor createMappingVisitor() {
return provider.createMappingVisitor(this);
}
public String explainToMySql(String sql) {
MySqlSelectQueryBlock query = explainToMySqlObject(sql);
query.accept(this.createMySqlMappingVisitor());
return SQLUtils.toMySqlString(query);
public SQLASTOutputVisitor createOutputVisitor(Appendable out) {
return provider.createOutputVisitor(this, out);
}
public SQLSelectQueryBlock explainToSQLObject(String sql) {
return provider.explainToSQLObject(this, sql);
}
public MySqlSelectQueryBlock explainToMySqlObject(String sql) {
MySqlSelectParser selectParser = new MySqlSelectParser(sql);
MySqlSelectQueryBlock query = (MySqlSelectQueryBlock) selectParser.query();
if (this.maxLimit != null) {
if (query.getLimit() == null) {
Limit limit = new Limit();
limit.setRowCount(new SQLNumberExpr(maxLimit));
query.setLimit(limit);
} else {
SQLNumberExpr rowCountExpr = (SQLNumberExpr) query.getLimit().getRowCount();
int rowCount = rowCountExpr.getNumber().intValue();
if (rowCount > maxLimit.intValue()) {
rowCountExpr.setNumber(maxLimit);
}
}
}
return query;
public String explainToSQL(String sql) {
SQLSelectQueryBlock query = explainToSQLObject(sql);
query.accept(this.createMappingVisitor());
StringBuilder out = new StringBuilder();
SQLASTOutputVisitor outputVisitor = createOutputVisitor(out);
query.accept(outputVisitor);
return out.toString();
}
}