Question

我正在使用jsqlparser解析SQL字符串并替换字符串中的表名称。
我的输入是

 SELECT id, test 
 FROM test1 JOIN test2 
 ON test1.aa = test2.bb 
 WHERE test1.conf = \"test\"
 LIMIT 10"

我的目标输出是

SELECT id, test 
FROM test1_suffix 
JOIN test2_suffix 
ON test1_suffix.aa = test2_suffix.bb 
WHERE test1_suffix.conf = \"test\"
LIMIT 10"

我设法通过扩展TablesNamesFinder来替换表名，但这给了我：

SELECT id, test
FROM test1_suffix 
JOIN test2_suffix 
ON test1.aa = test2.bb 
WHERE test1.conf = \"test\"
LIMIT 10

我说那是完成工作的一半，但是我怎么做剩下的工作呢？

Answer 1

在解析时添加一个别名，代替表名。

SELECT * 
FROM test a
WHERE a.conf = 'something'

然后应将其更改为，即where子句可以相同

SELECT * 
FROM test_suffix a
WHERE a.conf = 'something'

Answer 2

因此，这是一个完整的（希望）示例，可以替换表名称的所有出现。问题在于， JSqlParser 在别名和表名之间没有区别。如果您不想更正这些别名，则必须有一些逻辑来跳过它们的sql。

使用 TableNamesFinder 并不能完成全部任务，因为它仅遍历AST并查找表名并随后停止。这就是为什么我的示例使用解析器。

此代码将转换

select id, test from test where name = "test"

到

SELECT id, test FROM test_mytest WHERE name = "test"

和

select * from t2 join t1 on t1.aa = t2.bb where t1.a = "someCondition" limit 10

到

SELECT * FROM t2_mytest JOIN t1_mytest ON t1_mytest.aa = t2_mytest.bb WHERE t1_mytest.a = "someCondition" LIMIT 10

我认为这可以解决您的问题。

public class SimpleSqlParser24 {

    public static void main(String args[]) throws JSQLParserException {
        replaceTableName("select id, test from test where name = \"test\"");
        replaceTableName("select * from t2 join t1 on t1.aa = t2.bb where t1.a = \"someCondition\" limit 10");
    }

    private static void replaceTableName(String sql) throws JSQLParserException {
        Select select = (Select) CCJSqlParserUtil.parse(sql);

        StringBuilder buffer = new StringBuilder();
        ExpressionDeParser expressionDeParser = new ExpressionDeParser() {
            @Override
            public void visit(Column tableColumn) {
                if (tableColumn.getTable() != null) {
                    tableColumn.getTable().setName(tableColumn.getTable().getName() + "_mytest");
                }
                super.visit(tableColumn);
            }

        };
        SelectDeParser deparser = new SelectDeParser(expressionDeParser, buffer) {
            @Override
            public void visit(Table tableName) {
                tableName.setName(tableName.getName() + "_mytest");
                super.visit(tableName);
            }
        };
        expressionDeParser.setSelectVisitor(deparser);
        expressionDeParser.setBuffer(buffer);
        select.getSelectBody().accept(deparser);

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

使用jsqlparser

2 个答案: