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

create aliasQueryMap to store alias->expression, in case null returned in alias map

Browse Source
This commit is contained in:
Louis Xu
2016-04-21 17:24:22 +08:00
parent 098f603280
commit 48cf09635b
1 changed files with 26 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/dialect/teradata/visitor/TeradataSchemaStatVisitor.java
+26 -12
View File
@@ -15,10 +15,13 @@
*/
package com.alibaba.druid.sql.dialect.teradata.visitor;
import java.util.LinkedHashMap;
import java.util.Map;
import com.alibaba.druid.sql.ast.SQLName;
import com.alibaba.druid.sql.ast.SQLObject;
import com.alibaba.druid.sql.ast.expr.SQLAggregateExpr;
import com.alibaba.druid.sql.ast.expr.SQLCaseExpr;
import com.alibaba.druid.sql.ast.expr.SQLMethodInvokeExpr;
import com.alibaba.druid.sql.ast.statement.SQLSelectItem;
import com.alibaba.druid.sql.ast.statement.SQLSelectQuery;
@@ -32,6 +35,8 @@ import com.alibaba.druid.util.JdbcUtils;
public class TeradataSchemaStatVisitor extends SchemaStatVisitor implements TeradataASTVisitor {
protected final Map<String, SQLObject> aliasQueryMap = new LinkedHashMap<String, SQLObject>();
@Override
public String getDbType() {
return JdbcUtils.TERADATA;
@@ -98,25 +103,34 @@ public class TeradataSchemaStatVisitor extends SchemaStatVisitor implements Tera
if (alias != null && (!alias.isEmpty()) && aliasMap != null) {
if (x.getExpr() instanceof SQLName) {
putAliasMap(aliasMap, alias, x.getExpr().toString());
} else if (x.getExpr() instanceof SQLAggregateExpr
|| x.getExpr() instanceof SQLMethodInvokeExpr) {
} else {
TeradataSchemaStatVisitor visitor = new TeradataSchemaStatVisitor();
x.getExpr().accept(visitor);
for (Column col : visitor.getColumns()) {
if (col.getTable().equalsIgnoreCase("unknown")
|| this.getCurrentTable() != null) {
col.setTable(this.currentTable);
}
putAliasMap(aliasMap, alias, col.toString());
}
}
else {
putAliasMap(aliasMap, alias, null);
putAliasMap(aliasMap, alias, null);
// Add expression alias into aliasQuery map
// in order to retrieve column info from aliasQueryMap
// if aliasMap value is null
String uniqueAliasName = this.getCurrentTable() == null ?
alias : this.getCurrentTable() + "." + alias;
addAliasQuery(uniqueAliasName, x.getExpr());
}
}
return false;
}
protected void addAliasQuery(String alias, SQLObject query) {
String alias_lcase = alias.toLowerCase();
aliasQueryMap.put(alias_lcase, query);
}
public SQLObject getAliasQuery(String alias) {
String alias_lcase = alias.toLowerCase();
return aliasQueryMap.get(alias_lcase);
}
public Map<String, SQLObject> getAliasQueryMap() {
return aliasQueryMap;
}
}