From 4ad67c20b9c5e6625babefeee59ccee155cfc3aa Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Fri, 18 Jun 2021 16:54:10 +0800
Subject: [PATCH] Merge branch 'dev' of http://blit.7drlb.com:8888/r/moral into dev

---
 screen-manage/src/main/java/com/moral/api/service/impl/VersionServiceImpl.java |   41 +++++++++++++++++++++++++++++++++++++----
 1 files changed, 37 insertions(+), 4 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 d81b12f..26f3540 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,10 +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.Sensor;
-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;
@@ -41,6 +40,10 @@
     VersionMapper versionMapper;
     @Autowired
     VersionSensorUnitMapper versionSensorUnitMapper;
+    @Autowired
+    SensorMapper sensorMapper;
+    @Autowired
+    OrganizationUnitAlarmMapper organizationUnitAlarmMapper;
 
     @Override
     public VersionQueryDTO query(VersionQueryForm form) {
@@ -193,6 +196,11 @@
         deleteWrapper.eq("version_id", versionId);
         deleteWrapper.set("is_delete", Constants.DELETE);
         versionSensorUnitMapper.update(null, deleteWrapper);
+        //������������������id������code
+        for (VersionSensorUnit sensorUnit : sensorUnits) {
+            Sensor sensor = sensorMapper.selectById(sensorUnit.getSensorId());
+            sensorUnit.setSensorCode(sensor.getCode());
+        }
         //���������������������������������
         for (VersionSensorUnit sensorUnit : sensorUnits) {
             sensorUnit.setVersionId(versionId);
@@ -252,5 +260,30 @@
         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