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

[improvement](chat) Add a LLMParserLayer to interact with a Python service and add comments to certain classes (#388)

Browse Source
This commit is contained in:
lexluo09
2023-11-15 14:49:22 +08:00
committed by GitHub
parent 7ef3d92f2c
commit aa448b1ba3
11 changed files with 101 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
common/src/main/java/com/tencent/supersonic/common/util/jsqlparser/SqlParserSelectHelper.java
View File

@@ -166,7 +166,7 @@ public class SqlParserSelectHelper {
return new ArrayList<>(results);
}
private static ArrayList<String> getFieldsByPlainSelect(PlainSelect plainSelect) {
private static List<String> getFieldsByPlainSelect(PlainSelect plainSelect) {
if (Objects.isNull(plainSelect)) {
return new ArrayList<>();
}
@@ -396,9 +396,7 @@ public class SqlParserSelectHelper {
}
SelectBody selectBody = selectStatement.getSelectBody();
PlainSelect plainSelect = (PlainSelect) selectBody;
Table table = (Table) plainSelect.getFromItem();
return table;
return (Table) plainSelect.getFromItem();
}
public static String getDbTableName(String sql) {
@@ -406,5 +404,12 @@ public class SqlParserSelectHelper {
return table.getFullyQualifiedName();
}
public static String getNormalizedSql(String sql) {
Select selectStatement = getSelect(sql);
if (selectStatement == null) {
return null;
}
return selectStatement.toString();
}
}