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

Co-authored-by: lxwcodemonkey

headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/request/MetaBatchReq.java

@@ -22,6 +22,9 @@ public class MetaBatchReq {
      * 批量执行分类信息
      */
     private EventType type;
+
     private List<String> classifications;
 
+    private Integer sensitiveLevel;
+
 }

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

@@ -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,

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

@@ -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);

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

@@ -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();
+        List<MetricDO> metricDOS = getMetrics(metaBatchReq.getIds());
-        metricFilter.setIds(metaBatchReq.getIds());
-        List<MetricDO> metricDOS = metricRepository.getMetric(metricFilter);
         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();
+        List<MetricDO> metrics = getMetrics(metricIds);
-        metricFilter.setIds(metricIds);
-        List<MetricDO> metrics = metricRepository.getMetric(metricFilter);
         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();
+        List<MetricDO> metrics = getMetrics(metricIds);
-        metricFilter.setIds(metricIds);
-        List<MetricDO> metrics = metricRepository.getMetric(metricFilter);
         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();
+        List<MetricDO> metrics = getMetrics(metaBatchReq.getIds());
-        metricFilter.setIds(metaBatchReq.getIds());
-        List<MetricDO> metrics = metricRepository.getMetric(metricFilter);
         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();