From 2b83311bf491369ebfe1eea899bc5a475b1fa096 Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Fri, 11 Jun 2021 11:16:06 +0800
Subject: [PATCH] screen-manage                 修改分页功能

---
 screen-manage/src/main/java/com/moral/api/service/impl/VersionServiceImpl.java |   67 +++++++++++++++++++++++++++++++--
 1 files changed, 62 insertions(+), 5 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 d306c56..f99c16c 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,6 +4,7 @@
 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;
@@ -141,7 +142,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;
@@ -170,9 +171,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());
@@ -180,7 +181,63 @@
     }
 
     @Override
+    @Transactional
     public VersionDTO updateSensorUnits(VersionSensorUnitForm form) {
-        return null;
+        //������������������
+        VersionDTO dto = new VersionDTO();
+        //������
+        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);
+        //���������������������������������
+        for (VersionSensorUnit sensorUnit : sensorUnits) {
+            sensorUnit.setVersionId(versionId);
+            versionSensorUnitMapper.insert(sensorUnit);
+        }
+        //������������������
+        dto.setCode(ResponseCodeEnum.SUCCESS.getCode());
+        dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg());
+        return dto;
     }
+
+    @Override
+    public VersionDTO queryVersionById(VersionQueryByIdForm form) {
+        //������������������
+        VersionDTO dto = new VersionDTO();
+        //������
+        Integer id = form.getId();
+        Integer size = form.getSize();
+        Integer page = form.getPage();
+        //������������������������������������
+        List<Integer> versionsIds = new ArrayList<>();
+        versionsIds.add(id);
+        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);
+        return dto;
+    }
+
+
 }

--
Gitblit v1.8.0