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)(headless) Supplement the model name when obtaining metric information (#612)

Browse Source 
Co-authored-by: jolunoluo
This commit is contained in:
LXW
2024-01-10 10:13:45 +08:00
committed by GitHub
parent 87a60eeba2
commit 61da52650c
2 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/HeadlessQueryEngineImpl.java
View File

@@ -99,10 +99,11 @@ public class HeadlessQueryEngineImpl implements HeadlessQueryEngine {
}
private HeadlessModel getHeadLessModel(QueryStatement queryStatement) throws Exception {
HeadlessModel headlessModel = headlessSchemaManager.get(queryStatement.getQueryStructReq().getModelIdStr());
ItemDateResp itemDateResp = queryStructUtils.getItemDateResp(queryStatement.getQueryStructReq());
QueryStructReq queryStructReq = queryStatement.getQueryStructReq();
HeadlessModel headlessModel = headlessSchemaManager.get(queryStructReq.getModelIdStr());
ItemDateResp itemDateResp = queryStructUtils.getItemDateResp(queryStructReq);
headlessModel.setDataDate(itemDateResp);
List<ModelSchemaResp> modelSchemaResps = catalog.getModelSchema(queryStatement.getModelIds());
List<ModelSchemaResp> modelSchemaResps = catalog.getModelSchema(queryStructReq.getModelIds());
headlessModel.setModelSchemaResps(modelSchemaResps);
return headlessModel;
}

8
headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/MetricServiceImpl.java
View File

@@ -235,10 +235,14 @@ public class MetricServiceImpl implements MetricService {
@Override
public MetricResp getMetric(Long id, User user) {
MetricResp metricResp = getMetric(id);
if (metricResp == null) {
MetricDO metricDO = metricRepository.getMetricById(id);
if (metricDO == null) {
return null;
}
Map<Long, ModelResp> modelMap = modelService.getModelMap();
List<CollectDO> collectList = collectService.getCollectList(user.getName());
List<Long> collect = collectList.stream().map(CollectDO::getCollectId).collect(Collectors.toList());
MetricResp metricResp = MetricConverter.convert2MetricResp(metricDO, modelMap, collect);
fillAdminRes(Lists.newArrayList(metricResp), user);
return metricResp;
}