[add] 策略操作接口优化

signal-optimize-service/src/main/java/net/wanji/opt/synthesis/pojo/StrategyControlDataVO.java

@@ -15,6 +15,8 @@ import java.util.List;
 @Data
 @ApiModel(value = "StrategyControlDataVO", description = "策略控制操作数据实体")
 public class StrategyControlDataVO {
+    @ApiModelProperty(value = "id")
+    private Integer id;
     @ApiModelProperty(value = "路口干线Id")
     private String bizId;
     @ApiModelProperty(value = "类型 0-路口 1-干线")

signal-optimize-service/src/main/java/net/wanji/opt/synthesis/service/impl/StrategyControlServiceImpl.java

@@ -3,6 +3,7 @@ package net.wanji.opt.synthesis.service.impl;
 import cn.hutool.core.date.DateUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.fasterxml.jackson.core.JsonProcessingException;
 import lombok.extern.slf4j.Slf4j;
 import net.wanji.common.enums.DateStyle;
 import net.wanji.common.framework.Constants;
@@ -19,6 +20,7 @@ import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
 
 import javax.annotation.Resource;
+import java.lang.reflect.InvocationTargetException;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Objects;
@@ -41,15 +43,7 @@ public class StrategyControlServiceImpl implements StrategyControlService {
 
         try {
             // 存库
-            List<StrategyControlDataEntity> entities = new ArrayList<>();
-            List<StrategyControlDataVO> dataList = strategyControlVO.getDataList();
-            for (StrategyControlDataVO dataVO : dataList) {
-                StrategyControlDataEntity entity = new StrategyControlDataEntity();
-                BeanUtils.copyProperties(entity, dataVO);
-                entity.setTime(JacksonUtils.getInstance().writeValueAsString(dataVO.getTime()));
-                entities.add(entity);
-            }
-            strategyControlInfoMapper.insertBatch(entities);
+            insertAndUpdate(strategyControlVO);
 
             StrategyControlReq req = convertReq(strategyControlVO);
             Result result = pushStrategyControlService.push(req);
@@ -69,6 +63,22 @@ public class StrategyControlServiceImpl implements StrategyControlService {
         }
     }
 
+    private void insertAndUpdate(StrategyControlVO strategyControlVO) throws IllegalAccessException, InvocationTargetException, JsonProcessingException {
+        List<StrategyControlDataVO> dataList = strategyControlVO.getDataList();
+        for (StrategyControlDataVO dataVO : dataList) {
+            StrategyControlDataEntity entity = new StrategyControlDataEntity();
+            BeanUtils.copyProperties(entity, dataVO);
+            entity.setTime(JacksonUtils.getInstance().writeValueAsString(dataVO.getTime()));
+            if (StringUtils.equals("insert", dataVO.getAction())) {
+                strategyControlInfoMapper.insert(entity);
+            } else if (StringUtils.equals("update", dataVO.getAction())) {
+                strategyControlInfoMapper.updateById(entity);
+            } else if (StringUtils.equals("delete", dataVO.getAction())) {
+                strategyControlInfoMapper.deleteById(entity);
+            }
+        }
+    }
+
     private StrategyControlReq convertReq(StrategyControlVO vo) throws Exception {
         try {
             List<StrategyControlDataVO> dataList = vo.getDataList();