4 Commits

Author SHA1 Message Date
QJ_wonder
c75233e37f (fix)(chat) 修复插件功能,无法调用/插件结果被NL2SQL结果覆盖的问题 (#2277)
Some checks failed
supersonic CentOS CI / build (21) (push) Has been cancelled
supersonic mac CI / build (21) (push) Has been cancelled
supersonic ubuntu CI / build (21) (push) Has been cancelled
supersonic windows CI / build (21) (push) Has been cancelled
2025-07-27 09:03:26 +08:00
feelshana
785bda6cd9 (fix)[chat&headless]in mapping phase,DimValuesAlias do not take effect (#2340) 2025-07-27 09:03:05 +08:00
WDEP
6bd8970849 Fix model field update (#2343) 2025-07-27 09:02:53 +08:00
wangyong
c33a85b583 (fix)(webapp)修复问答对话右侧历史对话模块样式异常问题 (#2346) 2025-07-27 09:02:39 +08:00
27 changed files with 555 additions and 315 deletions

View File

@@ -88,10 +88,10 @@ public class WebServiceQuery extends PluginSemanticQuery {
restTemplate = ContextUtils.getBean(RestTemplate.class); restTemplate = ContextUtils.getBean(RestTemplate.class);
try { try {
responseEntity = responseEntity =
restTemplate.exchange(requestUrl, HttpMethod.POST, entity, Object.class); restTemplate.exchange(requestUrl, HttpMethod.POST, entity, String.class);
objectResponse = responseEntity.getBody(); objectResponse = responseEntity.getBody();
log.info("objectResponse:{}", objectResponse); log.info("objectResponse:{}", objectResponse);
Map<String, Object> response = JsonUtil.objectToMap(objectResponse); Map<String, Object> response = JSON.parseObject(objectResponse.toString());
webServiceResponse.setResult(response); webServiceResponse.setResult(response);
} catch (Exception e) { } catch (Exception e) {
log.info("Exception:{}", e.getMessage()); log.info("Exception:{}", e.getMessage());

View File

@@ -19,7 +19,7 @@ public class ParseContext {
} }
public boolean enableNL2SQL() { public boolean enableNL2SQL() {
return Objects.nonNull(agent) && agent.containsDatasetTool(); return Objects.nonNull(agent) && agent.containsDatasetTool()&&response.getSelectedParses().size() == 0;
} }
public boolean enableLLM() { public boolean enableLLM() {

View File

@@ -168,7 +168,6 @@ public class EmbeddingMatchStrategy extends BatchMatchStrategy<EmbeddingResult>
variable.put("retrievedInfo", JSONObject.toJSONString(results)); variable.put("retrievedInfo", JSONObject.toJSONString(results));
Prompt prompt = PromptTemplate.from(LLM_FILTER_PROMPT).apply(variable); Prompt prompt = PromptTemplate.from(LLM_FILTER_PROMPT).apply(variable);
ChatModelConfig chatModelConfig=null; ChatModelConfig chatModelConfig=null;
if(chatQueryContext.getRequest().getChatAppConfig()!=null if(chatQueryContext.getRequest().getChatAppConfig()!=null
&& chatQueryContext.getRequest().getChatAppConfig().containsKey("REWRITE_MULTI_TURN")){ && chatQueryContext.getRequest().getChatAppConfig().containsKey("REWRITE_MULTI_TURN")){

View File

@@ -1,10 +1,7 @@
package com.tencent.supersonic.headless.chat.mapper; package com.tencent.supersonic.headless.chat.mapper;
import com.tencent.supersonic.common.util.ContextUtils; import com.tencent.supersonic.common.util.ContextUtils;
import com.tencent.supersonic.headless.api.pojo.SchemaElement; import com.tencent.supersonic.headless.api.pojo.*;
import com.tencent.supersonic.headless.api.pojo.SchemaElementMatch;
import com.tencent.supersonic.headless.api.pojo.SchemaElementType;
import com.tencent.supersonic.headless.api.pojo.SchemaMapInfo;
import com.tencent.supersonic.headless.api.pojo.response.S2Term; import com.tencent.supersonic.headless.api.pojo.response.S2Term;
import com.tencent.supersonic.headless.chat.ChatQueryContext; import com.tencent.supersonic.headless.chat.ChatQueryContext;
import com.tencent.supersonic.headless.chat.knowledge.DatabaseMapResult; import com.tencent.supersonic.headless.chat.knowledge.DatabaseMapResult;
@@ -90,13 +87,14 @@ public class KeywordMapper extends BaseMapper {
.similarity(hanlpMapResult.getSimilarity()) .similarity(hanlpMapResult.getSimilarity())
.detectWord(hanlpMapResult.getDetectWord()).build(); .detectWord(hanlpMapResult.getDetectWord()).build();
// doDimValueAliasLogic 将维度值别名进行替换成真实维度值 // doDimValueAliasLogic 将维度值别名进行替换成真实维度值
doDimValueAliasLogic(schemaElementMatch); doDimValueAliasLogic(schemaElementMatch,chatQueryContext.getSemanticSchema().getDimensionValues());
addToSchemaMap(chatQueryContext.getMapInfo(), dataSetId, schemaElementMatch); addToSchemaMap(chatQueryContext.getMapInfo(), dataSetId, schemaElementMatch);
} }
} }
} }
private void doDimValueAliasLogic(SchemaElementMatch schemaElementMatch) { private void doDimValueAliasLogic(SchemaElementMatch schemaElementMatch,
List<SchemaElement> dimensionValues) {
SchemaElement element = schemaElementMatch.getElement(); SchemaElement element = schemaElementMatch.getElement();
if (SchemaElementType.VALUE.equals(element.getType())) { if (SchemaElementType.VALUE.equals(element.getType())) {
Long dimId = element.getId(); Long dimId = element.getId();
@@ -112,6 +110,18 @@ public class KeywordMapper extends BaseMapper {
schemaElementMatch.setWord(wordTech); schemaElementMatch.setWord(wordTech);
} }
} }
SchemaElement dimensionValue = dimensionValues.stream()
.filter(dimValue -> dimId.equals(dimValue.getId())).findFirst().orElse(null);
if (dimensionValue != null) {
SchemaValueMap dimValue =
dimensionValue.getSchemaValueMaps().stream().filter(schemaValueMap -> {
return StringUtils.equals(schemaValueMap.getBizName(), word)
|| schemaValueMap.getAlias().contains(word);
}).findFirst().orElse(null);
if (dimValue != null) {
schemaElementMatch.setWord(dimValue.getTechName());
}
}
} }
} }

View File

@@ -13,5 +13,7 @@ public interface DimensionDOCustomMapper {
void batchUpdateStatus(List<DimensionDO> dimensionDOS); void batchUpdateStatus(List<DimensionDO> dimensionDOS);
void batchUpdate(List<DimensionDO> dimensionDOS);
List<DimensionDO> queryDimensions(DimensionsFilter dimensionsFilter); List<DimensionDO> queryDimensions(DimensionsFilter dimensionsFilter);
} }

View File

@@ -13,6 +13,8 @@ public interface MetricDOCustomMapper {
void batchUpdateStatus(List<MetricDO> metricDOS); void batchUpdateStatus(List<MetricDO> metricDOS);
void batchUpdate(List<MetricDO> metricDOS);
void batchPublish(List<MetricDO> metricDOS); void batchPublish(List<MetricDO> metricDOS);
void batchUnPublish(List<MetricDO> metricDOS); void batchUnPublish(List<MetricDO> metricDOS);

View File

@@ -16,6 +16,8 @@ public interface DimensionRepository {
void batchUpdateStatus(List<DimensionDO> dimensionDOS); void batchUpdateStatus(List<DimensionDO> dimensionDOS);
void batchUpdate(List<DimensionDO> dimensionDOS);
DimensionDO getDimensionById(Long id); DimensionDO getDimensionById(Long id);
List<DimensionDO> getDimension(DimensionFilter dimensionFilter); List<DimensionDO> getDimension(DimensionFilter dimensionFilter);

View File

@@ -17,6 +17,8 @@ public interface MetricRepository {
void batchUpdateStatus(List<MetricDO> metricDOS); void batchUpdateStatus(List<MetricDO> metricDOS);
void batchUpdateMetric(List<MetricDO> metricDOS);
void batchPublish(List<MetricDO> metricDOS); void batchPublish(List<MetricDO> metricDOS);
void batchUnPublish(List<MetricDO> metricDOS); void batchUnPublish(List<MetricDO> metricDOS);

View File

@@ -46,6 +46,11 @@ public class DimensionRepositoryImpl implements DimensionRepository {
dimensionDOCustomMapper.batchUpdateStatus(dimensionDOS); dimensionDOCustomMapper.batchUpdateStatus(dimensionDOS);
} }
@Override
public void batchUpdate(List<DimensionDO> dimensionDOS) {
dimensionDOCustomMapper.batchUpdate(dimensionDOS);
}
@Override @Override
public DimensionDO getDimensionById(Long id) { public DimensionDO getDimensionById(Long id) {
return dimensionDOMapper.selectById(id); return dimensionDOMapper.selectById(id);

View File

@@ -53,6 +53,11 @@ public class MetricRepositoryImpl implements MetricRepository {
metricDOCustomMapper.batchUpdateStatus(metricDOS); metricDOCustomMapper.batchUpdateStatus(metricDOS);
} }
@Override
public void batchUpdateMetric(List<MetricDO> metricDOS) {
metricDOCustomMapper.batchUpdate(metricDOS);
}
@Override @Override
public void batchPublish(List<MetricDO> metricDOS) { public void batchPublish(List<MetricDO> metricDOS) {
metricDOCustomMapper.batchPublish(metricDOS); metricDOCustomMapper.batchPublish(metricDOS);

View File

@@ -27,10 +27,14 @@ public interface DimensionService {
DimensionResp createDimension(DimensionReq dimensionReq, User user) throws Exception; DimensionResp createDimension(DimensionReq dimensionReq, User user) throws Exception;
void alterDimensionBatch(List<DimensionReq> dimensionReqs, Long modelId, User user) throws Exception;
void createDimensionBatch(List<DimensionReq> dimensionReqs, User user) throws Exception; void createDimensionBatch(List<DimensionReq> dimensionReqs, User user) throws Exception;
void updateDimension(DimensionReq dimensionReq, User user) throws Exception; void updateDimension(DimensionReq dimensionReq, User user) throws Exception;
void updateDimensionBatch(List<DimensionReq> dimensionReqs, User user) throws Exception;
PageInfo<DimensionResp> queryDimension(PageDimensionReq pageDimensionReq); PageInfo<DimensionResp> queryDimension(PageDimensionReq pageDimensionReq);
List<DimensionResp> queryDimensions(DimensionsFilter dimensionsFilter); List<DimensionResp> queryDimensions(DimensionsFilter dimensionsFilter);
@@ -39,6 +43,8 @@ public interface DimensionService {
void deleteDimension(Long id, User user); void deleteDimension(Long id, User user);
void deleteDimensionBatch(List<Long> idList, User user);
List<DimensionResp> getDimensionInModelCluster(Long modelId); List<DimensionResp> getDimensionInModelCluster(Long modelId);
List<String> mockAlias(DimensionReq dimensionReq, String mockType, User user); List<String> mockAlias(DimensionReq dimensionReq, String mockType, User user);

View File

@@ -26,8 +26,12 @@ public interface MetricService {
void createMetricBatch(List<MetricReq> metricReqs, User user) throws Exception; void createMetricBatch(List<MetricReq> metricReqs, User user) throws Exception;
void alterMetricBatch(List<MetricReq> metricReqs, Long modelId, User user) throws Exception;
MetricResp updateMetric(MetricReq metricReq, User user) throws Exception; MetricResp updateMetric(MetricReq metricReq, User user) throws Exception;
void updateMetricBatch(List<MetricReq> metricReqs, User user) throws Exception;
void batchUpdateStatus(MetaBatchReq metaBatchReq, User user); void batchUpdateStatus(MetaBatchReq metaBatchReq, User user);
void batchPublish(List<Long> metricIds, User user); void batchPublish(List<Long> metricIds, User user);
@@ -40,6 +44,8 @@ public interface MetricService {
void deleteMetric(Long id, User user) throws Exception; void deleteMetric(Long id, User user) throws Exception;
void deleteMetricBatch(List<Long> idList, User user);
PageInfo<MetricResp> queryMetricMarket(PageMetricReq pageMetricReq, User user); PageInfo<MetricResp> queryMetricMarket(PageMetricReq pageMetricReq, User user);
PageInfo<MetricResp> queryMetric(PageMetricReq pageMetricReq, User user); PageInfo<MetricResp> queryMetric(PageMetricReq pageMetricReq, User user);

View File

@@ -11,6 +11,8 @@ import com.tencent.supersonic.headless.api.pojo.request.*;
import com.tencent.supersonic.headless.api.pojo.response.DatabaseResp; import com.tencent.supersonic.headless.api.pojo.response.DatabaseResp;
import com.tencent.supersonic.headless.api.pojo.response.ModelResp; import com.tencent.supersonic.headless.api.pojo.response.ModelResp;
import com.tencent.supersonic.headless.api.pojo.response.UnAvailableItemResp; import com.tencent.supersonic.headless.api.pojo.response.UnAvailableItemResp;
import com.tencent.supersonic.headless.server.persistence.dataobject.DimensionDO;
import com.tencent.supersonic.headless.server.persistence.dataobject.MetricDO;
import com.tencent.supersonic.headless.server.pojo.ModelFilter; import com.tencent.supersonic.headless.server.pojo.ModelFilter;
import java.sql.SQLException; import java.sql.SQLException;
@@ -53,5 +55,7 @@ public interface ModelService {
void batchUpdateStatus(MetaBatchReq metaBatchReq, User user); void batchUpdateStatus(MetaBatchReq metaBatchReq, User user);
Dimension updateDimension(DimensionReq dimensionReq, User user); void updateModelByDimAndMetric(Long modelId, List<DimensionReq> dimensionReqList, List<MetricReq> metricReqList, User user);
void deleteModelDetailByDimAndMetric(Long modelId, List<DimensionDO> dimensionReqList, List<MetricDO> metricReqList);
} }

View File

@@ -86,49 +86,70 @@ public class DimensionServiceImpl extends ServiceImpl<DimensionDOMapper, Dimensi
sendEventBatch(Lists.newArrayList(dimensionDO), EventType.ADD); sendEventBatch(Lists.newArrayList(dimensionDO), EventType.ADD);
// should update modelDetail // should update modelDetail
modelService.updateDimension(dimensionReq, user); modelService.updateModelByDimAndMetric(dimensionReq.getModelId(), Lists.newArrayList(dimensionReq), null, user);
return DimensionConverter.convert2DimensionResp(dimensionDO); return DimensionConverter.convert2DimensionResp(dimensionDO);
} }
@Override @Override
public void createDimensionBatch(List<DimensionReq> dimensionReqs, User user) { public void alterDimensionBatch(List<DimensionReq> dimensionReqs, Long modelId, User user) throws Exception {
if (CollectionUtils.isEmpty(dimensionReqs)) {
return;
}
Long modelId = dimensionReqs.get(0).getModelId();
List<DimensionResp> dimensionResps = getDimensions(modelId); List<DimensionResp> dimensionResps = getDimensions(modelId);
// get all dimension in model, only use bizname, because name can be changed to everything
Map<String, DimensionResp> bizNameMap = dimensionResps.stream() Map<String, DimensionResp> bizNameMap = dimensionResps.stream()
.collect(Collectors.toMap(DimensionResp::getBizName, a -> a, (k1, k2) -> k1)); .collect(Collectors.toMap(DimensionResp::getBizName, a -> a, (k1, k2) -> k1));
Map<String, DimensionResp> nameMap = dimensionResps.stream()
.collect(Collectors.toMap(DimensionResp::getName, a -> a, (k1, k2) -> k1));
List<DimensionReq> dimensionToInsert = Lists.newArrayList(); List<DimensionReq> dimensionToInsert = Lists.newArrayList();
List<DimensionReq> dimensionToUpdate = Lists.newArrayList();
List<Long> dimensionToDelete = Lists.newArrayList();
// look for which dimension need to insert, update, delete
dimensionReqs.stream().forEach(dimension -> { dimensionReqs.stream().forEach(dimension -> {
if (!bizNameMap.containsKey(dimension.getBizName()) if (!bizNameMap.containsKey(dimension.getBizName())) {
&& !nameMap.containsKey(dimension.getName())) {
dimensionToInsert.add(dimension); dimensionToInsert.add(dimension);
} else { } else {
DimensionResp dimensionRespByBizName = bizNameMap.get(dimension.getBizName()); DimensionResp dimensionRespByBizName = bizNameMap.get(dimension.getBizName());
DimensionResp dimensionRespByName = nameMap.get(dimension.getName());
if (null != dimensionRespByBizName && isChange(dimension, dimensionRespByBizName)) { if (null != dimensionRespByBizName && isChange(dimension, dimensionRespByBizName)) {
dimension.setId(dimensionRespByBizName.getId()); dimension.setId(dimensionRespByBizName.getId());
this.updateDimension(dimension, user); dimension.updatedBy(user.getName());
} else { dimensionToUpdate.add(dimension);
if (null != dimensionRespByName && isChange(dimension, dimensionRespByName)) {
dimension.setId(dimensionRespByName.getId());
this.updateDimension(dimension, user);
}
} }
} }
}); });
if (CollectionUtils.isEmpty(dimensionToInsert)) {
return; // the bizNames from alter dimensions
List<String> bizNames = dimensionReqs.stream().map(DimensionReq::getBizName).collect(Collectors.toList());
bizNameMap.keySet().forEach(bizNameInDb -> {
if (!bizNames.contains(bizNameInDb)) {
dimensionToDelete.add(bizNameMap.get(bizNameInDb).getId());
} }
});
// insert
if (!CollectionUtils.isEmpty(dimensionToInsert)) {
createDimensionBatch(dimensionToInsert, user);
}
// update
if (!CollectionUtils.isEmpty(dimensionToUpdate)) {
updateDimensionBatch(dimensionToUpdate, user);
}
// delete
if (!CollectionUtils.isEmpty(dimensionToDelete)) {
deleteDimensionBatch(dimensionToDelete, user);
}
}
@Override
public void createDimensionBatch(List<DimensionReq> dimensionReqs, User user) {
List<DimensionDO> dimensionDOS = List<DimensionDO> dimensionDOS =
dimensionToInsert.stream().peek(dimension -> dimension.createdBy(user.getName())) dimensionReqs.stream().peek(dimension -> dimension.createdBy(user.getName()))
.map(DimensionConverter::convert2DimensionDO).collect(Collectors.toList()); .map(DimensionConverter::convert2DimensionDO).collect(Collectors.toList());
dimensionRepository.createDimensionBatch(dimensionDOS); dimensionRepository.createDimensionBatch(dimensionDOS);
// should update modelDetail as well
modelService.updateModelByDimAndMetric(dimensionReqs.get(0).getModelId(), dimensionReqs, null, user);
sendEventBatch(dimensionDOS, EventType.ADD); sendEventBatch(dimensionDOS, EventType.ADD);
} }
@@ -141,13 +162,23 @@ public class DimensionServiceImpl extends ServiceImpl<DimensionDOMapper, Dimensi
DimensionConverter.convert(dimensionDO, dimensionReq); DimensionConverter.convert(dimensionDO, dimensionReq);
dimensionRepository.updateDimension(dimensionDO); dimensionRepository.updateDimension(dimensionDO);
// should update modelDetail as well // should update modelDetail as well
modelService.updateDimension(dimensionReq, user); modelService.updateModelByDimAndMetric(dimensionReq.getModelId(), Lists.newArrayList(dimensionReq), null, user);
if (!oldName.equals(dimensionDO.getName())) { if (!oldName.equals(dimensionDO.getName())) {
sendEvent(getDataItem(dimensionDO), EventType.UPDATE); sendEvent(getDataItem(dimensionDO), EventType.UPDATE);
} }
} }
@Override
public void updateDimensionBatch(List<DimensionReq> dimensionReqList, User user) {
checkExist(dimensionReqList);
List<DimensionDO> dimensionDOS = dimensionReqList.stream().map(DimensionConverter::convert2DimensionDO).collect(Collectors.toList());
dimensionRepository.batchUpdate(dimensionDOS);
// should update modelDetail as well
modelService.updateModelByDimAndMetric(dimensionReqList.get(0).getModelId(),dimensionReqList, null, user);
sendEventBatch(dimensionDOS, EventType.UPDATE);
}
@Override @Override
public void batchUpdateStatus(MetaBatchReq metaBatchReq, User user) { public void batchUpdateStatus(MetaBatchReq metaBatchReq, User user) {
if (CollectionUtils.isEmpty(metaBatchReq.getIds())) { if (CollectionUtils.isEmpty(metaBatchReq.getIds())) {
@@ -199,9 +230,30 @@ public class DimensionServiceImpl extends ServiceImpl<DimensionDOMapper, Dimensi
dimensionDO.setUpdatedAt(new Date()); dimensionDO.setUpdatedAt(new Date());
dimensionDO.setUpdatedBy(user.getName()); dimensionDO.setUpdatedBy(user.getName());
dimensionRepository.updateDimension(dimensionDO); dimensionRepository.updateDimension(dimensionDO);
// should update modelDetail
modelService.deleteModelDetailByDimAndMetric(dimensionDO.getModelId(), Lists.newArrayList(dimensionDO), null);
sendEventBatch(Lists.newArrayList(dimensionDO), EventType.DELETE); sendEventBatch(Lists.newArrayList(dimensionDO), EventType.DELETE);
} }
@Override
public void deleteDimensionBatch(List<Long> idList, User user) {
DimensionsFilter dimensionFilter = new DimensionsFilter();
dimensionFilter.setDimensionIds(idList);
List<DimensionDO> dimensionDOList = dimensionRepository.getDimensions(dimensionFilter);
if (CollectionUtils.isEmpty(dimensionDOList)) {
throw new RuntimeException(String.format("the dimension %s not exist", StringUtils.join(",",idList)));
}
dimensionDOList.forEach(dimensionDO -> {
dimensionDO.setStatus(StatusEnum.DELETED.getCode());
dimensionDO.setUpdatedAt(new Date());
dimensionDO.setUpdatedBy(user.getName());
});
dimensionRepository.batchUpdateStatus(dimensionDOList);
// should update modelDetail
modelService.deleteModelDetailByDimAndMetric(dimensionDOList.get(0).getModelId(), dimensionDOList, null);
sendEventBatch(dimensionDOList, EventType.DELETE);
}
@Override @Override
public DimensionResp getDimension(String bizName, Long modelId) { public DimensionResp getDimension(String bizName, Long modelId) {
List<DimensionResp> dimensionResps = getDimensions(modelId); List<DimensionResp> dimensionResps = getDimensions(modelId);

View File

@@ -11,23 +11,20 @@ import com.google.common.collect.Sets;
import com.tencent.supersonic.common.pojo.*; import com.tencent.supersonic.common.pojo.*;
import com.tencent.supersonic.common.pojo.enums.*; import com.tencent.supersonic.common.pojo.enums.*;
import com.tencent.supersonic.common.util.BeanMapper; import com.tencent.supersonic.common.util.BeanMapper;
import com.tencent.supersonic.common.util.JsonUtil;
import com.tencent.supersonic.headless.api.pojo.*; import com.tencent.supersonic.headless.api.pojo.*;
import com.tencent.supersonic.headless.api.pojo.enums.DimensionType;
import com.tencent.supersonic.headless.api.pojo.enums.MapModeEnum; import com.tencent.supersonic.headless.api.pojo.enums.MapModeEnum;
import com.tencent.supersonic.headless.api.pojo.enums.MetricDefineType; import com.tencent.supersonic.headless.api.pojo.enums.MetricDefineType;
import com.tencent.supersonic.headless.api.pojo.request.*; import com.tencent.supersonic.headless.api.pojo.request.*;
import com.tencent.supersonic.headless.api.pojo.response.*; import com.tencent.supersonic.headless.api.pojo.response.*;
import com.tencent.supersonic.headless.server.facade.service.ChatLayerService; import com.tencent.supersonic.headless.server.facade.service.ChatLayerService;
import com.tencent.supersonic.headless.server.persistence.dataobject.CollectDO; import com.tencent.supersonic.headless.server.persistence.dataobject.*;
import com.tencent.supersonic.headless.server.persistence.dataobject.MetricDO;
import com.tencent.supersonic.headless.server.persistence.dataobject.MetricQueryDefaultConfigDO;
import com.tencent.supersonic.headless.server.persistence.mapper.MetricDOMapper; import com.tencent.supersonic.headless.server.persistence.mapper.MetricDOMapper;
import com.tencent.supersonic.headless.server.persistence.repository.MetricRepository; import com.tencent.supersonic.headless.server.persistence.repository.MetricRepository;
import com.tencent.supersonic.headless.server.pojo.*; import com.tencent.supersonic.headless.server.pojo.*;
import com.tencent.supersonic.headless.server.service.*; import com.tencent.supersonic.headless.server.service.*;
import com.tencent.supersonic.headless.server.utils.AliasGenerateHelper; import com.tencent.supersonic.headless.server.utils.*;
import com.tencent.supersonic.headless.server.utils.MetricCheckUtils;
import com.tencent.supersonic.headless.server.utils.MetricConverter;
import com.tencent.supersonic.headless.server.utils.ModelClusterBuilder;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeanUtils; import org.springframework.beans.BeanUtils;
@@ -77,57 +74,82 @@ public class MetricServiceImpl extends ServiceImpl<MetricDOMapper, MetricDO>
@Override @Override
public MetricResp createMetric(MetricReq metricReq, User user) { public MetricResp createMetric(MetricReq metricReq, User user) {
checkExist(Lists.newArrayList(metricReq)); checkExist(Lists.newArrayList(metricReq));
MetricCheckUtils.checkParam(metricReq); MetricCheckUtils.checkParam(Lists.newArrayList(metricReq));
metricReq.createdBy(user.getName()); metricReq.createdBy(user.getName());
MetricDO metricDO = MetricConverter.convert2MetricDO(metricReq); MetricDO metricDO = MetricConverter.convert2MetricDO(metricReq);
metricRepository.createMetric(metricDO); metricRepository.createMetric(metricDO);
sendEventBatch(Lists.newArrayList(metricDO), EventType.ADD); sendEventBatch(Lists.newArrayList(metricDO), EventType.ADD);
// should update modelDetail as well
modelService.updateModelByDimAndMetric(metricReq.getModelId(),null, Lists.newArrayList(metricReq), user);
return MetricConverter.convert2MetricResp(metricDO); return MetricConverter.convert2MetricResp(metricDO);
} }
@Override @Override
public void createMetricBatch(List<MetricReq> metricReqs, User user) { public void createMetricBatch(List<MetricReq> metricReqs, User user) {
if (CollectionUtils.isEmpty(metricReqs)) {
return;
}
Long modelId = metricReqs.get(0).getModelId();
List<MetricResp> metricResps = getMetrics(new MetaFilter(Lists.newArrayList(modelId)));
Map<String, MetricResp> bizNameMap = metricResps.stream()
.collect(Collectors.toMap(MetricResp::getBizName, a -> a, (k1, k2) -> k1));
Map<String, MetricResp> nameMap = metricResps.stream()
.collect(Collectors.toMap(MetricResp::getName, a -> a, (k1, k2) -> k1));
List<MetricReq> metricToInsert = Lists.newArrayList();
metricReqs.stream().forEach(metric -> {
if (!bizNameMap.containsKey(metric.getBizName())
&& !nameMap.containsKey(metric.getName())) {
metricToInsert.add(metric);
} else {
MetricResp metricRespByBizName = bizNameMap.get(metric.getBizName());
MetricResp metricRespByName = nameMap.get(metric.getName());
if (null != metricRespByBizName) {
metric.setId(metricRespByBizName.getId());
this.updateMetric(metric, user);
} else {
if (null != metricRespByName) {
metric.setId(metricRespByName.getId());
this.updateMetric(metric, user);
}
}
}
});
if (CollectionUtils.isEmpty(metricToInsert)) {
return;
}
List<MetricDO> metricDOS = List<MetricDO> metricDOS =
metricToInsert.stream().peek(metric -> metric.createdBy(user.getName())) metricReqs.stream().peek(metric -> metric.createdBy(user.getName()))
.map(MetricConverter::convert2MetricDO).collect(Collectors.toList()); .map(MetricConverter::convert2MetricDO).collect(Collectors.toList());
metricRepository.createMetricBatch(metricDOS); metricRepository.createMetricBatch(metricDOS);
// should update modelDetail as well
modelService.updateModelByDimAndMetric(metricReqs.get(0).getModelId(), null, metricReqs, user);
sendEventBatch(metricDOS, EventType.ADD); sendEventBatch(metricDOS, EventType.ADD);
} }
@Override
public void alterMetricBatch(List<MetricReq> metricReqs, Long modelId, User user) {
List<MetricResp> metricResps = getMetrics(new MetaFilter(Lists.newArrayList(modelId)));
// get all metric in model, only use bizname, because name can be changed to everything
Map<String, MetricResp> bizNameMap = metricResps.stream()
.collect(Collectors.toMap(MetricResp::getBizName, a -> a, (k1, k2) -> k1));
List<MetricReq> metricToInsert = Lists.newArrayList();
List<MetricReq> metricToUpdate = Lists.newArrayList();
List<Long> metricToDelete = Lists.newArrayList();
metricReqs.stream().forEach(metric -> {
if (!bizNameMap.containsKey(metric.getBizName())) {
metricToInsert.add(metric);
} else {
MetricResp metricRespByBizName = bizNameMap.get(metric.getBizName());
if (null != metricRespByBizName && isChange(metric, metricRespByBizName)) {
metric.setId(metricRespByBizName.getId());
metric.updatedBy(user.getName());
metricToUpdate.add(metric);
}
}
});
// the bizNames from alter dimensions
List<String> bizNames = metricReqs.stream().map(MetricReq::getBizName).collect(Collectors.toList());
bizNameMap.keySet().forEach(bizNameInDb -> {
if (!bizNames.contains(bizNameInDb)) {
metricToDelete.add(bizNameMap.get(bizNameInDb).getId());
}
});
// insert
if (!CollectionUtils.isEmpty(metricToInsert)) {
createMetricBatch(metricToInsert, user);
}
// update
if (!CollectionUtils.isEmpty(metricToUpdate)) {
updateMetricBatch(metricToUpdate, user);
}
// delete
if (!CollectionUtils.isEmpty(metricToDelete)) {
deleteMetricBatch(metricToDelete, user);
}
}
@Override @Override
public MetricResp updateMetric(MetricReq metricReq, User user) { public MetricResp updateMetric(MetricReq metricReq, User user) {
MetricCheckUtils.checkParam(metricReq); MetricCheckUtils.checkParam(Lists.newArrayList(metricReq));
checkExist(Lists.newArrayList(metricReq)); checkExist(Lists.newArrayList(metricReq));
metricReq.updatedBy(user.getName()); metricReq.updatedBy(user.getName());
MetricDO metricDO = metricRepository.getMetricById(metricReq.getId()); MetricDO metricDO = metricRepository.getMetricById(metricReq.getId());
@@ -140,9 +162,23 @@ public class MetricServiceImpl extends ServiceImpl<MetricDOMapper, MetricDO>
dataItem.setNewName(metricDO.getName()); dataItem.setNewName(metricDO.getName());
sendEvent(dataItem, EventType.UPDATE); sendEvent(dataItem, EventType.UPDATE);
} }
// should update modelDetail as well
modelService.updateModelByDimAndMetric(metricReq.getModelId(), null, Lists.newArrayList(metricReq), user);
return MetricConverter.convert2MetricResp(metricDO); return MetricConverter.convert2MetricResp(metricDO);
} }
@Override
public void updateMetricBatch(List<MetricReq> metricReqs, User user) {
MetricCheckUtils.checkParam(metricReqs);
checkExist(metricReqs);
List<MetricDO> metricDOS = metricReqs.stream().map(MetricConverter::convert2MetricDO).collect(Collectors.toList());
metricRepository.batchUpdateMetric(metricDOS);
// should update modelDetail as well
modelService.updateModelByDimAndMetric(metricReqs.get(0).getModelId(), null, metricReqs, user);
sendEventBatch(metricDOS, EventType.UPDATE);
}
@Override @Override
public void batchUpdateStatus(MetaBatchReq metaBatchReq, User user) { public void batchUpdateStatus(MetaBatchReq metaBatchReq, User user) {
if (CollectionUtils.isEmpty(metaBatchReq.getIds())) { if (CollectionUtils.isEmpty(metaBatchReq.getIds())) {
@@ -240,9 +276,30 @@ public class MetricServiceImpl extends ServiceImpl<MetricDOMapper, MetricDO>
metricDO.setUpdatedAt(new Date()); metricDO.setUpdatedAt(new Date());
metricDO.setUpdatedBy(user.getName()); metricDO.setUpdatedBy(user.getName());
metricRepository.updateMetric(metricDO); metricRepository.updateMetric(metricDO);
// should update modelDetail
modelService.deleteModelDetailByDimAndMetric(metricDO.getModelId(), null, Lists.newArrayList(metricDO));
sendEventBatch(Lists.newArrayList(metricDO), EventType.DELETE); sendEventBatch(Lists.newArrayList(metricDO), EventType.DELETE);
} }
@Override
public void deleteMetricBatch(List<Long> idList, User user) {
MetricsFilter metricsFilter = new MetricsFilter();
metricsFilter.setMetricIds(idList);
List<MetricDO> metricDOList = metricRepository.getMetrics(metricsFilter);
if (CollectionUtils.isEmpty(metricDOList)) {
throw new RuntimeException(String.format("the metrics %s not exist", StringUtils.join(",",idList)));
}
metricDOList.forEach(metricDO -> {
metricDO.setStatus(StatusEnum.DELETED.getCode());
metricDO.setUpdatedAt(new Date());
metricDO.setUpdatedBy(user.getName());
});
metricRepository.batchUpdateStatus(metricDOList);
// should update modelDetail
modelService.deleteModelDetailByDimAndMetric(metricDOList.get(0).getModelId(), null, metricDOList);
sendEventBatch(metricDOList, EventType.DELETE);
}
@Override @Override
public PageInfo<MetricResp> queryMetricMarket(PageMetricReq pageMetricReq, User user) { public PageInfo<MetricResp> queryMetricMarket(PageMetricReq pageMetricReq, User user) {
// search by whole text // search by whole text
@@ -549,7 +606,7 @@ public class MetricServiceImpl extends ServiceImpl<MetricDOMapper, MetricDO>
return metricQueryDefaultConfig; return metricQueryDefaultConfig;
} }
private void checkExist(List<MetricBaseReq> metricReqs) { private void checkExist(List<MetricReq> metricReqs) {
Long modelId = metricReqs.get(0).getModelId(); Long modelId = metricReqs.get(0).getModelId();
MetaFilter metaFilter = new MetaFilter(); MetaFilter metaFilter = new MetaFilter();
metaFilter.setModelIds(Lists.newArrayList(modelId)); metaFilter.setModelIds(Lists.newArrayList(modelId));
@@ -830,8 +887,9 @@ public class MetricServiceImpl extends ServiceImpl<MetricDOMapper, MetricDO>
return modelResps.stream().map(ModelResp::getId).collect(Collectors.toSet()); return modelResps.stream().map(ModelResp::getId).collect(Collectors.toSet());
} }
private boolean isNameChange(MetricReq metricReq, MetricResp metricResp) { private boolean isChange(MetricReq metricReq, MetricResp metricResp) {
boolean isNameChange = !metricReq.getName().equals(metricResp.getName()); boolean isNameChange = !metricReq.getName().equals(metricResp.getName());
return isNameChange; boolean isBizNameChange = !Objects.equals(metricReq.getMetricDefineByMeasureParams(),metricResp.getMetricDefineByMeasureParams());
return isNameChange || isBizNameChange;
} }
} }

View File

@@ -15,6 +15,8 @@ import com.tencent.supersonic.headless.api.pojo.request.*;
import com.tencent.supersonic.headless.api.pojo.response.*; import com.tencent.supersonic.headless.api.pojo.response.*;
import com.tencent.supersonic.headless.server.modeller.SemanticModeller; import com.tencent.supersonic.headless.server.modeller.SemanticModeller;
import com.tencent.supersonic.headless.server.persistence.dataobject.DateInfoDO; import com.tencent.supersonic.headless.server.persistence.dataobject.DateInfoDO;
import com.tencent.supersonic.headless.server.persistence.dataobject.DimensionDO;
import com.tencent.supersonic.headless.server.persistence.dataobject.MetricDO;
import com.tencent.supersonic.headless.server.persistence.dataobject.ModelDO; import com.tencent.supersonic.headless.server.persistence.dataobject.ModelDO;
import com.tencent.supersonic.headless.server.persistence.repository.DateInfoRepository; import com.tencent.supersonic.headless.server.persistence.repository.DateInfoRepository;
import com.tencent.supersonic.headless.server.persistence.repository.ModelRepository; import com.tencent.supersonic.headless.server.persistence.repository.ModelRepository;
@@ -87,8 +89,12 @@ public class ModelServiceImpl implements ModelService {
// checkParams(modelReq); // checkParams(modelReq);
ModelDO modelDO = ModelConverter.convert(modelReq, user); ModelDO modelDO = ModelConverter.convert(modelReq, user);
modelRepository.createModel(modelDO); modelRepository.createModel(modelDO);
batchCreateDimension(modelDO, user); // create or update dimension
batchCreateMetric(modelDO, user); List<DimensionReq> dimensionReqs = ModelConverter.convertDimensionList(modelDO);
dimensionService.alterDimensionBatch(dimensionReqs, modelDO.getId(), user);
// create or update metric
List<MetricReq> metricReqs = ModelConverter.convertMetricList(modelDO);
metricService.alterMetricBatch(metricReqs, modelDO.getId(), user);
sendEvent(modelDO, EventType.ADD); sendEvent(modelDO, EventType.ADD);
return ModelConverter.convert(modelDO); return ModelConverter.convert(modelDO);
} }
@@ -117,8 +123,12 @@ public class ModelServiceImpl implements ModelService {
ModelDO modelDO = modelRepository.getModelById(modelReq.getId()); ModelDO modelDO = modelRepository.getModelById(modelReq.getId());
ModelConverter.convert(modelDO, modelReq, user); ModelConverter.convert(modelDO, modelReq, user);
modelRepository.updateModel(modelDO); modelRepository.updateModel(modelDO);
batchCreateDimension(modelDO, user); // create or update dimension
batchCreateMetric(modelDO, user); List<DimensionReq> dimensionReqs = ModelConverter.convertDimensionList(modelDO);
dimensionService.alterDimensionBatch(dimensionReqs, modelDO.getId(), user);
// create or update metric
List<MetricReq> metricReqs = ModelConverter.convertMetricList(modelDO);
metricService.alterMetricBatch(metricReqs, modelDO.getId(), user);
sendEvent(modelDO, EventType.UPDATE); sendEvent(modelDO, EventType.UPDATE);
return ModelConverter.convert(modelDO); return ModelConverter.convert(modelDO);
} }
@@ -239,32 +249,6 @@ public class ModelServiceImpl implements ModelService {
return dbSchema; return dbSchema;
} }
private void batchCreateDimension(ModelDO modelDO, User user) throws Exception {
List<DimensionReq> dimensionReqs = ModelConverter.convertDimensionList(modelDO);
List<DimensionReq> dimensionToCreate = new ArrayList<>();
for (DimensionReq dimensionReq : dimensionReqs) {
DimensionResp dimensionResp =
dimensionService.getDimension(dimensionReq.getBizName(), modelDO.getId());
if (dimensionResp == null) {
dimensionToCreate.add(dimensionReq);
}
}
dimensionService.createDimensionBatch(dimensionToCreate, user);
}
private void batchCreateMetric(ModelDO modelDO, User user) throws Exception {
List<MetricReq> metricReqs = ModelConverter.convertMetricList(modelDO);
List<MetricReq> metricToCreate = new ArrayList<>();
for (MetricReq metricReq : metricReqs) {
MetricResp metricResp =
metricService.getMetric(modelDO.getId(), metricReq.getBizName());
if (metricResp == null) {
metricToCreate.add(metricReq);
}
}
metricService.createMetricBatch(metricToCreate, user);
}
private void checkParams(ModelReq modelReq) { private void checkParams(ModelReq modelReq) {
String forbiddenCharacters = NameCheckUtils.findForbiddenCharacters(modelReq.getName()); String forbiddenCharacters = NameCheckUtils.findForbiddenCharacters(modelReq.getName());
if (StringUtils.isNotBlank(forbiddenCharacters)) { if (StringUtils.isNotBlank(forbiddenCharacters)) {
@@ -537,32 +521,83 @@ public class ModelServiceImpl implements ModelService {
} }
@Override @Override
public Dimension updateDimension(DimensionReq dimensionReq, User user) { public void updateModelByDimAndMetric(Long modelId, List<DimensionReq> dimensionReqList, List<MetricReq> metricReqList, User user) {
ModelDO modelDO = getModelDO(dimensionReq.getModelId()); ModelDO modelDO = getModelDO(modelId);
ModelDetail modelDetail = JsonUtil.toObject(modelDO.getModelDetail(), ModelDetail.class); ModelDetail modelDetail = JsonUtil.toObject(modelDO.getModelDetail(), ModelDetail.class);
if (!CollectionUtils.isEmpty(dimensionReqList)) {
dimensionReqList.forEach(dimensionReq -> {
Optional<Dimension> dimOptional = modelDetail.getDimensions().stream() Optional<Dimension> dimOptional = modelDetail.getDimensions().stream()
.filter(dimension -> dimension.getBizName().equals(dimensionReq.getBizName())) .filter(dimension -> dimension.getBizName().equals(dimensionReq.getBizName()))
.findFirst(); .findFirst();
Dimension result;
if (dimOptional.isPresent()) { if (dimOptional.isPresent()) {
Dimension dimension = dimOptional.get(); Dimension dimension = dimOptional.get();
dimension.setExpr(dimensionReq.getExpr()); dimension.setExpr(dimensionReq.getExpr());
dimension.setName(dimensionReq.getName()); dimension.setName(dimensionReq.getName());
dimension.setType(DimensionType.valueOf(dimensionReq.getType())); dimension.setType(DimensionType.valueOf(dimensionReq.getType()));
dimension.setDescription(dimensionReq.getDescription()); dimension.setDescription(dimensionReq.getDescription());
result = dimension;
} else { } else {
Dimension dimension = Dimension.builder().name(dimensionReq.getName()) Dimension dimension = Dimension.builder().name(dimensionReq.getName())
.bizName(dimensionReq.getBizName()).expr(dimensionReq.getExpr()) .bizName(dimensionReq.getBizName()).expr(dimensionReq.getExpr())
.type(DimensionType.valueOf(dimensionReq.getType())) .type(DimensionType.valueOf(dimensionReq.getType()))
.description(dimensionReq.getDescription()).build(); .description(dimensionReq.getDescription()).build();
modelDetail.getDimensions().add(dimension); modelDetail.getDimensions().add(dimension);
result = dimension; }
});
}
if (!CollectionUtils.isEmpty(metricReqList)) {
// 目前modeltail中的measure
Map<String, Measure> mesureMap = modelDetail.getMeasures().stream().collect(Collectors.toMap(Measure::getBizName, a -> a, (k1, k2) -> k1));
metricReqList.forEach(metricReq -> {
if (null != metricReq.getMetricDefineByMeasureParams() && !CollectionUtils.isEmpty(metricReq.getMetricDefineByMeasureParams().getMeasures())) {
for(Measure alterMeasure : metricReq.getMetricDefineByMeasureParams().getMeasures()) {
if (mesureMap.containsKey(alterMeasure.getBizName())) {
Measure measure = mesureMap.get(alterMeasure.getBizName());
BeanUtils.copyProperties(alterMeasure, measure);
} else {
modelDetail.getMeasures().add(alterMeasure);
}
}
} else {
modelDetail.getMeasures().clear();
}
});
}
modelDO.setModelDetail(JsonUtil.toString(modelDetail));
modelRepository.updateModel(modelDO);
}
@Override
public void deleteModelDetailByDimAndMetric(Long modelId, List<DimensionDO> dimensionList, List<MetricDO> metricReqList) {
ModelDO modelDO = getModelDO(modelId);
ModelDetail modelDetail = JsonUtil.toObject(modelDO.getModelDetail(), ModelDetail.class);
if (!CollectionUtils.isEmpty(dimensionList)) {
dimensionList.forEach(dimensionReq -> {
Optional<Dimension> dimOptional = modelDetail.getDimensions().stream()
.filter(dimension -> dimension.getBizName().equals(dimensionReq.getBizName()))
.findFirst();
if (dimOptional.isPresent()) {
Dimension dimension = dimOptional.get();
modelDetail.getDimensions().remove(dimension);
}
});
}
if (!CollectionUtils.isEmpty(metricReqList)) {
metricReqList.forEach(metricReq -> {
Optional<Measure> metricOptional = modelDetail.getMeasures().stream()
.filter(measure -> measure.getBizName().equals(metricReq.getBizName()))
.findFirst();
if (metricOptional.isPresent()) {
Measure measure = metricOptional.get();
modelDetail.getMeasures().remove(measure);
}
});
} }
modelDO.setModelDetail(JsonUtil.toString(modelDetail)); modelDO.setModelDetail(JsonUtil.toString(modelDetail));
modelRepository.updateModel(modelDO); modelRepository.updateModel(modelDO);
return result;
} }
protected ModelDO getModelDO(Long id) { protected ModelDO getModelDO(Long id) {

View File

@@ -113,7 +113,7 @@ public class DataSetSchemaBuilder {
if (dim.isTimeDimension()) { if (dim.isTimeDimension()) {
String timeFormat = String timeFormat =
String.valueOf(dim.getExt().get(DimensionConstants.DIMENSION_TIME_FORMAT)); String.valueOf(dim.getExt().get(DimensionConstants.DIMENSION_TIME_FORMAT));
setDefaultTimeFormat(dimToAdd, dim.getTypeParams(), timeFormat); setDefaultTimeFormat(dimToAdd, timeFormat);
} }
dimensions.add(dimToAdd); dimensions.add(dimToAdd);
} }
@@ -125,7 +125,13 @@ public class DataSetSchemaBuilder {
for (DimSchemaResp dim : resp.getDimensions()) { for (DimSchemaResp dim : resp.getDimensions()) {
Set<String> dimValueAlias = new HashSet<>(); Set<String> dimValueAlias = new HashSet<>();
List<DimValueMap> dimValueMaps = dim.getDimValueMaps(); List<DimValueMap> dimValueMaps = dim.getDimValueMaps();
List<SchemaValueMap> schemaValueMaps = new ArrayList<>();
if (!CollectionUtils.isEmpty(dimValueMaps)) { if (!CollectionUtils.isEmpty(dimValueMaps)) {
for (DimValueMap dimValueMap : dimValueMaps) {
SchemaValueMap schemaValueMap = new SchemaValueMap();
BeanUtils.copyProperties(dimValueMap, schemaValueMap);
schemaValueMaps.add(schemaValueMap);
}
for (DimValueMap dimValueMap : dimValueMaps) { for (DimValueMap dimValueMap : dimValueMaps) {
if (StringUtils.isNotEmpty(dimValueMap.getBizName())) { if (StringUtils.isNotEmpty(dimValueMap.getBizName())) {
dimValueAlias.add(dimValueMap.getBizName()); dimValueAlias.add(dimValueMap.getBizName());
@@ -138,7 +144,7 @@ public class DataSetSchemaBuilder {
SchemaElement dimValueToAdd = SchemaElement.builder().dataSetId(resp.getId()) SchemaElement dimValueToAdd = SchemaElement.builder().dataSetId(resp.getId())
.dataSetName(resp.getName()).model(dim.getModelId()).id(dim.getId()) .dataSetName(resp.getName()).model(dim.getModelId()).id(dim.getId())
.name(dim.getName()).bizName(dim.getBizName()).type(SchemaElementType.VALUE) .name(dim.getName()).bizName(dim.getBizName()).type(SchemaElementType.VALUE)
.useCnt(dim.getUseCnt()) .schemaValueMaps(schemaValueMaps).useCnt(dim.getUseCnt())
.alias(new ArrayList<>(Arrays.asList(dimValueAlias.toArray(new String[0])))) .alias(new ArrayList<>(Arrays.asList(dimValueAlias.toArray(new String[0]))))
.isTag(dim.getIsTag()).description(dim.getDescription()).build(); .isTag(dim.getIsTag()).description(dim.getDescription()).build();
dimensionValues.add(dimValueToAdd); dimensionValues.add(dimValueToAdd);
@@ -193,7 +199,7 @@ public class DataSetSchemaBuilder {
} }
private static void setDefaultTimeFormat(SchemaElement dimToAdd, private static void setDefaultTimeFormat(SchemaElement dimToAdd,
DimensionTimeTypeParams dimensionTimeTypeParams, String timeFormat) { String timeFormat) {
dimToAdd.getExtInfo().put(DimensionConstants.DIMENSION_TIME_FORMAT, timeFormat); dimToAdd.getExtInfo().put(DimensionConstants.DIMENSION_TIME_FORMAT, timeFormat);
} }
} }

View File

@@ -70,6 +70,9 @@ public class DimensionConverter {
if (dimensionReq.getExt() != null) { if (dimensionReq.getExt() != null) {
dimensionDO.setExt(JSONObject.toJSONString(dimensionReq.getExt())); dimensionDO.setExt(JSONObject.toJSONString(dimensionReq.getExt()));
} }
if (dimensionReq.getTypeParams() != null) {
dimensionDO.setTypeParams(JSONObject.toJSONString(dimensionReq.getTypeParams()));
}
dimensionDO.setStatus(StatusEnum.ONLINE.getCode()); dimensionDO.setStatus(StatusEnum.ONLINE.getCode());
return dimensionDO; return dimensionDO;
} }

View File

@@ -10,9 +10,12 @@ import com.tencent.supersonic.headless.api.pojo.request.MetricReq;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
import org.springframework.util.CollectionUtils; import org.springframework.util.CollectionUtils;
import java.util.List;
public class MetricCheckUtils { public class MetricCheckUtils {
public static void checkParam(MetricReq metricReq) { public static void checkParam(List<MetricReq> metricReqList) {
metricReqList.forEach(metricReq -> {
String expr = ""; String expr = "";
if (MetricDefineType.METRIC.equals(metricReq.getMetricDefineType())) { if (MetricDefineType.METRIC.equals(metricReq.getMetricDefineType())) {
MetricDefineByMetricParams typeParams = metricReq.getMetricDefineByMetricParams(); MetricDefineByMetricParams typeParams = metricReq.getMetricDefineByMetricParams();
@@ -58,6 +61,7 @@ public class MetricCheckUtils {
throw new InvalidArgumentException( throw new InvalidArgumentException(
String.format("名称包含特殊字符%s, 请修改", forbiddenCharacters)); String.format("名称包含特殊字符%s, 请修改", forbiddenCharacters));
} }
});
} }
private static boolean hasAggregateFunction(String expr) { private static boolean hasAggregateFunction(String expr) {

View File

@@ -28,7 +28,7 @@ public class ModelConverter {
public static ModelDO convert(ModelReq modelReq, User user) { public static ModelDO convert(ModelReq modelReq, User user) {
ModelDO modelDO = new ModelDO(); ModelDO modelDO = new ModelDO();
ModelDetail modelDetail = createModelDetail(modelReq); ModelDetail modelDetail = convert(modelReq);
modelReq.createdBy(user.getName()); modelReq.createdBy(user.getName());
BeanMapper.mapper(modelReq, modelDO); BeanMapper.mapper(modelReq, modelDO);
modelDO.setStatus(StatusEnum.ONLINE.getCode()); modelDO.setStatus(StatusEnum.ONLINE.getCode());
@@ -68,7 +68,7 @@ public class ModelConverter {
} }
public static ModelDO convert(ModelDO modelDO, ModelReq modelReq, User user) { public static ModelDO convert(ModelDO modelDO, ModelReq modelReq, User user) {
ModelDetail modelDetail = updateModelDetail(modelReq); ModelDetail modelDetail = convert(modelReq);
BeanMapper.mapper(modelReq, modelDO); BeanMapper.mapper(modelReq, modelDO);
if (modelReq.getDrillDownDimensions() != null) { if (modelReq.getDrillDownDimensions() != null) {
modelDO.setDrillDownDimensions( modelDO.setDrillDownDimensions(
@@ -288,18 +288,21 @@ public class ModelConverter {
.collect(Collectors.toList()); .collect(Collectors.toList());
} }
private static ModelDetail createModelDetail(ModelReq modelReq) { private static ModelDetail convert(ModelReq modelReq) {
ModelDetail modelDetail = new ModelDetail(); ModelDetail modelDetail = new ModelDetail();
List<Measure> measures = modelReq.getModelDetail().getMeasures(); List<Measure> measures = modelReq.getModelDetail().getMeasures();
List<Dimension> dimensions = modelReq.getModelDetail().getDimensions(); List<Dimension> dimensions = modelReq.getModelDetail().getDimensions();
List<Identify> identifiers = modelReq.getModelDetail().getIdentifiers(); List<Identify> identifiers = modelReq.getModelDetail().getIdentifiers();
List<Field> fields = modelReq.getModelDetail().getFields(); List<Field> fields = modelReq.getModelDetail().getFields();
List<String> fieldNames = fields.stream().map(Field::getFieldName).collect(Collectors.toList());
if (measures != null) { if (measures != null) {
for (Measure measure : measures) { for (Measure measure : measures) {
if (StringUtils.isNotBlank(measure.getBizName()) if (StringUtils.isNotBlank(measure.getBizName())
&& StringUtils.isBlank(measure.getExpr())) { && StringUtils.isBlank(measure.getExpr())) {
measure.setExpr(measure.getBizName()); measure.setExpr(measure.getBizName());
}
if (StringUtils.isNotBlank(measure.getBizName()) && !fieldNames.contains(measure.getBizName())) {
fields.add(new Field(measure.getBizName(), "")); fields.add(new Field(measure.getBizName(), ""));
} }
} }
@@ -309,53 +312,24 @@ public class ModelConverter {
if (StringUtils.isNotBlank(dimension.getBizName()) if (StringUtils.isNotBlank(dimension.getBizName())
&& StringUtils.isBlank(dimension.getExpr())) { && StringUtils.isBlank(dimension.getExpr())) {
dimension.setExpr(dimension.getBizName()); dimension.setExpr(dimension.getBizName());
}
if (StringUtils.isNotBlank(dimension.getBizName()) && !fieldNames.contains(dimension.getBizName())) {
fields.add(new Field(dimension.getBizName(), "")); fields.add(new Field(dimension.getBizName(), ""));
} }
} }
} }
if (identifiers != null) { if (identifiers != null) {
for (Identify identify : identifiers) { for (Identify identify : identifiers) {
if (StringUtils.isNotBlank(identify.getBizName()) if (StringUtils.isNotBlank(identify.getBizName()) && StringUtils.isBlank(identify.getName())) {
&& StringUtils.isBlank(identify.getName())) {
identify.setName(identify.getBizName()); identify.setName(identify.getBizName());
} }
identify.setIsCreateDimension(1); identify.setIsCreateDimension(1);
if (StringUtils.isNotBlank(identify.getBizName()) && !fieldNames.contains(identify.getBizName())) {
fields.add(new Field(identify.getBizName(), "")); fields.add(new Field(identify.getBizName(), ""));
} }
} }
BeanMapper.mapper(modelReq.getModelDetail(), modelDetail);
return modelDetail;
} }
private static ModelDetail updateModelDetail(ModelReq modelReq) {
ModelDetail modelDetail = new ModelDetail();
List<Measure> measures = modelReq.getModelDetail().getMeasures();
List<Dimension> dimensions = modelReq.getModelDetail().getDimensions();
if (!CollectionUtils.isEmpty(dimensions)) {
for (Dimension dimension : dimensions) {
if (StringUtils.isNotBlank(dimension.getBizName())
&& StringUtils.isBlank(dimension.getExpr())) {
dimension.setExpr(dimension.getBizName());
}
}
}
if (measures == null) {
measures = Lists.newArrayList();
}
for (Measure measure : measures) {
if (StringUtils.isBlank(measure.getBizName())) {
continue;
}
// Compatible with front-end tmp
String oriFieldName =
measure.getBizName().replaceFirst(modelReq.getBizName() + "_", "");
measure.setExpr(oriFieldName);
if (!measure.getBizName().startsWith(modelReq.getBizName())) {
measure.setBizName(String.format("%s_%s", modelReq.getBizName(), oriFieldName));
}
}
BeanMapper.mapper(modelReq.getModelDetail(), modelDetail); BeanMapper.mapper(modelReq.getModelDetail(), modelDetail);
return modelDetail; return modelDetail;
} }

View File

@@ -3,26 +3,27 @@
"http://mybatis.org/dtd/mybatis-3-mapper.dtd"> "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.tencent.supersonic.headless.server.persistence.mapper.DimensionDOCustomMapper"> <mapper namespace="com.tencent.supersonic.headless.server.persistence.mapper.DimensionDOCustomMapper">
<resultMap id="BaseResultMap" type="com.tencent.supersonic.headless.server.persistence.dataobject.DimensionDO"> <resultMap id="BaseResultMap" type="com.tencent.supersonic.headless.server.persistence.dataobject.DimensionDO">
<id column="id" jdbcType="BIGINT" property="id" /> <id column="id" jdbcType="BIGINT" property="id"/>
<result column="model_id" jdbcType="BIGINT" property="modelId" /> <result column="model_id" jdbcType="BIGINT" property="modelId"/>
<result column="name" jdbcType="VARCHAR" property="name" /> <result column="name" jdbcType="VARCHAR" property="name"/>
<result column="biz_name" jdbcType="VARCHAR" property="bizName" /> <result column="biz_name" jdbcType="VARCHAR" property="bizName"/>
<result column="description" jdbcType="VARCHAR" property="description" /> <result column="description" jdbcType="VARCHAR" property="description"/>
<result column="status" jdbcType="INTEGER" property="status" /> <result column="status" jdbcType="INTEGER" property="status"/>
<result column="sensitive_level" jdbcType="INTEGER" property="sensitiveLevel" /> <result column="sensitive_level" jdbcType="INTEGER" property="sensitiveLevel"/>
<result column="type" jdbcType="VARCHAR" property="type" /> <result column="type" jdbcType="VARCHAR" property="type"/>
<result column="created_at" jdbcType="TIMESTAMP" property="createdAt" /> <result column="created_at" jdbcType="TIMESTAMP" property="createdAt"/>
<result column="created_by" jdbcType="VARCHAR" property="createdBy" /> <result column="created_by" jdbcType="VARCHAR" property="createdBy"/>
<result column="updated_at" jdbcType="TIMESTAMP" property="updatedAt" /> <result column="updated_at" jdbcType="TIMESTAMP" property="updatedAt"/>
<result column="updated_by" jdbcType="VARCHAR" property="updatedBy" /> <result column="updated_by" jdbcType="VARCHAR" property="updatedBy"/>
<result column="semantic_type" jdbcType="VARCHAR" property="semanticType" /> <result column="semantic_type" jdbcType="VARCHAR" property="semanticType"/>
<result column="alias" jdbcType="VARCHAR" property="alias" /> <result column="alias" jdbcType="VARCHAR" property="alias"/>
<result column="default_values" jdbcType="VARCHAR" property="defaultValues" /> <result column="default_values" jdbcType="VARCHAR" property="defaultValues"/>
<result column="dim_value_maps" jdbcType="VARCHAR" property="dimValueMaps" /> <result column="dim_value_maps" jdbcType="VARCHAR" property="dimValueMaps"/>
</resultMap> </resultMap>
<resultMap extends="BaseResultMap" id="ResultMapWithBLOBs" type="com.tencent.supersonic.headless.server.persistence.dataobject.DimensionDO"> <resultMap extends="BaseResultMap" id="ResultMapWithBLOBs"
<result column="type_params" jdbcType="LONGVARCHAR" property="typeParams" /> type="com.tencent.supersonic.headless.server.persistence.dataobject.DimensionDO">
<result column="expr" jdbcType="LONGVARCHAR" property="expr" /> <result column="type_params" jdbcType="LONGVARCHAR" property="typeParams"/>
<result column="expr" jdbcType="LONGVARCHAR" property="expr"/>
</resultMap> </resultMap>
@@ -59,7 +60,7 @@
</sql> </sql>
<sql id="Base_Column_List"> <sql id="Base_Column_List">
id id
, name, biz_name, description, status, model_id, type, type_params, expr, datasource_id, , name, biz_name, description, status, model_id, type, type_params, expr,
created_at, created_by, updated_by, updated_at, semantic_type created_at, created_by, updated_by, updated_at, semantic_type
</sql> </sql>
@@ -86,21 +87,36 @@
<update id="batchUpdate" parameterType="java.util.List"> <update id="batchUpdate" parameterType="java.util.List">
<foreach collection="list" item="dimension" separator=";"> <foreach collection="list" item="dimension" separator=";">
update s2_dimension update s2_dimension
set name = #{dimension.name,jdbcType=VARCHAR}, set
biz_name = #{dimension.bizName,jdbcType=VARCHAR}, <if test="dimension.name != null and dimension.name !=''">name = #{dimension.name,jdbcType=VARCHAR},</if>
description = #{dimension.description,jdbcType=VARCHAR}, <if test="dimension.bizName != null and dimension.bizName !=''">biz_name =
status = #{dimension.status,jdbcType=INTEGER}, #{dimension.bizName,jdbcType=VARCHAR},
model_id = #{dimension.modelId,jdbcType=BIGINT}, </if>
type = #{dimension.type,jdbcType=VARCHAR}, <if test="dimension.description != null and dimension.description !=''">description =
type_params = #{dimension.typeParams,jdbcType=VARCHAR}, #{dimension.description,jdbcType=VARCHAR},
datasource_id = #{dimension.datasourceId,jdbcType=BIGINT}, </if>
created_at = #{dimension.createdAt,jdbcType=TIMESTAMP}, <if test="dimension.status != null">status = #{dimension.status,jdbcType=INTEGER},</if>
created_by = #{dimension.createdBy,jdbcType=VARCHAR}, <if test="dimension.modelId != null">model_id = #{dimension.modelId,jdbcType=BIGINT},</if>
updated_by = #{dimension.updatedBy,jdbcType=VARCHAR}, <if test="dimension.type != null and dimension.type !=''">type = #{dimension.type,jdbcType=VARCHAR},</if>
updated_at = #{dimension.updatedAt,jdbcType=TIMESTAMP}, <if test="dimension.typeParams != null and dimension.typeParams !=''">type_params =
semantic_type = #{dimension.semanticType,jdbcType=VARCHAR}, #{dimension.typeParams,jdbcType=VARCHAR},
sensitive_level = #{dimension.sensitiveLevel,jdbcType=INTEGER}, </if>
expr = #{dimension.expr,jdbcType=LONGVARCHAR} <if test="dimension.createdAt != null">created_at = #{dimension.createdAt,jdbcType=TIMESTAMP},</if>
<if test="dimension.createdBy != null and dimension.createdBy !=''">created_by =
#{dimension.createdBy,jdbcType=VARCHAR},
</if>
<if test="dimension.semanticType != null and dimension.semanticType !=''">semantic_type =
#{dimension.semanticType,jdbcType=VARCHAR},
</if>
<if test="dimension.sensitiveLevel != null and dimension.sensitiveLevel !=''">sensitive_level =
#{dimension.sensitiveLevel,jdbcType=INTEGER},
</if>
<if test="dimension.expr != null and dimension.expr !=''">expr = #{dimension.expr,jdbcType=LONGVARCHAR},
</if>
<if test="dimension.updatedBy != null and dimension.updatedBy !=''">updated_by =
#{dimension.updatedBy,jdbcType=VARCHAR},
</if>
<if test="dimension.updatedAt != null">updated_at = #{dimension.updatedAt,jdbcType=TIMESTAMP}</if>
where id = #{dimension.id,jdbcType=BIGINT} where id = #{dimension.id,jdbcType=BIGINT}
</foreach> </foreach>
</update> </update>

View File

@@ -97,6 +97,35 @@
</foreach> </foreach>
</update> </update>
<update id="batchUpdate" parameterType="java.util.List">
<foreach collection="list" item="metric" separator=";">
update s2_metric
set
<if test="metric.name != null and metric.name !=''">name = #{metric.name,jdbcType=VARCHAR},</if>
<if test="metric.bizName != null and metric.bizName !=''">biz_name = #{metric.bizName,jdbcType=VARCHAR},
</if>
<if test="metric.description != null and metric.description !=''">description =
#{metric.description,jdbcType=VARCHAR},
</if>
<if test="metric.status != null">status = #{metric.status,jdbcType=INTEGER},</if>
<if test="metric.modelId != null">model_id = #{metric.modelId,jdbcType=BIGINT},</if>
<if test="metric.type != null and metric.type !=''">type = #{metric.type,jdbcType=VARCHAR},</if>
<if test="metric.typeParams != null and metric.typeParams !=''">type_params =
#{metric.typeParams,jdbcType=VARCHAR},
</if>
<if test="metric.createdAt != null">created_at = #{metric.createdAt,jdbcType=TIMESTAMP},</if>
<if test="metric.createdBy != null and dimension.createdBy !=''">created_by =
#{metric.createdBy,jdbcType=VARCHAR},
</if>
<if test="metric.sensitiveLevel != null">sensitive_level = #{metric.sensitiveLevel,jdbcType=INTEGER},</if>
<if test="metric.updatedBy != null and metric.updatedBy !=''">updated_by =
#{metric.updatedBy,jdbcType=VARCHAR},
</if>
<if test="metric.updatedAt != null">updated_at = #{metric.updatedAt,jdbcType=TIMESTAMP}</if>
where id = #{metric.id,jdbcType=BIGINT}
</foreach>
</update>
<update id="batchPublish" parameterType="java.util.List"> <update id="batchPublish" parameterType="java.util.List">
<foreach collection="list" item="metric" separator=";"> <foreach collection="list" item="metric" separator=";">
update s2_metric update s2_metric

View File

@@ -1,8 +1,11 @@
.conversation { .conversation {
position: relative; position: relative;
width: 0; width: 248px;
height: 100%; height: 100%;
background: #fff; background: #f9f9f9;
padding: 10px 10px;
border-right: 1px solid #f1f1f1;
.rightSection { .rightSection {
width: 100%; width: 100%;

View File

@@ -435,17 +435,17 @@ const ClassDimensionTable: React.FC<Props> = ({}) => {
queryDimensionList({ ...filterParams, ...currentPagin }); queryDimensionList({ ...filterParams, ...currentPagin });
}} }}
toolBarRender={() => [ toolBarRender={() => [
<Button // <Button
key="create" // key="create"
type="primary" // type="primary"
onClick={() => { // onClick={() => {
toDimensionEditPage(domainId, modelId!, 0); // toDimensionEditPage(domainId, modelId!, 0);
// setDimensionItem(undefined); // // setDimensionItem(undefined);
// setCreateModalVisible(true); // // setCreateModalVisible(true);
}} // }}
> // >
// 创建维度
</Button>, // </Button>,
<BatchCtrlDropDownButton <BatchCtrlDropDownButton
key="ctrlBtnList" key="ctrlBtnList"
extenderList={['batchSensitiveLevel', 'exportTagButton']} extenderList={['batchSensitiveLevel', 'exportTagButton']}

View File

@@ -367,6 +367,15 @@ const MetricInfoCreateForm: React.FC<CreateFormProps> = ({
message.error('请输入度量表达式'); message.error('请输入度量表达式');
return; return;
} }
if (defineType === METRIC_DEFINE_TYPE.MEASURE) {
const { bizName, name, metricDefineByMeasureParams } = queryParams;
queryParams[queryParamsTypeParamsKey[METRIC_DEFINE_TYPE.MEASURE]].measures =
metricDefineByMeasureParams.measures.map((item: ISemantic.IMeasure) => {
return item.bizName === bizName && name ? { ...item, name } : item;
});
}
if (!dataFormatType) { if (!dataFormatType) {
delete queryParams.dataFormat; delete queryParams.dataFormat;
} }

View File

@@ -75,6 +75,14 @@ const MetricMeasuresFormTable: React.FC<Props> = ({
return tableData; return tableData;
}; };
useEffect(() => {
if (measuresParams?.measures) {
setSelectedKeys(measuresParams.measures.map((item: ISemantic.IMeasure) => item.bizName));
}
typeParams?.expr && setExprString(typeParams.expr);
}, [measuresParams, typeParams?.expr]);
useEffect(() => { useEffect(() => {
setTableData(getTableData()); setTableData(getTableData());
}, [measuresList, measuresParams]); }, [measuresList, measuresParams]);