Question

我有一个SQL查询：

select 
    t1.name, t2.address 
from 
    Table1 t1 
inner join 
    Table2 t2 on t1.id = t2.id;

和地图：

Map<String,String> map = new HashMap<String,String>();
map.put("testTable", "hive.DB1");
map.put("testTable", "mongo.DB2");

我只想使用此地图将上面提到的查询转换为：

select 
    t1.name, t2.address 
from 
    hive.DB1.`Table1` t1 
inner join 
    mongo.DB2.`Table2` t2 on t1.id = t2.id;

哪种开源SQL解析器适合此目的。

Answer 1

你真的需要一个SQL Parser吗？为什么不使用sed？

一张桌子的前言：

sed -e＆＃34; s / Table1 / hive.DB1.TABLE1 / g＆＃34; C：\ test.sql＆gt; C：\ new.sql

Answer 2

我使用JSqlParser。

我从SQL查询中提取了表名：

Statement statement = CCJSqlParserUtil.parse("select t1.name, t2.address from Table1 t1 inner join Table2 t2 on t1.id = t2.id;");
Select selectStatement = (Select) statement;
TablesNamesFinder tablesNamesFinder = new TablesNamesFinder();
List<String> tableList = tablesNamesFinder.getTableList(selectStatement);

然后我根据地图修改了表名。

Answer 3

需要解析器才能从“from”部分更改表引用。某些数据库不接受表达式中的完全限定名称。

此代码使用JSqlParser 0.9.5-SNAPSHOT并打印出已修改的SQL：

public class SimpleTableReplacer {
    public static void main(String args[]) throws JSQLParserException {
        String sql = "select t1.name, t2.address from Table1 t1 inner join Table2 t2 on t1.id = t2.id";
        Select select = (Select)CCJSqlParserUtil.parse(sql);

        //Replace Table1 with hive.DB1.Table1 and Table2 with mongo.DB2.Table2
        StringBuilder buffer = new StringBuilder();
        ExpressionDeParser expressionDeParser = new ExpressionDeParser();
        SelectDeParser deparser = new SelectDeParser(expressionDeParser,buffer ) {
            @Override
            public void visit(Table tableName) {
                switch (tableName.getName()) {
                    case "Table1": getBuffer().append("hive.DB1.Table1").append(' ').append(tableName.getAlias().getName());break;
                    case "Table2": getBuffer().append("mongo.DB2.Table2").append(' ').append(tableName.getAlias().getName());break;
                    default:
                        super.visit(tableName);
                }
            }
        };
        expressionDeParser.setSelectVisitor(deparser);
        expressionDeParser.setBuffer(buffer);
        select.getSelectBody().accept(deparser);
        //End of value modification

        System.out.println(buffer.toString());
    }
}

结果：SELECT t1.name, t2.address FROM hive.DB1.Table1 t1 INNER JOIN mongo.DB2.Table2 t2 ON t1.id = t2.id。

当然，您可以使用此代码修改类层次结构本身，这意味着更改表 - 对象名称。

此外，您可以使用JSqlParser的新功能为SQL的某些部分提供AST节点。您可以在SQL中提取表名的确切位置，并在那里替换文本。这可以用这种方式编码：

SimpleNode node = (SimpleNode) CCJSqlParserUtil.parseAST(sql);
    node.childrenAccept(new CCJSqlParserDefaultVisitor() {
        @Override
        public Object visit(SimpleNode node, Object data) {
            if (node.getId() == CCJSqlParserTreeConstants.JJTTABLE) {
                System.out.println("table name '" + node.jjtGetValue() + "' found at " + node.jjtGetFirstToken().beginColumn + "-" + node.jjtGetLastToken().endColumn);
            }
            return super.visit(node, data); 
        }
    }, null);

Answer 4

如果您只需要从SQL中提取表名并执行某些逻辑，则需要使用超轻，超快的库来执行此操作

只需在你的pom中添加以下内容

即可

<dependency>
    <groupId>com.github.mnadeem</groupId>
    <artifactId>sql-table-name-parser</artifactId>
    <version>0.0.1</version>
</dependency>

并使用以下说明

new TableNameParser(sql).tables()

有关详细信息，请参阅project

免责声明：我是所有者

SQL解析器改变表名

4 个答案: