From d673c2d0cbeeeb3ab788607ecef3d3b953054d2d Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Wed, 23 Jun 2021 14:48:10 +0800
Subject: [PATCH] screen-manage               修复update语句bug

---
 screen-manage/src/main/java/com/moral/api/service/impl/VersionServiceImpl.java |  163 ++++++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 144 insertions(+), 19 deletions(-)

diff --git a/screen-manage/src/main/java/com/moral/api/service/impl/VersionServiceImpl.java b/screen-manage/src/main/java/com/moral/api/service/impl/VersionServiceImpl.java
index 4e79d21..92e0bf6 100644
--- a/screen-manage/src/main/java/com/moral/api/service/impl/VersionServiceImpl.java
+++ b/screen-manage/src/main/java/com/moral/api/service/impl/VersionServiceImpl.java
@@ -4,9 +4,9 @@
 import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.moral.api.config.mybatis.wrapper.NullFilterWrapper;
-import com.moral.api.entity.User;
-import com.moral.api.entity.Version;
-import com.moral.api.entity.VersionSensorUnit;
+import com.moral.api.entity.*;
+import com.moral.api.mapper.OrganizationUnitAlarmMapper;
+import com.moral.api.mapper.SensorMapper;
 import com.moral.api.mapper.VersionMapper;
 import com.moral.api.mapper.VersionSensorUnitMapper;
 import com.moral.api.pojo.dto.version.VersionDTO;
@@ -40,6 +40,10 @@
     VersionMapper versionMapper;
     @Autowired
     VersionSensorUnitMapper versionSensorUnitMapper;
