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 queryData by dimension error (#1666)

Browse Source
This commit is contained in:
jipeli
2024-09-13 14:37:14 +08:00
committed by GitHub
parent e901ff8370
commit 5f4d24348a
3 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/ChatQueryServiceImpl.java
View File

@@ -200,6 +200,7 @@ public class ChatQueryServiceImpl implements ChatQueryService {
SemanticParseInfo parseInfo = SemanticParseInfo parseInfo =
chatManageService.getParseInfo(chatQueryDataReq.getQueryId(), parseId); chatManageService.getParseInfo(chatQueryDataReq.getQueryId(), parseId);
parseInfo = mergeParseInfo(parseInfo, chatQueryDataReq); parseInfo = mergeParseInfo(parseInfo, chatQueryDataReq);
parseInfo.setSqlInfo(new SqlInfo());
DataSetSchema dataSetSchema = DataSetSchema dataSetSchema =
semanticLayerService.getDataSetSchema(parseInfo.getDataSetId()); semanticLayerService.getDataSetSchema(parseInfo.getDataSetId());

4
headless/core/src/main/java/com/tencent/supersonic/headless/core/translator/DefaultSemanticTranslator.java
View File

@@ -89,9 +89,7 @@ public class DefaultSemanticTranslator implements SemanticTranslator {
QueryStatement tableSql = QueryStatement tableSql =
parserSql( parserSql(
metricTable, isSingleTable, dataSetQueryParam, queryStatement); metricTable, isSingleTable, dataSetQueryParam, queryStatement);
if (isSingleTable if (isSingleTable && StringUtils.isNotBlank(tableSql.getDataSetSimplifySql())) {
&& Objects.nonNull(tableSql.getDataSetQueryParam())
&& !tableSql.getDataSetSimplifySql().isEmpty()) {
queryStatement.setSql(tableSql.getDataSetSimplifySql()); queryStatement.setSql(tableSql.getDataSetSimplifySql());
queryStatement.setDataSetQueryParam(dataSetQueryParam); queryStatement.setDataSetQueryParam(dataSetQueryParam);
return queryStatement; return queryStatement;

4
launchers/standalone/src/test/java/com/tencent/supersonic/headless/QueryBySqlTest.java
View File

@@ -129,6 +129,8 @@ public class QueryBySqlTest extends BaseTest {
SemanticQueryResp semanticQueryResp = SemanticQueryResp semanticQueryResp =
queryBySql("SELECT SUM(pv) FROM 超音数PVUV统计 WHERE department ='HR'", tom); queryBySql("SELECT SUM(pv) FROM 超音数PVUV统计 WHERE department ='HR'", tom);
Assertions.assertNotNull(semanticQueryResp.getQueryAuthorization().getMessage()); Assertions.assertNotNull(semanticQueryResp.getQueryAuthorization().getMessage());
Assertions.assertTrue(semanticQueryResp.getSql().contains("user_name = 'tom'")); Assertions.assertTrue(
semanticQueryResp.getSql().contains("user_name = 'tom'")
|| semanticQueryResp.getSql().contains("`user_name` = 'tom'"));
} }
} }