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

(improvement)(Headless) Obtain metric default agg (#984) (#985)

Browse Source
This commit is contained in:
jipeli
2024-05-13 12:05:02 +08:00
committed by GitHub
parent 210591e28f
commit 947a01e8ba
2 changed files with 64 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
common/src/main/java/com/tencent/supersonic/common/util/jsqlparser/SqlSelectFunctionHelper.java
View File

@@ -1,11 +1,14 @@
package com.tencent.supersonic.common.util.jsqlparser;
import com.tencent.supersonic.common.pojo.enums.AggOperatorEnum;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;
import lombok.extern.slf4j.Slf4j;
import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.expression.Function;
@@ -23,6 +26,8 @@ import org.springframework.util.CollectionUtils;
@Slf4j
public class SqlSelectFunctionHelper {
public static List<String> aggregateFunctionName = Arrays.asList("SUM", "COUNT", "MAX", "MIN", "AVG");
public static boolean hasAggregateFunction(String sql) {
if (!CollectionUtils.isEmpty(getFunctions(sql))) {
return true;
@@ -84,5 +89,22 @@ public class SqlSelectFunctionHelper {
return sumFunction;
}
public static String getFirstAggregateFunctions(String expr) {
List<String> functions = getAggregateFunctions(expr);
return CollectionUtils.isEmpty(functions) ? "" : functions.get(0);
}
public static List<String> getAggregateFunctions(String expr) {
Expression expression = QueryExpressionReplaceVisitor.getExpression(expr);
if (Objects.nonNull(expression)) {
FunctionVisitor visitor = new FunctionVisitor();
expression.accept(visitor);
Set<String> functions = visitor.getFunctionNames();
return functions.stream()
.filter(t -> aggregateFunctionName.contains(t.toUpperCase())).collect(Collectors.toList());
}
return new ArrayList<>();
}
}