feat: 新增手动生成排班接口

guzijian
1 parent 010828aa
Showing 8 changed files with 306 additions and 10 deletions
Bsth-admin/src/main/java/com/ruoyi/controller/AttendanceController.java
Bsth-admin/src/main/java/com/ruoyi/mapper/RuleNumSettingMapper.java
Bsth-admin/src/main/java/com/ruoyi/service/AttendanceService.java
Bsth-admin/src/main/java/com/ruoyi/service/RuleNumSettingService.java
Bsth-admin/src/main/java/com/ruoyi/service/impl/AttendanceServiceImpl.java
Bsth-admin/src/main/java/com/ruoyi/service/impl/RuleNumSettingServiceImpl.java
Bsth-admin/src/main/java/com/ruoyi/utils/ConstDateUtil.java
Bsth-admin/src/main/resources/mapper/RuleNumSettingMapper.xml
@@ -150,4 +150,12 @@ public class AttendanceController extends BaseController {
         return Result.OK(attendanceService.getTodayFreeUser(date));
     }
+    @ApiOperation("生成指定月份的排班")
+    @PostMapping("/create/attendance/{month}")
+    public Result<?> createAttendance(@PathVariable("month") String month){
+        attendanceService.createAttendance(month);
+        return Result.OK();
+    }
+
+
 }
@@ -6,6 +6,7 @@ import com.ruoyi.driver.domain.Driver;
 import com.ruoyi.pojo.dto.AttendanceDto;
 import com.ruoyi.pojo.vo.SchedulingRequestVo;
 import com.ruoyi.pojo.vo.SchedulingResponseVo;
+import org.apache.ibatis.annotations.Param;
 import java.util.List;
@@ -24,6 +25,13 @@ public interface RuleNumSettingMapper extends BaseMapper&lt;RuleNumSetting&gt; {
      * @return
      */
     List<AttendanceDto> getPeopleList();
+
+    /**
+     * 获取指定月份的日期
+     * @param month
+     * @return
+     */
+    List<AttendanceDto> getPeopleListByMonth(@Param("month") String month);
 }
@@ -74,4 +74,10 @@ public interface AttendanceService {
      * @return
      */
     List<PeopleResponseVo> getTodayFreeUser(String date);
+
+    /**
+     * 生成指定月份的日期排班
+     * @param month
+     */
+    void createAttendance(String month);
 }
@@ -2,6 +2,7 @@ package com.ruoyi.service;
 import com.baomidou.mybatisplus.extension.service.IService;
 import com.ruoyi.domain.RuleNumSetting;
+import com.ruoyi.pojo.dto.AttendanceDto;
 import com.ruoyi.pojo.dto.RuleNumDto;
 import com.ruoyi.pojo.vo.SchedulingRequestVo;
 import com.ruoyi.pojo.vo.SchedulingResponseVo;
@@ -38,4 +39,11 @@ public interface RuleNumSettingService extends IService&lt;RuleNumSetting&gt; {
      * 手动添加班次
      */
     void manualAddBcCache();
+
+    /**
+     * 获取指定月份的人员
+     * @param month
+     * @return
+     */
+    List<AttendanceDto> getPeopleListByMonth(String month);
 }
@@ -13,10 +13,7 @@ import com.ruoyi.domain.*;
 import com.ruoyi.driver.mapper.DriverMapper;
 import com.ruoyi.num.domain.RuleNum;
 import com.ruoyi.num.service.IRuleNumService;
-import com.ruoyi.pojo.dto.DriverDto;
-import com.ruoyi.pojo.dto.RuleSchedulingDto;
-import com.ruoyi.pojo.dto.SchedulingDto;
-import com.ruoyi.pojo.dto.SchedulingSettingDto;
+import com.ruoyi.pojo.dto.*;
 import com.ruoyi.pojo.vo.*;
 import com.ruoyi.scheduling.domain.RuleScheduling;
 import com.ruoyi.scheduling.service.IRuleSchedulingService;
@@ -31,12 +28,14 @@ import org.springframework.transaction.annotation.Transactional;
 import javax.validation.constraints.NotBlank;
 import java.time.LocalDate;
 import java.time.ZoneId;
