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

(improvement)(headless) Batch Update Metric sensitiveLevel (#1167)

Browse Source 
Co-authored-by: lxwcodemonkey
This commit is contained in:
LXW
2024-06-20 11:56:30 +08:00
committed by GitHub
parent 5474fe1d04
commit 0d9afb5d87
4 changed files with 42 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
headless/server/src/main/java/com/tencent/supersonic/headless/server/rest/MetricController.java
View File

@@ -92,6 +92,15 @@ public class MetricController {
return true;
}
@PostMapping("/batchUpdateSensitiveLevel")
public Boolean batchUpdateSensitiveLevel(@RequestBody MetaBatchReq metaBatchReq,
HttpServletRequest request,
HttpServletResponse response) {
User user = UserHolder.findUser(request, response);
metricService.batchUpdateSensitiveLevel(metaBatchReq, user);
return true;
}
@PostMapping("/mockMetricAlias")
public List<String> mockMetricAlias(@RequestBody MetricBaseReq metricReq,
HttpServletRequest request,

2
headless/server/src/main/java/com/tencent/supersonic/headless/server/service/MetricService.java
View File

@@ -36,6 +36,8 @@ public interface MetricService {
void batchUpdateClassifications(MetaBatchReq metaBatchReq, User user);
void batchUpdateSensitiveLevel(MetaBatchReq metaBatchReq, User user);
void deleteMetric(Long id, User user) throws Exception;
PageInfo<MetricResp> queryMetricMarket(PageMetricReq pageMetricReq, User user);

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

@@ -2,6 +2,7 @@ package com.tencent.supersonic.headless.server.service.impl;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.TypeReference;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.google.common.collect.Lists;
@@ -17,6 +18,7 @@ import com.tencent.supersonic.common.pojo.enums.EventType;
import com.tencent.supersonic.common.pojo.enums.StatusEnum;
import com.tencent.supersonic.common.pojo.enums.TypeEnums;
import com.tencent.supersonic.common.util.BeanMapper;
import com.tencent.supersonic.headless.server.persistence.mapper.MetricDOMapper;
import com.tencent.supersonic.headless.server.utils.AliasGenerateHelper;
import com.tencent.supersonic.common.jsqlparser.SqlSelectFunctionHelper;
import com.tencent.supersonic.headless.api.pojo.DrillDownDimension;
@@ -89,7 +91,8 @@ import java.util.stream.Collectors;
@Service
@Slf4j
public class MetricServiceImpl implements MetricService {
public class MetricServiceImpl extends ServiceImpl<MetricDOMapper, MetricDO>
implements MetricService {
private MetricRepository metricRepository;
@@ -186,9 +189,7 @@ public class MetricServiceImpl implements MetricService {
if (CollectionUtils.isEmpty(metaBatchReq.getIds())) {
return;
}
MetricFilter metricFilter = new MetricFilter();
metricFilter.setIds(metaBatchReq.getIds());
List<MetricDO> metricDOS = metricRepository.getMetric(metricFilter);
List<MetricDO> metricDOS = getMetrics(metaBatchReq.getIds());
if (CollectionUtils.isEmpty(metricDOS)) {
return;
}
@@ -210,9 +211,7 @@ public class MetricServiceImpl implements MetricService {
@Override
public void batchPublish(List<Long> metricIds, User user) {
MetricFilter metricFilter = new MetricFilter();
metricFilter.setIds(metricIds);
List<MetricDO> metrics = metricRepository.getMetric(metricFilter);
List<MetricDO> metrics = getMetrics(metricIds);
for (MetricDO metricDO : metrics) {
metricDO.setUpdatedAt(new Date());
metricDO.setUpdatedBy(user.getName());
@@ -222,9 +221,7 @@ public class MetricServiceImpl implements MetricService {
@Override
public void batchUnPublish(List<Long> metricIds, User user) {
MetricFilter metricFilter = new MetricFilter();
metricFilter.setIds(metricIds);
List<MetricDO> metrics = metricRepository.getMetric(metricFilter);
List<MetricDO> metrics = getMetrics(metricIds);
for (MetricDO metricDO : metrics) {
metricDO.setUpdatedAt(new Date());
metricDO.setUpdatedBy(user.getName());
@@ -234,9 +231,7 @@ public class MetricServiceImpl implements MetricService {
@Override
public void batchUpdateClassifications(MetaBatchReq metaBatchReq, User user) {
MetricFilter metricFilter = new MetricFilter();
metricFilter.setIds(metaBatchReq.getIds());
List<MetricDO> metrics = metricRepository.getMetric(metricFilter);
List<MetricDO> metrics = getMetrics(metaBatchReq.getIds());
for (MetricDO metricDO : metrics) {
metricDO.setUpdatedAt(new Date());
metricDO.setUpdatedBy(user.getName());
@@ -267,6 +262,17 @@ public class MetricServiceImpl implements MetricService {
metricDO.setClassifications(classifications);
}
@Override
public void batchUpdateSensitiveLevel(MetaBatchReq metaBatchReq, User user) {
List<MetricDO> metrics = getMetrics(metaBatchReq.getIds());
for (MetricDO metricDO : metrics) {
metricDO.setUpdatedAt(new Date());
metricDO.setUpdatedBy(user.getName());
metricDO.setSensitiveLevel(metaBatchReq.getSensitiveLevel());
}
updateBatchById(metrics);
}
@Override
public void deleteMetric(Long id, User user) {
MetricDO metricDO = metricRepository.getMetricById(id);
@@ -349,6 +355,15 @@ public class MetricServiceImpl implements MetricService {
return metricRepository.getMetric(metricFilter);
}
private List<MetricDO> getMetrics(List<Long> ids) {
if (CollectionUtils.isEmpty(ids)) {
return Lists.newArrayList();
}
MetricFilter metricFilter = new MetricFilter();
metricFilter.setIds(ids);
return metricRepository.getMetric(metricFilter);
}
@Override
public List<MetricResp> getMetrics(MetaFilter metaFilter) {
MetricFilter metricFilter = new MetricFilter();