From 460992d2583b6377989369721b096b924a532d57 Mon Sep 17 00:00:00 2001
From: lishuai <1793460677@qq.com>
Date: Fri, 24 Nov 2023 13:11:59 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8E=92=E5=BA=8F=E3=80=81=E5=88=9D=E5=A7=8B?=
 =?UTF-8?q?=E5=8C=96=E7=AD=96=E7=95=A5=E5=8A=A0=E8=BD=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../server/impl/CustomPolicyServiceImpl.java  | 27 ++++++++++++++++---
 1 file changed, 24 insertions(+), 3 deletions(-)

diff --git a/acs/nladmin-system/nlsso-server/src/main/java/org/nl/acs/custompolicy/server/impl/CustomPolicyServiceImpl.java b/acs/nladmin-system/nlsso-server/src/main/java/org/nl/acs/custompolicy/server/impl/CustomPolicyServiceImpl.java
index b7a8dd0..fee24c3 100644
--- a/acs/nladmin-system/nlsso-server/src/main/java/org/nl/acs/custompolicy/server/impl/CustomPolicyServiceImpl.java
+++ b/acs/nladmin-system/nlsso-server/src/main/java/org/nl/acs/custompolicy/server/impl/CustomPolicyServiceImpl.java
@@ -2,6 +2,7 @@ package org.nl.acs.custompolicy.server.impl;
 
 
 import cn.hutool.core.bean.BeanUtil;
+import cn.hutool.core.collection.CollectionUtil;
 import com.alibaba.fastjson.JSON;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
@@ -44,11 +45,24 @@ public class CustomPolicyServiceImpl extends CommonServiceImpl<CustomPolicyMappe
     List<DeviceStrategy> instructionStrategys;
 
     private final CustomPolicyMapper customPolicyMapper;
+    private final HashMap<String,List<CustomPolicyPlantDTO>> customPolicyPlantDTOMap;
 
+    /**
+     * 初始化策略
+     * @throws Exception
+     */
     @Override
     public void autoInitial() throws Exception {
-
-
+        LambdaQueryWrapper<CustomPolicy> wrapper = new LambdaQueryWrapper<>();
+        wrapper.eq(CustomPolicy::getIs_delete,0) ;
+        List<CustomPolicy> customPolicyList = customPolicyMapper.selectList(wrapper);
+        if (CollectionUtil.isNotEmpty(customPolicyList)){
+            List<CustomPolicy> customPolicies = customPolicyList.stream().filter(item -> item.getKey_value() != null).collect(Collectors.toList());
+            customPolicies.forEach(item -> {
+                CustomPolicyPlantVO customPolicyPlantVO = JSON.parseObject(item.getKey_value(), CustomPolicyPlantVO.class);
+                customPolicyPlantDTOMap.put(item.getUnit_code(),customPolicyPlantVO.getPlans());
+            });
+        }
     }
 
 
@@ -100,7 +114,14 @@ public class CustomPolicyServiceImpl extends CommonServiceImpl<CustomPolicyMappe
         customPolicy.setId(customPolicyPlantVO.getId());
         customPolicy.setUpdate_user(currentUsername);
         customPolicy.setUpdate_time(now);
-        customPolicy.setKey_value(JSON.toJSONString(customPolicyPlantVO));
+        if (CollectionUtil.isNotEmpty(customPolicyPlantVO.getPlans())) {
+            for (int i = 0; i < customPolicyPlantVO.getPlans().size(); i++) {
+                customPolicyPlantVO.getPlans().get(i).setSort(i);
+            }
+            customPolicy.setKey_value(JSON.toJSONString(customPolicyPlantVO));
+        }else {
+            customPolicy.setKey_value(null);
+        }
         customPolicyMapper.updateById(customPolicy);
     }