+import java.time.temporal.ChronoUnit;
 import java.util.*;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 import static com.ruoyi.common.RuleSchedulingProperties.*;
 import static com.ruoyi.common.WeekEnum.FREE;
+import static com.ruoyi.service.impl.RuleNumSettingServiceImpl.getFreeRuleSchedulingDto;
 /**
  * @author 20412
@@ -68,6 +67,9 @@ public class AttendanceServiceImpl implements AttendanceService {
     private RuleAttendanceMainService attendanceMainService;
     @Autowired
+    private IRuleSchedulingService schedulingService;
+
+    @Autowired
     private RuleAttendanceMainHelpService helpService;
     @Autowired
@@ -179,7 +181,7 @@ public class AttendanceServiceImpl implements AttendanceService {
         attendanceMainService.updateByJobCodeTotalQuantity(vo, mainList);
     }
-    private static  List<RuleAttendanceMain> handleRuleUpdate(UpdateAttendanceVo vo, RuleAttendanceMain main, RuleScheduling ruleScheduling) {
+    private static List<RuleAttendanceMain> handleRuleUpdate(UpdateAttendanceVo vo, RuleAttendanceMain main, RuleScheduling ruleScheduling) {
         List<RuleAttendanceMain> mainList = new ArrayList<>();
         LocalDate startDate = LocalDate.parse(vo.getStartDate());
         LocalDate endDate = LocalDate.parse(vo.getEndDate());
@@ -192,7 +194,7 @@ public class AttendanceServiceImpl implements AttendanceService {
     private static void handleSegmentation(RuleAttendanceMain main, RuleScheduling ruleScheduling, List<RuleAttendanceMain> mainList, LocalDate startDate) {
         RuleAttendanceMain nowMain = new RuleAttendanceMain();
-        BeanUtils.copyProperties(main,nowMain);
+        BeanUtils.copyProperties(main, nowMain);
         nowMain.setFirstWorkSignInTime(ConstDateUtil.dateAddition(startDate.toString(), ConstDateUtil.formatDate("HH:mm:ss", nowMain.getFirstWorkSignInTime())));
         if (HAVE_SEGMENTATION.equals(ruleScheduling.getSecondFlag())) {
             nowMain.setFirstQuittingSignInTime(ConstDateUtil.dateAddition(startDate.toString(), ConstDateUtil.formatDate("HH:mm:ss", nowMain.getFirstQuittingSignInTime())));
@@ -268,6 +270,219 @@ public class AttendanceServiceImpl implements AttendanceService {
         return peopleList;
     }
+    @Override
+    public void createAttendance(String month) {
+        if (StringUtils.isEmpty(month) || !StringUtils.contains(month, "-")) {
+            return;
+        }
+        // 不能生成以前的排班
+        if (ConstDateUtil.formatDate("yyyy-MM").compareTo(month) > 0){
+            throw new RuntimeException("只能生成未来的排班");
+        }
+        String[] split = month.split("-");
+        List<AttendanceDto> peopleList = ruleNumSettingService.getPeopleListByMonth(month);
+        List<Integer> settingList = peopleList.stream().map(AttendanceDto::getSettingId).distinct().collect(Collectors.toList());
+        // 获取规则集合
+        List<RuleSchedulingDto> ruleList = schedulingService.selectRuleSchedulingListVoBySettingIds(settingList);
+        // 获取班次
+        List<RuleNumDto> ruleNumDtoList = ruleNumSettingService.selectRuleNumBySettingId(settingList);
+        // 处理规则
+        List<LocalDate> dateList = ConstDateUtil.getDateSetFromTheCurrentDayToTheEndOfTheMonth(Integer.parseInt(split[0]), Integer.parseInt(split[1]));
+        List<RuleAttendanceMain> attendanceMains = handleAttendanceMain(peopleList, ruleList, ruleNumDtoList, dateList, month);
+        attendanceMainService.saveBatch(attendanceMains);
+    }
+
+
+    private List<RuleAttendanceMain> handleAttendanceMain(List<AttendanceDto> peopleList, List<RuleSchedulingDto> ruleList, List<RuleNumDto> ruleNumDtoList, List<LocalDate> dateList, String month) {
+        if (CollectionUtil.isEmpty(peopleList)) {
+            return null;
+        }
+        Map<Integer, List<AttendanceDto>> peopleMap = transformMapByPeople(peopleList);
+        Map<Integer, RuleNumDto> ruleNumMap = transformMapByRuleNum(ruleNumDtoList);
+        Map<Integer, List<RuleSchedulingDto>> ruleMap = transformMapByRule(ruleList);
+        List<RuleAttendanceMain> ruleAttendanceMainList = new ArrayList<>();
+        for (Map.Entry<Integer, List<AttendanceDto>> entry : peopleMap.entrySet()) {
+            Integer settingId = entry.getKey();
+            List<AttendanceDto> attendanceDtoList = entry.getValue();
+            RuleNumDto dto = ruleNumMap.get(settingId);
+            List<RuleAttendanceMain> mainList = new ArrayList<>(entry.getValue().size() * dateList.size());
+            for (AttendanceDto attendanceDto : attendanceDtoList) {
+                for (LocalDate date : dateList) {
+                    // 处理固定规则翻班
+                    if (RULE_TYPE_FIXED.equals(dto.getRuleType())) {
+                        handleFixed(date, mainList, ruleMap.get(settingId), attendanceDto, dto.getRuleDictName());
+                    }
+                    // 处理星期翻班
+                    else {
+                        handleWeek(date, mainList, ruleMap.get(settingId), attendanceDto, dto.getRuleDictName());
+                    }
+                }
+            }
+            ruleAttendanceMainList.addAll(mainList);
+        }
+        backCreatePeopleListLog(peopleList, ruleNumMap, dateList);
+        return ruleAttendanceMainList;
+    }
+
+    private void handleFixed(LocalDate nowDate, List<RuleAttendanceMain> mainList, List<RuleSchedulingDto> ruleSchedulingDtoList, AttendanceDto attendanceDto, String ruleDictName) {
+        RuleAttendanceMain main = new RuleAttendanceMain();
+        main.setPosts(attendanceDto.getPosts());
+        main.setRuleDictName(ruleDictName);
+        main.setSchedulingDate(Date.from(nowDate.atStartOfDay(ZoneId.systemDefault()).toInstant()));
+        main.setFleetName(attendanceDto.getFleetName());
+        Date startDate = attendanceDto.getStartDate();
+        // 匹配当天应设置的规则
+        RuleSchedulingDto dto = getNowRuleByFixed(nowDate, startDate, ruleSchedulingDtoList);
+        main.setWorkFlag(dto.getId() == 0 ? FREE_FLAG : WORK_FLAG);
+        BeanUtils.copyProperties(dto, main);
+        main.setName(attendanceDto.getName());
+        main.setJobCode(attendanceDto.getJobCode());
+        // 处理是否分段
+        handleSegmentationByDate(nowDate, main, dto);
+        mainList.add(main);
+    }
+
+    private RuleSchedulingDto getNowRuleByFixed(LocalDate nowDate, Date startDate, List<RuleSchedulingDto> ruleSchedulingDtoList) {
+        LocalDate start = startDate.toInstant().atZone(ZoneId.systemDefault()).toLocalDate();
+        int daysDifference = new Long(ChronoUnit.DAYS.between(start, nowDate)).intValue();
+        if (daysDifference < 0) {
+            return getFreeRuleSchedulingDto(startDate);
+        }
+        int length = ruleSchedulingDtoList.size();
+        return ruleSchedulingDtoList.get(daysDifference % length);
+    }
+
+
+    public static void handleSegmentationByDate(LocalDate date, RuleAttendanceMain main, RuleSchedulingDto dto) {
+        main.setFirstWorkSignInTime(ConstDateUtil.dateAddition(date.toString(), ConstDateUtil.formatDate("HH:mm:ss", main.getFirstWorkSignInTime())));
+        // 不存在分段
+        if (NO_SEGMENTATION.equals(dto.getSecondFlag())) {
+            if (TOMORROW_YES.equals(dto.getFirstSignInDayTomorrow())) {
+                main.setFirstQuittingSignInTime(handleTimeTomorrowByRule(date, main.getFirstQuittingSignInTime()));
+            } else {
+                main.setFirstQuittingSignInTime(ConstDateUtil.dateAddition(date.toString(), ConstDateUtil.formatDate("HH:mm:ss", main.getFirstQuittingSignInTime())));
+            }
+        }
+        // 存在分段
+        else {
+            main.setFirstQuittingSignInTime(ConstDateUtil.dateAddition(date.toString(), ConstDateUtil.formatDate("HH:mm:ss", main.getFirstQuittingSignInTime())));
+            main.setSecondWorkSignInTime(ConstDateUtil.dateAddition(date.toString(), ConstDateUtil.formatDate("HH:mm:ss", main.getSecondWorkSignInTime())));
+            if (TOMORROW_YES.equals(dto.getFirstSignInDayTomorrow())) {
+                main.setSecondQuittingSignInTime(handleTimeTomorrowByRule(date, main.getSecondQuittingSignInTime()));
+            } else {
+                main.setSecondQuittingSignInTime(ConstDateUtil.dateAddition(date.toString(), ConstDateUtil.formatDate("HH:mm:ss", main.getSecondQuittingSignInTime())));
+            }
+        }
+    }
+
+    private static Date handleTimeTomorrowByRule(LocalDate date, Date time) {
+        // 隔天日期需要加1
+        return ConstDateUtil.dateAddition(date.plusDays(1).toString(), ConstDateUtil.formatDate("HH:mm:ss", time));
+    }
+
+
+    private void handleWeek(LocalDate date, List<RuleAttendanceMain> mainList, List<RuleSchedulingDto> ruleSchedulingDtoList, AttendanceDto attendanceDto, String ruleDictName) {
+        RuleAttendanceMain main = new RuleAttendanceMain();
+        main.setSchedulingDate(Date.from(date.atStartOfDay(ZoneId.systemDefault()).toInstant()));
+        main.setPosts(attendanceDto.getPosts());
+        main.setRuleDictName(ruleDictName);
+        main.setFleetName(attendanceDto.getFleetName());
+        RuleSchedulingDto dto = null;
+        dto = getRuleByWeek(main, ruleSchedulingDtoList, date, attendanceDto.getStartDate());
+        main.setWorkFlag(dto.getId() == 0 ? FREE_FLAG : WORK_FLAG);
+        BeanUtils.copyProperties(dto, main);
+        main.setName(attendanceDto.getName());
+        main.setJobCode(attendanceDto.getJobCode());
+        handleSegmentationByDate(date, main, dto);
+        mainList.add(main);
+    }
+
+    private static RuleSchedulingDto getRuleByWeek(RuleAttendanceMain main, List<RuleSchedulingDto> ruleSchedulingDtoList, LocalDate nowDate, Date startDate) {
+        RuleSchedulingDto dto = null;
+        switch (nowDate.getDayOfWeek().toString()) {
+            case "MONDAY":
+                dto = ruleSchedulingDtoList.get(0);
+                break;
+            case "TUESDAY":
+                dto = ruleSchedulingDtoList.get(1);
+                break;
+            case "WEDNESDAY":
+                dto = ruleSchedulingDtoList.get(2);
+                break;
+            case "THURSDAY":
+                dto = ruleSchedulingDtoList.get(3);
+                break;
+            case "FRIDAY":
+                dto = ruleSchedulingDtoList.get(4);
+                break;
+            case "SATURDAY":
+                dto = ruleSchedulingDtoList.get(5);
+                break;
+            case "SUNDAY":
+                dto = ruleSchedulingDtoList.get(6);
+                break;
+        }
+        LocalDate start = startDate.toInstant().atZone(ZoneId.systemDefault()).toLocalDate();
+        int daysDifference = new Long(ChronoUnit.DAYS.between(start, nowDate)).intValue();
+        if (daysDifference < 0) {
+            return getFreeRuleSchedulingDto(startDate);
+        }
+        return dto;
+    }
+
+
+
+    private Map<Integer, List<AttendanceDto>> transformMapByPeople(List<AttendanceDto> peopleList) {
+        Map<Integer, List<AttendanceDto>> map = new HashMap<>(16);
+        for (AttendanceDto dto : peopleList) {
+            List<AttendanceDto> dtoList = map.get(dto.getSettingId());
+            if (CollectionUtil.isEmpty(dtoList)) {
+                map.put(dto.getSettingId(), new ArrayList<>(Arrays.asList(dto)));
+            } else {
+                dtoList.add(dto);
+            }
+        }
+        return map;
+    }
+
+    private Map<Integer, RuleNumDto> transformMapByRuleNum(List<RuleNumDto> ruleNumDtoList) {
+        Map<Integer, RuleNumDto> map = new HashMap<>(16);
+        for (RuleNumDto ruleNumDto : ruleNumDtoList) {
+            map.put(ruleNumDto.getSettingId(), ruleNumDto);
+        }
+        return map;
+    }
+
+    private Map<Integer, List<RuleSchedulingDto>> transformMapByRule(List<RuleSchedulingDto> ruleList) {
+        Map<Integer, List<RuleSchedulingDto>> map = new HashMap<>(16);
+        for (RuleSchedulingDto dto : ruleList) {
+            List<RuleSchedulingDto> dtoList = map.get(dto.getSettingId());
+            if (CollectionUtil.isEmpty(dtoList)) {
+                map.put(dto.getSettingId(), new ArrayList<>(Arrays.asList(dto)));
+            } else {
+                dtoList.add(dto);
+            }
+        }
+        for (Map.Entry<Integer, List<RuleSchedulingDto>> entry : map.entrySet()) {
+            entry.getValue().sort(Comparator.comparing(RuleSchedulingDto::getSort));
+        }
+        return map;
+    }
+
+    private void backCreatePeopleListLog(List<AttendanceDto> peopleList, Map<Integer, RuleNumDto> ruleNumDtoList, List<LocalDate> dateList) {
+        List<RuleAttendanceMainHelp> helpList = new ArrayList<>(peopleList.size());
+        LocalDate localDate = dateList.get(0);
+        Date date = Date.from(localDate.atStartOfDay(ZoneId.systemDefault()).toInstant());
+        for (AttendanceDto dto : peopleList) {
+            RuleAttendanceMainHelp help = new RuleAttendanceMainHelp();
+            BeanUtils.copyProperties(dto, help);
+            help.setDateFlag(date);
+            help.setRuleDictName(ruleNumDtoList.get(dto.getSettingId()).getRuleDictName());
+            helpList.add(help);
+        }
+        helpService.saveBatch(helpList);
+    }
+
     private List<PeopleResponseVo> mainTransformPeopleResponseVo(List<RuleAttendanceMain> mainList) {
         List<PeopleResponseVo> vos = new ArrayList<>(mainList.size());
         for (RuleAttendanceMain main : mainList) {
@@ -89,7 +89,8 @@ public class RuleNumSettingServiceImpl extends ServiceImpl&lt;RuleNumSettingMapper,
             // 获取班次
             List<RuleNumDto> ruleNumDtoList = ruleNumSettingService.selectRuleNumBySettingId(settingList);
             // 处理规则
-            List<RuleAttendanceMain> attendanceMains = handleAttendanceMain(peopleList, ruleList, ruleNumDtoList);
+            List<LocalDate> dateList = ConstDateUtil.getDateSetFromTheCurrentDayToTheEndOfTheMonth();
+            List<RuleAttendanceMain> attendanceMains = handleAttendanceMain(peopleList, ruleList, ruleNumDtoList, dateList);
             attendanceMainService.saveBatch(attendanceMains);
             log.info("排班明细生成完毕");
         } catch (Exception e) {
@@ -111,7 +112,7 @@ public class RuleNumSettingServiceImpl extends ServiceImpl&lt;RuleNumSettingMapper,
         helpService.saveBatch(helpList);
     }
-    private List<RuleAttendanceMain> handleAttendanceMain(List<AttendanceDto> peopleList, List<RuleSchedulingDto> ruleList, List<RuleNumDto> ruleNumDtoList) {
+    private List<RuleAttendanceMain> handleAttendanceMain(List<AttendanceDto> peopleList, List<RuleSchedulingDto> ruleList, List<RuleNumDto> ruleNumDtoList, List<LocalDate> dateList) {
         if (CollectionUtil.isEmpty(peopleList)) {
             return null;
         }
@@ -119,7 +120,6 @@ public class RuleNumSettingServiceImpl extends ServiceImpl&lt;RuleNumSettingMapper,
         Map<Integer, RuleNumDto> ruleNumMap = transformMapByRuleNum(ruleNumDtoList);
         Map<Integer, List<RuleSchedulingDto>> ruleMap = transformMapByRule(ruleList);
         List<RuleAttendanceMain> ruleAttendanceMainList = new ArrayList<>();
-        List<LocalDate> dateList = ConstDateUtil.getDateSetFromTheCurrentDayToTheEndOfTheMonth();
         for (Map.Entry<Integer, List<AttendanceDto>> entry : peopleMap.entrySet()) {
             Integer settingId = entry.getKey();
             List<AttendanceDto> attendanceDtoList = entry.getValue();
@@ -245,6 +245,7 @@ public class RuleNumSettingServiceImpl extends ServiceImpl&lt;RuleNumSettingMapper,
     /**
      * 获取休息rule
+     *
      * @param startDate
      * @return
      */
