From 6221c8b5df5eb0ee62dce79048c9c3f0dc3d0fc1 Mon Sep 17 00:00:00 2001
From: fuliqi <fuliqi@qq.com>
Date: 星期六, 31 八月 2024 20:15:57 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java |   30 ++++++++++++++++++------------
 1 files changed, 18 insertions(+), 12 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java b/ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java
index 0cdb6d0..309b427 100644
--- a/ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java
+++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java
@@ -248,10 +248,10 @@
     }
 
     @Override
-    public Result importData(MultipartFile file, Integer unitId, Date startTime, Date endTime) throws IOException {
+    public Result importData(MultipartFile file, Integer unitId, Date startTime, Date endTime, Boolean needUpdateUnit) throws IOException {
         Consumer<List<PointExport>> consumer = (dataList) -> {
             try {
-                this.updatePoint(dataList, unitId, startTime, endTime);
+                this.updatePoint(dataList, unitId, startTime, endTime, needUpdateUnit);
             } catch (ExecutionException e) {
                 e.printStackTrace();
             } catch (InterruptedException e) {
@@ -269,7 +269,7 @@
      * @param unitId
      */
     @Transactional(rollbackFor = Exception.class)
-    public void updatePoint(List<PointExport> dataList, Integer unitId, Date startTime, Date endTime) throws ExecutionException, InterruptedException {
+    public void updatePoint(List<PointExport> dataList, Integer unitId, Date startTime, Date endTime, Boolean needUpdateUnit) throws ExecutionException, InterruptedException {
         if (CollectionUtils.isEmpty(dataList)) {
             throw new RuntimeException("瀵煎叆鏁版嵁涓嶈兘涓虹┖");
         }
@@ -277,11 +277,19 @@
             YwPoint point = new YwPoint();
             point.setImportantCommandImageTag("鏄�".equals(item.getImportantCommandImageTagString()));
             point.setProvinceTag("鏄�".equals(item.getProvinceTagString()));
-            if (Objects.nonNull(unitId)) {
-                point.setUnitId(Long.valueOf(unitId));
+            point.setDeptTag("鏄�".equals(item.getDeptTag()));
+            if (needUpdateUnit) {
+                if (Objects.nonNull(unitId)) {
+                    point.setUnitId(Long.valueOf(unitId));
+                }
+                point.setStartTime(startTime);
+                point.setEndTime(endTime);
+            } else {
+                point.setUnitId(null);
+                point.setStartTime(null);
+                point.setEndTime(null);
             }
-            point.setStartTime(startTime);
-            point.setEndTime(endTime);
+
             point.setPointName(item.getPointName());
             point.setSerialNumber(item.getSerialNumber());
             return point;
@@ -290,19 +298,17 @@
     }
 
     public void waitAllFinishAndGetResult(List<YwPoint> dataList) throws InterruptedException, ExecutionException {
-        List<FutureTask<Boolean>> resultList = new ArrayList<>(512);
-        List<Boolean> data = new ArrayList<>(512);
         int start = 0;
         Date startTime = new Date();
         while (true) {
-            if (dataList.size() < start + 100) {
+            if (dataList.size() < start + 800) {
                 List<YwPoint> list = dataList.subList(start, dataList.size() - 1);
                 this.baseMapper.updatePoint(list);
                 break;
             } else {
-                List<YwPoint> list = dataList.subList(start, start + 100);
+                List<YwPoint> list = dataList.subList(start, start + 800);
                 this.baseMapper.updatePoint(list);
-                start += 100;
+                start += 800;
             }
         }
         Date endTime = new Date();

--
Gitblit v1.8.0