topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-12 04:27:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[improvement][headless] Fix the issue with the query when using the with clause along with a join (#1779)

Browse Source
This commit is contained in:
lexluo09
2024-10-10 21:53:29 +08:00
committed by GitHub
parent bf1f2ed60a
commit 484861571f
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
common/src/main/java/com/tencent/supersonic/common/jsqlparser/SqlReplaceHelper.java
View File

@@ -439,16 +439,18 @@ public class SqlReplaceHelper {
private static void replaceJoins(PlainSelect plainSelect, String tableName, private static void replaceJoins(PlainSelect plainSelect, String tableName,
List<String> withNameList) { List<String> withNameList) {
List<Join> joins = plainSelect.getJoins(); List<Join> joins = plainSelect.getJoins();
TableNameReplaceVisitor fromItemVisitor =
new TableNameReplaceVisitor(tableName, new HashSet<>(withNameList));
if (!CollectionUtils.isEmpty(joins)) { if (!CollectionUtils.isEmpty(joins)) {
for (Join join : joins) { for (Join join : joins) {
if (join.getRightItem() instanceof ParenthesedFromItem) { if (join.getRightItem() instanceof ParenthesedFromItem) {
List<PlainSelect> subPlainSelects = SqlSelectHelper.getPlainSelects( List<PlainSelect> subPlainSelects = SqlSelectHelper.getPlainSelects(
Collections.singletonList((PlainSelect) join.getRightItem())); Collections.singletonList((PlainSelect) join.getRightItem()));
subPlainSelects.forEach(subPlainSelect -> subPlainSelect.getFromItem().accept( subPlainSelects.forEach(subPlainSelect -> {
new TableNameReplaceVisitor(tableName, new HashSet<>(withNameList)))); subPlainSelect.getFromItem().accept(fromItemVisitor);
});
} else if (join.getRightItem() instanceof Table) { } else if (join.getRightItem() instanceof Table) {
Table table = (Table) join.getRightItem(); join.getRightItem().accept(fromItemVisitor);
table.setName(tableName);
} }
} }
} }