@@ -333,6 +334,11 @@ public class RuleNumSettingServiceImpl extends ServiceImpl&lt;RuleNumSettingMapper,
         cacheScheduling.putAll(resultMap);
     }
+    @Override
+    public List<AttendanceDto> getPeopleListByMonth(String month) {
+        return baseMapper.getPeopleListByMonth(month);
+    }
+
     private List<DriverScheduling> handleOtherPostsScheduling() {
         QueryWrapper<RuleAttendanceMain> qw = new QueryWrapper<>();
         qw.lambda()
@@ -104,6 +104,29 @@ public class ConstDateUtil {
         return dateList;
     }
+    public static List<LocalDate> getDateSetFromTheCurrentDayToTheEndOfTheMonth(int year,int month){
+        LocalDate today = LocalDate.of(year,month,1);
+
+        // 获取本月的最后一天
+        LocalDate endDay;
+        if (month == 12) {
+            endDay = LocalDate.of(year, month, 31);
+        } else {
+            endDay = LocalDate.of(year, month + 1, 1).minusDays(1);
+        }
+
+        // 生成日期列表
+        List<LocalDate> dateList = new ArrayList<>();
+        LocalDate currentDay = today;
+        while (!currentDay.isAfter(endDay)) {
+            dateList.add(currentDay);
+            currentDay = currentDay.plusDays(1);
+        }
+
+        return dateList;
+    }
+
+
     /**
      * 获取指定月份所有日期
      */
@@ -87,7 +87,29 @@
             FROM
                 rule_attendance_main
             WHERE
-                DATE_FORMAT( scheduling_date, '%y-%m' ) = DATE_FORMAT( NOW(), '%y-%m' )
+                DATE_FORMAT( scheduling_date, '%Y-%m' ) = DATE_FORMAT( NOW(), '%Y-%m' )
+        )
+    </select>
+    <select id="getPeopleListByMonth" resultType="com.ruoyi.pojo.dto.AttendanceDto" resultMap="AttendanceResult">
+        SELECT
+            driver.job_code,
+            driver.personnel_name,
+            driver.fleet_name,
+            driver.posts,
+            rule_setting_driver.setting_id,
+            rule_setting_driver.start_date
+        FROM
+            driver,
+            rule_setting_driver
+        WHERE
+            driver.job_code = rule_setting_driver.job_code
+          AND rule_setting_driver.job_code NOT IN (
+            SELECT DISTINCT
+                job_code
+            FROM
+                rule_attendance_main
+            WHERE
+                DATE_FORMAT( scheduling_date, '%Y-%m' ) = #{month}
         )
     </select>