From 7e51076a4cccc27df31bcdd98ca14c2f9d4928e9 Mon Sep 17 00:00:00 2001
From: value <z1292839451@163.com>
Date: 星期四, 09 五月 2024 07:27:04 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 performance-server/src/main/java/com/yuanchu/mom/service/impl/PerformanceShiftServiceImpl.java |  181 +++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 181 insertions(+), 0 deletions(-)

diff --git a/performance-server/src/main/java/com/yuanchu/mom/service/impl/PerformanceShiftServiceImpl.java b/performance-server/src/main/java/com/yuanchu/mom/service/impl/PerformanceShiftServiceImpl.java
new file mode 100644
index 0000000..b69566d
--- /dev/null
+++ b/performance-server/src/main/java/com/yuanchu/mom/service/impl/PerformanceShiftServiceImpl.java
@@ -0,0 +1,181 @@
+package com.yuanchu.mom.service.impl;
+
+import cn.hutool.core.date.DateUtil;
+import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.yuanchu.mom.dto.PerformanceShiftAddDto;
+import com.yuanchu.mom.mapper.PerformanceShiftMapper;
+import com.yuanchu.mom.pojo.Enums;
+import com.yuanchu.mom.pojo.PerformanceShift;
+import com.yuanchu.mom.service.EnumService;
+import com.yuanchu.mom.service.PerformanceShiftService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.text.SimpleDateFormat;
+import java.time.LocalDate;
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
+import java.time.temporal.TemporalAdjusters;
+import java.util.*;
+
+/**
+ * <p>
+ * 缁╂晥绠$悊-鐝 鏈嶅姟瀹炵幇绫�
+ * </p>
+ *
+ * @author 姹熻嫃榈烽洀缃戠粶绉戞妧鏈夐檺鍏徃
+ * @since 2024-05-08 09:12:04
+ */
+@Service
+public class PerformanceShiftServiceImpl extends ServiceImpl<PerformanceShiftMapper, PerformanceShift> implements PerformanceShiftService {
+
+    @Autowired
+    private EnumService enumService;
+
+    public List<PerformanceShift> list = new ArrayList<>();
+
+    @Transactional(rollbackFor = Exception.class)
+    @Override
+    public void performanceShiftAdd(PerformanceShiftAddDto performanceShiftAddDto) {
+        DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd");
+        String formattedDateTime = performanceShiftAddDto.getStartWeek().format(formatter);
+        boolean exists = baseMapper.exists(Wrappers.<PerformanceShift>lambdaQuery()
+                .eq(PerformanceShift::getWorkTime, formattedDateTime));
+        String[] splitUserId = performanceShiftAddDto.getUserId().split(",");
+        // 濡傛灉涓嶅瓨鍦ㄦ坊鍔犳暟鎹�
+        if (!exists) {
+            LocalDate firstDayOfMonth = performanceShiftAddDto.getEndWeek().toLocalDate().withDayOfMonth(1);
+            LocalDate lastDayOfMonth = performanceShiftAddDto.getEndWeek().toLocalDate().with(TemporalAdjusters.lastDayOfMonth());
+            List<LocalDateTime> localDateTimesBetween = getLocalDateTimesBetween(firstDayOfMonth.atStartOfDay(), lastDayOfMonth.atStartOfDay());
+            localDateTimesBetween.forEach(i -> {
+                for (String s : splitUserId) {
+                    PerformanceShift performanceShift = new PerformanceShift();
+                    performanceShift.setUserId(Integer.valueOf(s));
+                    performanceShift.setWorkTime(i);
+                    performanceShift.setShift("");
+                    list.add(performanceShift);
+                }
+                if (list.size() >= 1000) {
+                    baseMapper.insertBatchSomeColumn(list);
+                    list.clear();
+                }
+            });
+            if (!list.isEmpty()) {
+                baseMapper.insertBatchSomeColumn(list);
+            }
+        }
+        // 鍐嶆鏇存柊
+        List<LocalDateTime> datesBetween = getLocalDateTimesBetween(performanceShiftAddDto.getStartWeek(), performanceShiftAddDto.getEndWeek());
+        for (LocalDateTime date : datesBetween) {
+            for (String s : splitUserId) {
+                PerformanceShift performanceShift = new PerformanceShift();
+                performanceShift.setShift(performanceShiftAddDto.getShift());
+                performanceShift.setUserId(Integer.valueOf(s));
+                performanceShift.setWorkTime(date);
+                String formatterDateTime = date.format(formatter);
+                baseMapper.update(new PerformanceShift(), Wrappers.<PerformanceShift>lambdaUpdate()
+                        .set(PerformanceShift::getShift, performanceShiftAddDto.getShift())
+                        .eq(PerformanceShift::getUserId, s)
+                        .eq(PerformanceShift::getWorkTime, formatterDateTime));
+            }
+        }
+    }
+
+    @Override
+    public IPage<Map<String, Object>> performanceShiftPage(Page<Object> page, String time, String userName, String laboratory) {
+        IPage<Map<String, Object>> mapIPage = baseMapper.performanceShiftPage(page, time, userName, laboratory);
+        List<Enums> shiftType = enumService.selectEnumByCategory("鐝绫诲瀷");
+        for (Map<String, Object> i : mapIPage.getRecords()) {
+            String[] shiftTimes = i.get("shiftTime").toString().split(";");
+            int totalAttendance = 0;
+            List<Object> map = new ArrayList<>();
+            // 鍒嗗壊鏃ユ湡
+            for (String shiftTime : shiftTimes) {
+                Map<Object, Object> hashMap = new HashMap<>();
+                String[] shiftTimeAndShift = shiftTime.split("锛�");
+                for (Enums enums : shiftType) {
+                    if (!i.containsKey(enums.getLabel())) {
+                        i.put(enums.getLabel(), 0);
+                    }
+                    if (enums.getValue().equals(shiftTimeAndShift[1])) {
+                        Integer num = (Integer) i.get(enums.getLabel());
+                        i.put(enums.getLabel(), num += 1);
+                    }
+                }
+                if (shiftTimeAndShift[1].equals("1") || shiftTimeAndShift[1].equals("2") || shiftTimeAndShift[1].equals("0")) {
+                    i.put("totalAttendance", totalAttendance += 1);
+                }
+                hashMap.put("id", shiftTimeAndShift[2]);
+                hashMap.put("shift", shiftTimeAndShift[1]);
+                hashMap.put("weekly", DateUtil.weekOfYear(DateUtil.parse(shiftTimeAndShift[0])));
+                hashMap.put("time", shiftTimeAndShift[0]);
+                hashMap.put("headerTime", getWeek(shiftTimeAndShift[0]));
+                map.add(hashMap);
+            }
+            i.put("list", map);
+            i.remove("shiftTime");
+        }
+        return mapIPage;
+    }
+
+    @Override
+    public void performanceShiftUpdate(String shift, Integer id) {
+        PerformanceShift performanceShift = new PerformanceShift();
+        performanceShift.setId(id);
+        performanceShift.setShift(shift);
+        baseMapper.updateById(performanceShift);
+    }
+
+    // 鑾峰彇涓や釜localDateTime鐨勬瘡涓�澶�
+    public static List<LocalDateTime> getLocalDateTimesBetween(LocalDateTime start, LocalDateTime end) {
+        List<LocalDateTime> localDateTimes = new ArrayList<>();
+        LocalDate currentDate = start.toLocalDate();
+        LocalDateTime currentLocalDateTime = start;
+        while (!currentDate.isAfter(end.toLocalDate())) {
+            localDateTimes.add(currentLocalDateTime);
+            currentLocalDateTime = currentLocalDateTime.plusDays(1);
+            currentDate = currentDate.plusDays(1);
+        }
+        return localDateTimes;
+    }
+
+    public static String getWeek(String dayStr) {
+        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+        try {
+            Date date = sdf.parse(dayStr);
+            Calendar calendar = Calendar.getInstance();
+            calendar.setTime(date);
+            int dayOfWeek = calendar.get(Calendar.DAY_OF_WEEK);
+            int day = calendar.get(Calendar.DAY_OF_MONTH);
+            return day + " " + getWeekDay(dayOfWeek);
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        return null;
+    }
+
+    public static String getWeekDay(int dayOfWeek) {
+        switch (dayOfWeek) {
+            case Calendar.MONDAY:
+                return "鍛ㄤ竴";
+            case Calendar.TUESDAY:
+                return "鍛ㄤ簩";
+            case Calendar.WEDNESDAY:
+                return "鍛ㄤ笁";
+            case Calendar.THURSDAY:
+                return "鍛ㄥ洓";
+            case Calendar.FRIDAY:
+                return "鍛ㄤ簲";
+            case Calendar.SATURDAY:
+                return "鍛ㄥ叚";
+            case Calendar.SUNDAY:
+                return "鍛ㄦ棩";
+            default:
+                return "鏈煡";
+        }
+    }
+}

--
Gitblit v1.9.3