From 369a252bfd68f5c02e285cd06d9bf482401e9d0e Mon Sep 17 00:00:00 2001
From: xiangpei <xiangpei@timesnew.cn>
Date: 星期三, 07 八月 2024 11:08:39 +0800
Subject: [PATCH] 点位时间优化

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/YwPointServiceImpl.java |    9 ++++++---
 1 files changed, 6 insertions(+), 3 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 eef5dce..354b073 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
@@ -36,6 +36,7 @@
 import org.springframework.web.multipart.MultipartFile;
 
 import java.io.IOException;
+import java.util.Date;
 import java.util.List;
 import java.util.Objects;
 import java.util.function.Consumer;
@@ -233,9 +234,9 @@
     }
 
     @Override
-    public Result importData(MultipartFile file, Integer unitId) throws IOException {
+    public Result importData(MultipartFile file, Integer unitId, Date startTime, Date endTime) throws IOException {
         Consumer<List<PointExport>> consumer = (dataList) -> {
-            this.updatePoint(dataList, unitId);
+            this.updatePoint(dataList, unitId, startTime, endTime);
         };
         EasyExcel.read(file.getInputStream(), PointExport.class , new CurrencyDataListener(consumer)).headRowNumber(1).doReadAll();
         return Result.ok();
@@ -247,7 +248,7 @@
      * @param dataList
      * @param unitId
      */
-    private void updatePoint(List<PointExport> dataList, Integer unitId) {
+    private void updatePoint(List<PointExport> dataList, Integer unitId, Date startTime, Date endTime) {
         if (CollectionUtils.isEmpty(dataList)) {
             throw new RuntimeException("瀵煎叆鏁版嵁涓嶈兘涓虹┖");
         }
@@ -255,6 +256,8 @@
         new LambdaUpdateChainWrapper<>(baseMapper)
                 .in(YwPoint::getSerialNumber, pointList)
                 .set(YwPoint::getUnitId, unitId)
+                .set(YwPoint::getStartTime, startTime)
+                .set(YwPoint::getEndTime, endTime)
                 .update();
     }
 

--
Gitblit v1.8.0