计算请假时长和出差时长

ゞ浴缸里的玫瑰 2022-11-26 07:47 200阅读 0赞

import lombok.Data;
    
    import java.text.ParseException;
    import java.text.SimpleDateFormat;
    import java.time.*;
    import java.time.format.DateTimeFormatter;
    import java.util.ArrayList;
    import java.util.Date;
    
    public class MyTest {
        public static void main(String[] args) {
            MyTest test = new MyTest();
            DateTime dateTime = new DateTime();
            dateTime.setStartTime("2020-10-10 14:38:00");
            dateTime.setEndTime("2020-10-10 15:28:00");
            System.out.println(test.Test(dateTime));
        }
    
        public static Double Test(DateTime dateTime) {
     Double dayHours = 0.0;
            SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:SS");
            try {
                // 获取请假开始时间
                Date startTime = format.parse(dateTime.getStartTime());
                // 获取请假结束时间
                Date endTime = format.parse(dateTime.getEndTime());
                LocalDate localDate = startTime.toInstant().atZone(ZoneId.systemDefault()).toLocalDate();
                LocalDate localDateEnd = endTime.toInstant().atZone(ZoneId.systemDefault()).toLocalDate();
                int days = localDateEnd.getDayOfYear() - localDate.getDayOfYear() - 1;
                int sum = days;
    
                // 请假整天数
                for (int index = 0; index < days; index++) {
                    LocalDate date = localDate.plusDays(index + 1);
                    DayOfWeek day = date.getDayOfWeek();
                    // 如果是星期六星期天，需要减去一天
                    String todayStr = date.format(DateTimeFormatter.ofPattern("yyyy-MM-dd"));
                    if (DayOfWeek.SUNDAY.equals(day) || DayOfWeek.SATURDAY.equals(day)) {
                        if (!dateTime.getMakeUpLeaveList().contains(todayStr)) {
                            sum--;
                        }
                    }
                    if (dateTime.getHolidayList().contains(todayStr) && !DayOfWeek.SUNDAY.equals(day) && !DayOfWeek.SATURDAY.equals(day)) {
                        sum--;
                    }
                }
                dayHours = sum <= dayHours ? dayHours : sum * 8;
                // 计算开始当天的时间
                LocalTime localDateTime = startTime.toInstant().atZone(ZoneId.systemDefault()).toLocalTime();
                LocalTime localDateEndTime = endTime.toInstant().atZone(ZoneId.systemDefault()).toLocalTime();
                if (isCalculate(localDate, dateTime)) {
    
                    if (localDateTime.getHour() == dateTime.getAmWorkStart() && localDateEnd.getDayOfYear() - localDate.getDayOfYear() > 0) {
                        dayHours += 8;
                    } else {
                        int time = localDateTime.getHour() < dateTime.getPmWorkStart()
                                && localDateTime.getHour() > dateTime.getAmWorkEnd() ? dateTime.getPmWorkStart() : localDateTime.getHour();
                        int minute = localDateTime.getHour() < dateTime.getPmWorkStart()
                                && localDateTime.getHour() > dateTime.getAmWorkEnd() ? 0 : localDateTime.getMinute();
                        if (localDate.getDayOfYear() == localDateEnd.getDayOfYear()
                                && localDateEndTime.getHour() <= dateTime.getAmWorkEnd()
                                && localDateTime.getHour() <= dateTime.getAmWorkEnd()
                                ||
                                localDate.getDayOfYear() == localDateEnd.getDayOfYear()
                                        && localDateEndTime.getHour() <= dateTime.getPmWorkEnd()
                                        && time <= dateTime.getPmWorkEnd()
                                        && localDateEndTime.getHour() >= dateTime.getPmWorkStart()
                                        && time >= dateTime.getPmWorkStart()
                        ) {
    
                            if (localDateTime.getHour() < dateTime.getAmWorkEnd() || localDateTime.getHour() >= dateTime.getPmWorkStart()) {
                                int h = localDateEndTime.getHour() - time;
                                int m = localDateEndTime.getMinute() - minute;
                                dayHours += h + (m > 0 ? 1 : 0);
                            }
                        } else {
                            // 早上的时间
                            if (localDateTime.getHour() < dateTime.getAmWorkEnd()) {
                                int AmDate = dateTime.getAmWorkStart() > localDateTime.getHour()
                                        ? dateTime.getAmWorkEnd() - dateTime.getAmWorkStart()
                                        : dateTime.getAmWorkEnd() - localDateTime.getHour();
                                if (localDateEnd.getDayOfYear() - localDate.getDayOfYear() > 0) {
                                    dayHours += AmDate + 4;
                                } else {
                                    dayHours += AmDate;
                                }
                            } else if (time >= dateTime.getPmWorkStart()) {
                                // 下午的时间
                                int pMDate = dateTime.getPmWorkEnd() > time ? dateTime.getPmWorkEnd() - time : 0;
                                dayHours += pMDate == 0 && dateTime.getPmWorkEnd() > time ? 1.0 : pMDate;
                            }
                        }
                    }
                }
                // 计算开始结束的时间
                if (isCalculate(localDateEnd, dateTime)) {
                    if (localDateEndTime.getHour() == dateTime.getPmWorkEnd() && localDateEnd.getDayOfYear() - localDate.getDayOfYear() > 0) {
                        dayHours += 8;
                    } else {
                        int time = localDateEndTime.getHour() < dateTime.getPmWorkStart() ? dateTime.getPmWorkStart() : localDateEndTime.getHour();
                        if (localDate.getDayOfYear() == localDateEnd.getDayOfYear()
                                && localDateEndTime.getHour() <= dateTime.getAmWorkEnd()
                                && localDateTime.getHour() <= dateTime.getAmWorkEnd()
                                ||
                                localDate.getDayOfYear() == localDateEnd.getDayOfYear()
                                        && localDateTime.getHour() <= dateTime.getPmWorkEnd()
                                        && time <= dateTime.getPmWorkEnd()
                                        && localDateTime.getHour() >= dateTime.getPmWorkStart()
                                        && time >= dateTime.getPmWorkStart()
                        ) {
    
                        } else {
                            if (localDateEndTime.getHour() < dateTime.getAmWorkEnd()) {
                                int AmDate = dateTime.getAmWorkStart() > localDateEndTime.getHour()
                                        ? dateTime.getAmWorkEnd() - dateTime.getAmWorkStart()
                                        : localDateEndTime.getHour() - dateTime.getAmWorkStart();
                                dayHours += AmDate == 0 ? 1.0 : AmDate;
                            } else {
                                // 下午的时间
                                int pMDate = dateTime.getPmWorkEnd() > time ? time - dateTime.getPmWorkStart() : 0;
                                if (localDateEndTime.getMinute() > 0 && localDateEndTime.getHour() >= dateTime.getPmWorkStart() && localDateEndTime.getHour() <= dateTime.getPmWorkEnd()) {
                                    pMDate += 1.0;
                                }
                                if (localDateEnd.getDayOfYear() - localDate.getDayOfYear() > 0) {
                                    if (localDateEndTime.getHour() > dateTime.getPmWorkEnd()) {
                                        dayHours += pMDate + 8;
                                    } else {
                                        dayHours += pMDate + dateTime.getAmWorkEnd() - dateTime.getAmWorkStart();
                                    }
                                } else {
                                    dayHours += pMDate;
                                }
                            }
                        }
                    }
                }
            } catch (ParseException e) {
                e.printStackTrace();
            }
    
            return dayHours;
           }
    }
    
    @Data
    /**
     * 这个实体的数据可以从数据中获取
     */
    class DateTime {
        private final SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:SS");
        private String startTime;
        private String endTime;
        private Integer AmWorkStart = 9;
        private Integer AmWorkEnd = 12;
        private Integer PmWorkStart = 14;
        private Integer PmWorkEnd = 18;
        private ArrayList<String> holidayList = new ArrayList<String>() {
       {
            add("2020-10-01");
            add("2020-10-02");
            add("2020-10-03");
            add("2020-10-04");
            add("2020-10-05");
            add("2020-10-06");
            add("2020-10-07");
            add("2020-10-08");
        }};
        private ArrayList<String> makeUpLeaveList = new ArrayList<String>() {
       {
            add("2020-10-10");
        }};
    
    
    }