Java Code Examples for com.alibaba.druid.sql.dialect.mysql.visitor.MySqlOutputVisitor

The following examples show how to use com.alibaba.druid.sql.dialect.mysql.visitor.MySqlOutputVisitor. These examples are extracted from open source projects. You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
Source Project: dts   Source File: SelectParser.java    License: Apache License 2.0 6 votes vote down vote up
@Override
protected String getTableName(SQLSelectStatement parseSqlStatement) {
    SQLSelectQueryBlock selectQueryBlock = parseSqlStatement.getSelect().getQueryBlock();
    SQLTableSource tableSource = selectQueryBlock.getFrom();
    StringBuffer sb = new StringBuffer();
    MySqlOutputVisitor visitor = new MySqlOutputVisitor(sb) {

        @Override
        public boolean visit(SQLExprTableSource x) {
            printTableSourceExpr(x.getExpr());
            return false;
        }
    };
    visitor.visit((SQLExprTableSource)tableSource);
    return sb.toString();
}
 
Example 2
Source Project: dble   Source File: MysqlVisitor.java    License: GNU General Public License v2.0 6 votes vote down vote up
void buildTableName(TableNode tableNode, StringBuilder sb) {
    String tableName = "`" + tableNode.getPureName() + "`";
    String fullName = "`" + tableNode.getPureSchema() + "`." + tableName;
    mapTableToSimple.put(fullName, tableName);
    sb.append(" ").append(fullName);
    String alias = tableNode.getAlias();
    if (alias != null) {
        sb.append(" `").append(alias).append("`");
    }
    List<SQLHint> hintList = tableNode.getHintList();
    if (hintList != null && !hintList.isEmpty()) {
        sb.append(' ');
        boolean isFirst = true;
        for (SQLHint hint : hintList) {
            if (isFirst)
                isFirst = false;
            else
                sb.append(" ");
            MySqlOutputVisitor ov = new MySqlOutputVisitor(sb);
            ov.setShardingSupport(false);
            hint.accept(ov);
        }
    }
}
 
Example 3
Source Project: baymax   Source File: MySqlSqlParser.java    License: Apache License 2.0 6 votes vote down vote up
@Override
public void changeSql(ParseResult result, ExecutePlan plan) {
    if (ExecuteType.NO == plan.getExecuteType()){
    }else {
        for (TrargetSqlEntity sql : plan.getSqlList()){
            ReplaceTableNameVisitor replaceVisitor = new ReplaceTableNameVisitor(sql.getLogicTableName(), sql.getTargetTableName());
            StringBuilder out = new StringBuilder();
            MySqlOutputVisitor outPutVisitor = new MySqlOutputVisitor(out);
            // 替换表名
            statement.accept(replaceVisitor);
            // 输出sql
            statement.accept(outPutVisitor);
            sql.setOriginalSql(result.getSql());
            sql.setTargetSql(out.toString());
            // 输出sql后要还原statement以便下次替换表名
            replaceVisitor.reset();
        }
    }
}
 
Example 4
Source Project: baymax   Source File: ReplaceTableNameVisitorTest.java    License: Apache License 2.0 6 votes vote down vote up
public void test(String sql, String logicName, String targetName){

        MySqlStatementParser parser = new MySqlStatementParser(sql);
        SQLStatement statement = parser.parseStatement();

        ReplaceTableNameVisitor replaceVisitor = new ReplaceTableNameVisitor(logicName, targetName);
        StringBuilder out = new StringBuilder();
        MySqlOutputVisitor outPutVisitor = new MySqlOutputVisitor(out);
        // 替换表名
        statement.accept(replaceVisitor);
        // 输出sql
        statement.accept(outPutVisitor);

        System.out.println();
        System.out.println(sql);
        System.out.println(out.toString());
        // 输出sql后要还原statement以便下次替换表名
        replaceVisitor.reset();
    }
 
Example 5
Source Project: dble   Source File: MySQLItemVisitor.java    License: GNU General Public License v2.0 5 votes vote down vote up
private void initName(SQLExpr expr) {
    StringBuilder sb = new StringBuilder();
    MySqlOutputVisitor ov = new MySqlOutputVisitor(sb);
    ov.setShardingSupport(false);
    expr.accept(ov);
    item.setItemName(sb.toString());
}
 
Example 6
Source Project: dble   Source File: Item.java    License: GNU General Public License v2.0 5 votes vote down vote up
public final String getItemName() {
    if (itemName == null || itemName.length() == 0) {
        SQLExpr expr = toExpression();
        StringBuilder sb = new StringBuilder();
        MySqlOutputVisitor ov = new MySqlOutputVisitor(sb);
        ov.setShardingSupport(false);
        expr.accept(ov);
        itemName = sb.toString();
    }
    return itemName;
}
 
Example 7
Source Project: dble   Source File: DefaultDruidParser.java    License: GNU General Public License v2.0 5 votes vote down vote up
String statementToString(SQLStatement statement) {
    StringBuffer buf = new StringBuffer();
    MySqlOutputVisitor visitor = new MySqlOutputVisitor(buf);
    visitor.setShardingSupport(false);
    statement.accept(visitor);
    return buf.toString();
}
 
Example 8
Source Project: baymax   Source File: MySqlSelectAggParserTest.java    License: Apache License 2.0 5 votes vote down vote up
/**
 * //agg
 * getMergeColumns
 * getAliaColumns
 *
 * //group by
 * setGroupbyColumns
 *
 * @param sql
 */
public void test(String sql){
    ParseResult result = new ParseResult();
    ExecutePlan plan = new ExecutePlan();

    parser.init(sql, null);
    parser.parse(result);

    plan.setExecuteType(ExecuteType.PARTITION);

    SQLSelectStatement stmt = (SQLSelectStatement) parser.statement;

    parser.parseMysqlQueary(result, plan, (MySqlSelectQueryBlock) stmt.getSelect().getQuery());

    StringBuilder out = new StringBuilder();
    MySqlOutputVisitor outPutVisitor = new MySqlOutputVisitor(out);
    stmt.accept(outPutVisitor);

    System.out.println();
    System.out.println("/***********************agg*************************/");
    System.out.println(plan.getMergeColumns());
    System.out.println("setHasAllColumnExpr:" + result.isHasAllColumnExpr());
    System.out.println(out.toString());
    PrintUtil.printFildAlisMap(result.getAliaColumns());
    System.out.println("/*********************group by**********************/");
    System.out.println(plan.getGroupbyColumns());
    System.out.println("/*********************order by**********************/");
    System.out.println(plan.getOrderbyColumns());

}