topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-10 11:07:06 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(improvement)(Headless) Support model batch update status (#1035)

Browse Source
This commit is contained in:
LXW
2024-05-27 20:49:34 +08:00
committed by GitHub
parent f10cc90262
commit 46e2683338
3 changed files with 32 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/ModelServiceImpl.java
View File

@@ -114,32 +114,13 @@ public class ModelServiceImpl implements ModelService {
public ModelResp updateModel(ModelReq modelReq, User user) throws Exception {
checkName(modelReq);
ModelDO modelDO = modelRepository.getModelById(modelReq.getId());
int oldStatus = modelDO.getStatus();
ModelConverter.convert(modelDO, modelReq, user);
modelRepository.updateModel(modelDO);
batchCreateDimension(modelDO, user);
batchCreateMetric(modelDO, user);
statusPublish(oldStatus, modelDO);
return ModelConverter.convert(modelDO);
}
private void statusPublish(Integer oldStatus, ModelDO modelDO) {
if (oldStatus.equals(modelDO.getStatus())) {
return;
}
EventType eventType = null;
if (oldStatus.equals(StatusEnum.ONLINE.getCode())
&& modelDO.getStatus().equals(StatusEnum.OFFLINE.getCode())) {
eventType = EventType.DELETE;
} else if (oldStatus.equals(StatusEnum.OFFLINE.getCode())
&& modelDO.getStatus().equals(StatusEnum.ONLINE.getCode())) {
eventType = EventType.ADD;
}
log.info("model:{} status from {} to {}", modelDO.getId(), oldStatus, modelDO.getStatus());
metricService.sendMetricEventBatch(Lists.newArrayList(modelDO.getId()), eventType);
dimensionService.sendDimensionEventBatch(Lists.newArrayList(modelDO.getId()), eventType);
}
@Override
public List<ModelResp> getModelList(MetaFilter metaFilter) {
ModelFilter modelFilter = new ModelFilter();
@@ -342,7 +323,7 @@ public class ModelServiceImpl implements ModelService {
}
if (authTypeEnum.equals(AuthType.VISIBLE)) {
modelWithAuth = modelResps.stream()
.filter(domainResp -> checkDataSeterPermission(orgIds, user, domainResp))
.filter(domainResp -> checkDataSetPermission(orgIds, user, domainResp))
.collect(Collectors.toList());
}
return modelWithAuth;
@@ -429,6 +410,14 @@ public class ModelServiceImpl implements ModelService {
modelDO.setStatus(metaBatchReq.getStatus());
modelDO.setUpdatedAt(new Date());
modelDO.setUpdatedBy(user.getName());
if (StatusEnum.OFFLINE.getCode().equals(metaBatchReq.getStatus())
|| StatusEnum.DELETED.getCode().equals(metaBatchReq.getStatus())) {
metricService.sendMetricEventBatch(Lists.newArrayList(modelDO.getId()), EventType.DELETE);
dimensionService.sendDimensionEventBatch(Lists.newArrayList(modelDO.getId()), EventType.DELETE);
} else if (StatusEnum.ONLINE.getCode().equals(metaBatchReq.getStatus())) {
metricService.sendMetricEventBatch(Lists.newArrayList(modelDO.getId()), EventType.ADD);
dimensionService.sendDimensionEventBatch(Lists.newArrayList(modelDO.getId()), EventType.ADD);
}
})
.collect(Collectors.toList());
modelRepository.batchUpdate(modelDOS);
@@ -470,7 +459,7 @@ public class ModelServiceImpl implements ModelService {
return false;
}
public static boolean checkDataSeterPermission(Set<String> orgIds, User user, ModelResp modelResp) {
public static boolean checkDataSetPermission(Set<String> orgIds, User user, ModelResp modelResp) {
if (checkAdminPermission(orgIds, user, modelResp)) {
return true;
}

6
headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/TermServiceImpl.java
View File

@@ -11,6 +11,9 @@ import com.tencent.supersonic.headless.server.persistence.dataobject.TermDO;
import com.tencent.supersonic.headless.server.persistence.mapper.TermMapper;
import com.tencent.supersonic.headless.server.service.TermService;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
@@ -50,6 +53,9 @@ public class TermServiceImpl extends ServiceImpl<TermMapper, TermDO> implements
@Override
public Map<Long, List<TermResp>> getTermSets(Set<Long> domainIds) {
if (CollectionUtils.isEmpty(domainIds)) {
return new HashMap<>();
}
QueryWrapper<TermDO> queryWrapper = new QueryWrapper<>();
queryWrapper.lambda().in(TermDO::getDomainId, domainIds);
List<TermDO> list = list(queryWrapper);

16
headless/server/src/main/resources/mapper/custom/ModelDOCustomMapper.xml Normal file
View File

@@ -0,0 +1,16 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.tencent.supersonic.headless.server.persistence.mapper.ModelDOCustomMapper">
<update id="batchUpdateStatus" parameterType="java.util.List">
<foreach collection="list" item="model" separator=";">
update s2_model
set status = #{model.status,jdbcType=INTEGER},
updated_at = #{model.updatedAt,jdbcType=TIMESTAMP},
updated_by = #{model.updatedBy,jdbcType=VARCHAR}
where id = #{model.id,jdbcType=BIGINT}
</foreach>
</update>
</mapper>