From c659ff50dfce120d02fa906a33f7b4ab20ffddc2 Mon Sep 17 00:00:00 2001
From: Lius <Lius2225@163.com>
Date: 星期一, 11 十一月 2024 11:37:21 +0800
Subject: [PATCH] update

---
 lxzn-module-mdc/src/main/java/org/jeecg/modules/mdc/service/impl/MdcDeviceCalendarServiceImpl.java |   37 ++++++++++++++++++++++---------------
 1 files changed, 22 insertions(+), 15 deletions(-)

diff --git a/lxzn-module-mdc/src/main/java/org/jeecg/modules/mdc/service/impl/MdcDeviceCalendarServiceImpl.java b/lxzn-module-mdc/src/main/java/org/jeecg/modules/mdc/service/impl/MdcDeviceCalendarServiceImpl.java
index 238de3a..ac17d84 100644
--- a/lxzn-module-mdc/src/main/java/org/jeecg/modules/mdc/service/impl/MdcDeviceCalendarServiceImpl.java
+++ b/lxzn-module-mdc/src/main/java/org/jeecg/modules/mdc/service/impl/MdcDeviceCalendarServiceImpl.java
@@ -7,10 +7,7 @@
 import org.apache.commons.lang3.StringUtils;
 import org.jeecg.modules.mdc.entity.*;
 import org.jeecg.modules.mdc.mapper.MdcDeviceCalendarMapper;
-import org.jeecg.modules.mdc.service.IMdcDeviceCalendarService;
-import org.jeecg.modules.mdc.service.IMdcEquipmentService;
-import org.jeecg.modules.mdc.service.IMdcShiftSubService;
-import org.jeecg.modules.mdc.service.IMdcVacationManagementService;
+import org.jeecg.modules.mdc.service.*;
 import org.jeecg.modules.mdc.vo.EquipmentCalendarVo;
 import org.jeecg.modules.mdc.vo.MdcDeviceCalendarQueryVo;
 import org.jeecg.modules.mdc.vo.MdcDeviceCalendarVo;
@@ -41,6 +38,9 @@
 
     @Resource
     private IMdcVacationManagementService mdcVacationManagementService;
+
+    @Resource
+    private IMdcEquipmentOvertimeService mdcEquipmentOvertimeService;
 
     /**
      * 鍒嗛〉鍒楄〃鏌ヨ
@@ -114,19 +114,16 @@
      */
     @Override
     public List<MdcDeviceCalendarVo> listByEquipmentAndDate(String equipmentId, List<String> stringDates) {
-        List<MdcDeviceCalendarVo> result = new ArrayList<>();
         try {
-            for (String stringDate : stringDates) {
-                List<MdcDeviceCalendarVo> mdcDeviceCalendarVos = this.baseMapper.listByEquipmentAndDate(equipmentId, stringDate);
-                if (mdcDeviceCalendarVos != null && !mdcDeviceCalendarVos.isEmpty()) {
-                    result.addAll(mdcDeviceCalendarVos);
-                }
+            List<MdcDeviceCalendarVo> mdcDeviceCalendarVos = this.baseMapper.listByEquipmentAndDates(equipmentId, stringDates);
+            if (mdcDeviceCalendarVos != null && !mdcDeviceCalendarVos.isEmpty()) {
+                return mdcDeviceCalendarVos;
+            } else {
+                return null;
             }
         } catch (Exception e) {
-            e.printStackTrace();
+            return null;
         }
-
-        return result;
     }
 
     /**
@@ -167,7 +164,7 @@
     public BigDecimal computeShiftTimeCount(String equipmentId, String validDate) {
         validDate = validDate.replaceAll("-", "");
         Integer shiftCount = this.baseMapper.computeShiftTimeCount(equipmentId, validDate);
-        return new BigDecimal(shiftCount).multiply(new BigDecimal("8")).multiply(new BigDecimal("60"));
+        return new BigDecimal(shiftCount);
     }
 
     /**
@@ -212,7 +209,17 @@
 
     @Override
     public List<MdcDeviceCalendarVo> listByEquipmentIdAndDate(String equipmentId, String date) {
-        return this.baseMapper.listByEquipmentAndDate(equipmentId, date);
+        List<MdcDeviceCalendarVo> mdcDeviceCalendarVos = this.baseMapper.listByEquipmentAndDate(equipmentId, date);
+        if (mdcDeviceCalendarVos != null && !mdcDeviceCalendarVos.isEmpty()) {
+            for (MdcDeviceCalendarVo mdcDeviceCalendarVo : mdcDeviceCalendarVos) {
+                List<MdcEquipmentOvertime> list = mdcEquipmentOvertimeService.list(new LambdaQueryWrapper<MdcEquipmentOvertime>().eq(MdcEquipmentOvertime::getEquipmentId, mdcDeviceCalendarVo.getEquipmentId()).eq(MdcEquipmentOvertime::getTheDate, mdcDeviceCalendarVo.getEndDate()));
+                if (list != null && !list.isEmpty()) {
+                    mdcDeviceCalendarVo.setOvertimeStartTime(list.get(0).getStartTime());
+                    mdcDeviceCalendarVo.setOvertimeEndTime(list.get(0).getEndTime());
+                }
+            }
+        }
+        return mdcDeviceCalendarVos;
     }
 
 }

--
Gitblit v1.9.3