+    @Autowired
+    SensorMapper sensorMapper;
+    @Autowired
+    OrganizationUnitAlarmMapper organizationUnitAlarmMapper;
 
     @Override
     public VersionQueryDTO query(VersionQueryForm form) {
@@ -141,7 +145,7 @@
         QueryWrapper<Version> wrapper = new QueryWrapper<>();
         wrapper.setEntity(existVersion);
         Version existVersionResult = versionMapper.selectOne(wrapper);
-        if(!ObjectUtils.isEmpty(existVersionResult)){
+        if (!ObjectUtils.isEmpty(existVersionResult)) {
             dto.setCode(ResponseCodeEnum.VERSION_EXIST.getCode());
             dto.setMsg(ResponseCodeEnum.VERSION_EXIST.getMsg());
             return dto;
@@ -161,6 +165,16 @@
         VersionDTO dto = new VersionDTO();
         //������
         Integer id = form.getId();
+        //������������������������������������������������������������
+        QueryWrapper<OrganizationUnitAlarm> queryWrapper = new QueryWrapper<>();
+        queryWrapper.eq("is_delete",Constants.NOT_DELETE);
+        queryWrapper.eq("version_id",id);
+        List<OrganizationUnitAlarm> organizationUnitAlarms = organizationUnitAlarmMapper.selectList(queryWrapper);
+        if(!ObjectUtils.isEmpty(organizationUnitAlarms)){
+            dto.setCode(ResponseCodeEnum.VERSION_USED.getCode());
+            dto.setMsg(ResponseCodeEnum.VERSION_USED.getMsg());
+            return dto;
+        }
         //������������������������������������������
         Version oldVersion = versionMapper.selectById(id);
         //������������
@@ -170,9 +184,9 @@
         versionMapper.updateById(deleteVersion);
         //������������������������������������
         UpdateWrapper deleteSensorUnitWrapper = new UpdateWrapper();
-        deleteSensorUnitWrapper.eq("version_id",id);
-        deleteSensorUnitWrapper.set("is_delete",Constants.DELETE);
-        versionSensorUnitMapper.update(null,deleteSensorUnitWrapper);
+        deleteSensorUnitWrapper.eq("version_id", id);
+        deleteSensorUnitWrapper.set("is_delete", Constants.DELETE);
+        versionSensorUnitMapper.update(null, deleteSensorUnitWrapper);
         //������������������
         dto.setCode(ResponseCodeEnum.SUCCESS.getCode());
         dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg());
@@ -187,15 +201,67 @@
         //������
         List<VersionSensorUnit> sensorUnits = form.getSensorUnits();
         Integer versionId = form.getVersionId();
-        //������������������������������������
-        UpdateWrapper deleteWrapper = new UpdateWrapper();
-        deleteWrapper.eq("version_id",versionId);
-        deleteWrapper.set("is_delete",Constants.DELETE);
-        versionSensorUnitMapper.update(null,deleteWrapper);
-        //���������������������������������
+        //������������������id������code
         for (VersionSensorUnit sensorUnit : sensorUnits) {
-            sensorUnit.setVersionId(versionId);
+            Sensor sensor = sensorMapper.selectById(sensorUnit.getSensorId());
+            sensorUnit.setSensorCode(sensor.getCode());
+        }
+        //������������������������������������
+        QueryWrapper<VersionSensorUnit> queryOldWrapper = new QueryWrapper<>();
+        queryOldWrapper.eq("version_id",versionId);
+        queryOldWrapper.eq("is_delete",Constants.NOT_DELETE);
+        List<VersionSensorUnit> oldSensorUnits = versionSensorUnitMapper.selectList(queryOldWrapper);
+        //������������������������(������������������������)
+        List<VersionSensorUnit> insertList = new ArrayList<>();
+        List<VersionSensorUnit> deleteList = new ArrayList<>();
+        sensorUnits.forEach(value->{
+            value.setVersionId(versionId);
+            if(!oldSensorUnits.contains(value))
+                insertList.add(value);
+        });
+        oldSensorUnits.forEach(value->{
+            if(!sensorUnits.contains(value))
+                deleteList.add(value);
+        });
+        //������������
+        UpdateWrapper deleteWrapper = new UpdateWrapper();
+        List<Integer> deleteIds = new ArrayList<>();
+        deleteList.forEach(value->deleteIds.add(value.getId()));
+        deleteWrapper.in("id",deleteIds);
+        deleteWrapper.set("is_delete", Constants.DELETE);
+        versionSensorUnitMapper.update(null, deleteWrapper);
+        //������������
+        for (VersionSensorUnit sensorUnit : insertList) {
             versionSensorUnitMapper.insert(sensorUnit);
+        }
+        //���������������������������
+        QueryWrapper<OrganizationUnitAlarm> queryOrgUnitAlarmWrapper = new QueryWrapper<>();
+        queryOrgUnitAlarmWrapper.select("distinct organization_id");
+        queryOrgUnitAlarmWrapper.eq("version_id",versionId);
+        queryOrgUnitAlarmWrapper.eq("is_delete",Constants.NOT_DELETE);
+        List<OrganizationUnitAlarm> organizationUnitAlarms = organizationUnitAlarmMapper.selectList(queryOrgUnitAlarmWrapper);
+        List<Integer> organizationIds = new ArrayList<>();
+        organizationUnitAlarms.forEach(value->organizationIds.add(value.getOrganizationId()));
+        for (Integer organizationId : organizationIds) {
+            //������������
+            for (VersionSensorUnit versionSensorUnit : deleteList) {
+                UpdateWrapper deleteOrganizationUnitAlarmWrapper = new UpdateWrapper();
+                deleteOrganizationUnitAlarmWrapper.eq("organization_id",organizationId);
+                deleteOrganizationUnitAlarmWrapper.eq("version_id",versionId);
+                deleteOrganizationUnitAlarmWrapper.eq("sensor_code",versionSensorUnit.getSensorCode());
+                deleteOrganizationUnitAlarmWrapper.set("is_delete",Constants.DELETE);
+                organizationUnitAlarmMapper.update(null,deleteOrganizationUnitAlarmWrapper);
+            }
+            //������������
+            for (VersionSensorUnit versionSensorUnit : insertList) {
+                OrganizationUnitAlarm organizationUnitAlarm = new OrganizationUnitAlarm();
+                organizationUnitAlarm.setOrganizationId(organizationId);
+                organizationUnitAlarm.setVersionId(versionId);
+                organizationUnitAlarm.setSensorCode(versionSensorUnit.getSensorCode());
+                organizationUnitAlarm.setUnitKey(versionSensorUnit.getUnitKey());
+                organizationUnitAlarm.setShowUnitKey(versionSensorUnit.getUnitKey());
+                organizationUnitAlarmMapper.insert(organizationUnitAlarm);
+            }
         }
         //������������������
         dto.setCode(ResponseCodeEnum.SUCCESS.getCode());
@@ -204,18 +270,77 @@
     }
 
     @Override
-    public VersionDTO queryVersionById(Integer id) {
+    public VersionQueryDTO queryVersionById(VersionQueryByIdForm form) {
         //������������������
-        VersionDTO dto = new VersionDTO();
+        VersionQueryDTO dto = new VersionQueryDTO();
+        //������
+        Integer id = form.getId();
+        Integer size = form.getSize();
+        Integer page = form.getPage();
         //������������������������������������
         List<Integer> versionsIds = new ArrayList<>();
         versionsIds.add(id);
-        List<Version> version = new ArrayList<>();
-        version = versionMapper.queryVersionsAndSensorUnitByIds(versionsIds);
+        List<Version> versions = new ArrayList<>();
+        versions = versionMapper.queryVersionsAndSensorUnitByIds(versionsIds);
+        //������������������������������������
+        Version version = versions.get(0);
+        List<Sensor> querySensors = version.getSensors();
+        if (!ObjectUtils.isEmpty(querySensors)) {
+            int startIndex = (page - 1) * size;
+            int endIndex = startIndex + size - 1;
+            List<Sensor> resultSensors = new ArrayList<>();
+
+            for (int i = startIndex; i <= endIndex; i++) {
+                if (i >= querySensors.size())
+                    break;
+                resultSensors.add(querySensors.get(i));
+            }
+            version.setSensors(resultSensors);
+        }
         //������������������
         dto.setCode(ResponseCodeEnum.SUCCESS.getCode());
         dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg());
-        dto.setVersion(version.get(0));
+        if (ObjectUtils.isEmpty(querySensors)) {
+            dto.setTotal(0);
+            dto.setPages(0);
+        }else{
+            dto.setTotal(querySensors.size());
+            double querySize = (double)querySensors.size();
+            double dSize = (double)size;
+            dto.setPages((int)Math.ceil(querySize/dSize));
+        }
+        dto.setSize(size);
+        dto.setCurrent(page);
+        VersionDTO versionDTO = new VersionDTO();
+        versionDTO.setVersion(version);
+        dto.setVersionDTOS(Arrays.asList(versionDTO));
         return dto;
     }
+
+    @Override
+    public VersionQueryDTO queryByOrganizationId(Integer organizationId) {
+        //������������������
+        VersionQueryDTO dto = new VersionQueryDTO();
+        //������������id
+        QueryWrapper<OrganizationUnitAlarm> queryVersionIdsWrapper = new QueryWrapper<>();
+        queryVersionIdsWrapper.select("DISTINCT version_id").eq("is_delete",Constants.NOT_DELETE).eq("organization_id",organizationId);
+        List<OrganizationUnitAlarm> organizationUnitAlarms = organizationUnitAlarmMapper.selectList(queryVersionIdsWrapper);
+        List<Integer> versionIds = new ArrayList<>();
+        for (OrganizationUnitAlarm organizationUnitAlarm : organizationUnitAlarms) {
+            versionIds.add(organizationUnitAlarm.getVersionId());
+        }
+        //������������id������������
+        List<Version> versions = new ArrayList<>();
+        if(!ObjectUtils.isEmpty(versionIds))
+            versions = versionMapper.selectBatchIds(versionIds);
+        //������������������
+        List<VersionDTO> versionDTOS = new ArrayList<>();
+        versions.forEach(value->versionDTOS.add(new VersionDTO(value)));
+        dto.setCode(ResponseCodeEnum.SUCCESS.getCode());
+        dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg());
+        dto.setVersionDTOS(versionDTOS);
+        return dto;
+    }
+
+
 }

--
Gitblit v1.8.0