Crunchy
2024-05-08 f6fd1f8aed6e7bb0c4dce40cf1cc48b8f4597432
performance-server/src/main/java/com/yuanchu/mom/service/impl/PerformanceShiftServiceImpl.java
@@ -1,5 +1,6 @@
package com.yuanchu.mom.service.impl;
import cn.hutool.core.date.DateUtil;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
@@ -12,11 +13,10 @@
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.text.SimpleDateFormat;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.*;
/**
 * <p>
@@ -36,11 +36,14 @@
    public void performanceShiftAdd(PerformanceShiftAddDto performanceShiftAddDto) {
        List<LocalDateTime> datesBetween = getLocalDateTimesBetween(performanceShiftAddDto.getStartWeek(), performanceShiftAddDto.getEndWeek());
        for (LocalDateTime date : datesBetween) {
            PerformanceShift performanceShift = new PerformanceShift();
            performanceShift.setShift(performanceShiftAddDto.getShift());
            performanceShift.setUserId(performanceShiftAddDto.getUserId());
            performanceShift.setWorkTime(date);
            baseMapper.insert(performanceShift);
            String[] splitUserId = performanceShiftAddDto.getUserId().split(",");
            for (String s : splitUserId) {
                PerformanceShift performanceShift = new PerformanceShift();
                performanceShift.setShift(performanceShiftAddDto.getShift());
                performanceShift.setUserId(Integer.valueOf(s));
                performanceShift.setWorkTime(date);
                baseMapper.insert(performanceShift);
            }
        }
    }
@@ -51,20 +54,28 @@
        for (Map<String, Object> i : mapIPage.getRecords()) {
            String[] shiftTimes = i.get("shiftTime").toString().split(";");
            int totalAttendance = 0;
            Map<Object, Object> map = new HashMap<>();
            for (Enums enums : shiftType) {
                Integer num = 0;
                int num = 0;
                // 分割日期
                for (String shiftTime : shiftTimes) {
                    Map<Object, Object> hashMap = new HashMap<>();
                    String[] shiftTimeAndShift = shiftTime.split(":");
                    if (enums.getValue().equals(shiftTimeAndShift[1])) {
                        num++;
                    }
                    // 3:休假;4:请假
                    if (!enums.getValue().equals("3") && enums.getValue().equals("4")) {
                        totalAttendance++;
                    }
                    hashMap.put("id", shiftTimeAndShift[2]);
                    hashMap.put("shift", shiftTimeAndShift[1]);
                    hashMap.put("weekly", DateUtil.weekOfYear(DateUtil.parse(shiftTimeAndShift[0])));
                    hashMap.put("time", shiftTimeAndShift[0]);
                    map.put(getWeek(shiftTimeAndShift[0]), hashMap);
                }
                i.put(enums.getLabel(), num);
                i.put("totalAttendance", totalAttendance);
                i.put("list", map);
                if(!enums.getValue().equals("3") && !enums.getValue().equals("4")) {
                    i.put("totalAttendance", totalAttendance += num);
                }
                i.remove("shiftTime");
            }
        }
        return mapIPage;
@@ -90,4 +101,40 @@
        }
        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 "未知";
        }
    }
}