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

(improvement)(headless)Remove tag-related logic when determining query type.

Browse Source
This commit is contained in:
jerryjzhang
2024-08-24 09:47:42 +08:00
parent bef652892b
commit 5229fdc8b5
1 changed files with 3 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/QueryTypeParser.java
View File

@@ -52,6 +52,7 @@ public class QueryTypeParser implements SemanticParser {
if (Objects.isNull(sqlInfo) || StringUtils.isBlank(sqlInfo.getParsedS2SQL())) { if (Objects.isNull(sqlInfo) || StringUtils.isBlank(sqlInfo.getParsedS2SQL())) {
return QueryType.DETAIL; return QueryType.DETAIL;
} }
//1. entity queryType //1. entity queryType
Long dataSetId = parseInfo.getDataSetId(); Long dataSetId = parseInfo.getDataSetId();
SemanticSchema semanticSchema = chatQueryContext.getSemanticSchema(); SemanticSchema semanticSchema = chatQueryContext.getSemanticSchema();
@@ -66,23 +67,13 @@ public class QueryTypeParser implements SemanticParser {
return QueryType.ID; return QueryType.ID;
} }
} }
List<String> selectFields = SqlSelectHelper.getSelectFields(sqlInfo.getParsedS2SQL());
selectFields.addAll(whereFields);
List<String> selectWhereFilterByTimeFields = filterByTimeFields(selectFields);
if (CollectionUtils.isNotEmpty(selectWhereFilterByTimeFields)) {
Set<String> tags = semanticSchema.getTags(dataSetId).stream().map(SchemaElement::getName)
.collect(Collectors.toSet());
//If all the fields in the SELECT/WHERE statement are of tag type.
if (CollectionUtils.isNotEmpty(tags)
&& tags.containsAll(selectWhereFilterByTimeFields)) {
return QueryType.DETAIL;
}
}
} }
//2. metric queryType //2. metric queryType
if (selectContainsMetric(sqlInfo, dataSetId, semanticSchema)) { if (selectContainsMetric(sqlInfo, dataSetId, semanticSchema)) {
return QueryType.METRIC; return QueryType.METRIC;
} }
return QueryType.DETAIL; return QueryType.DETAIL;
} }