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

(improvement)(chat) dsl parser and corrector support having function (#94)

Browse Source
This commit is contained in:
lexluo09
2023-09-15 15:38:33 +08:00
committed by GitHub
parent 45ed5648c4
commit 3701ade05f
4 changed files with 34 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -117,9 +117,19 @@ public class SqlParserSelectHelper {
getWhereFields(plainSelect, result);
getHavingFields(plainSelect, result);
return new ArrayList<>(result);
}
private static void getHavingFields(PlainSelect plainSelect, Set<String> result) {
Expression having = plainSelect.getHaving();
if (Objects.nonNull(having)) {
having.accept(new FieldAcquireVisitor(result));
}
}
public static List<String> getOrderByFields(String sql) {
PlainSelect plainSelect = getPlainSelect(sql);
if (Objects.isNull(plainSelect)) {

5
common/src/main/java/com/tencent/supersonic/common/util/jsqlparser/SqlParserUpdateHelper.java
View File

@@ -94,6 +94,11 @@ public class SqlParserUpdateHelper {
if (Objects.nonNull(groupByElement)) {
groupByElement.accept(new GroupByReplaceVisitor(fieldToBizName, exactReplace));
}
//5. replace having fields
Expression having = plainSelect.getHaving();
if (Objects.nonNull(having)) {
having.accept(visitor);
}
return selectStatement.toString();
}