排序、初始化策略加载

This commit is contained in:
2023-11-24 13:11:59 +08:00
parent 5a63c40d70
commit 460992d258

View File

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