123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663 |
- package com.persagy.calendar.handle;
- import java.util.ArrayList;
- import java.util.Date;
- import java.util.HashMap;
- import java.util.Iterator;
- import java.util.List;
- import java.util.Map;
- import java.util.Set;
- import java.util.stream.Collectors;
- import org.springframework.beans.BeanUtils;
- import org.springframework.beans.factory.annotation.Autowired;
- import org.springframework.stereotype.Component;
- import org.springframework.transaction.annotation.Transactional;
- import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
- import com.google.common.collect.Lists;
- import com.persagy.calendar.constant.WorkCalendarConstant;
- import com.persagy.calendar.pojo.dto.WorkCalendar;
- import com.persagy.calendar.pojo.dto.WorkCalendarDate;
- import com.persagy.calendar.pojo.dto.WorkCalendarRule;
- import com.persagy.calendar.pojo.vo.CalendarDateResult;
- import com.persagy.calendar.pojo.vo.CustomCalendarVO;
- import com.persagy.calendar.pojo.vo.WorkCalendarDateCreateVO;
- import com.persagy.calendar.pojo.vo.WorkCalendarDateQueryVO;
- import com.persagy.calendar.pojo.vo.WorkCalendarDateUpdateVO;
- import com.persagy.calendar.pojo.vo.WorkCalendarMoreDateCreateVO;
- import com.persagy.calendar.service.IWorkCalendarDateService;
- import com.persagy.calendar.service.IWorkCalendarRuleService;
- import com.persagy.calendar.service.IWorkCalendarService;
- import com.persagy.common.exception.BusinessException;
- import com.persagy.common.utils.DateUtil;
- import com.persagy.common.utils.IdGenerator;
- import com.persagy.common.utils.StringUtil;
- import cn.hutool.core.collection.CollectionUtil;
- import cn.hutool.core.date.DateField;
- import cn.hutool.core.date.DateTime;
- import cn.hutool.core.date.DateUnit;
- import cn.hutool.core.util.BooleanUtil;
- import lombok.extern.slf4j.Slf4j;
- @Slf4j
- @Component
- public class WorkCalendarDateHandler {
- @Autowired
- private IWorkCalendarService workCalendarService;
-
- @Autowired
- private IWorkCalendarRuleService workCalendarRuleService;
-
- @Autowired
- private IWorkCalendarDateService workCalendarDateService;
-
-
- public List<CalendarDateResult> queryWorkCalendarDateList(WorkCalendarDateQueryVO queryVO) {
-
- Map<String, List<WorkCalendarDate>> resultMap = new HashMap<String, List<WorkCalendarDate>>();
-
- Map<String, Boolean> existsCustom = new HashMap<String, Boolean>();
-
-
- this.selectHistoryDate(resultMap, existsCustom, queryVO);
-
-
- String format = DateUtil.format(new Date(), DateUtil.FORMAT_DATE_YYYYMMDD);
- DateTime currentDate = DateUtil.parse(format, DateUtil.FORMAT_DATE_YYYYMMDD);
- String dateStartStr = queryVO.getWorkDateStart();
- String dateEndStr = queryVO.getWorkDateEnd();
- DateTime dateStart = DateUtil.parse(dateStartStr, DateUtil.FORMAT_DATE_YYYYMMDD);
- DateTime dateEnd = DateUtil.parse(dateEndStr, DateUtil.FORMAT_DATE_YYYYMMDD);
- if (currentDate.before(dateStart)) {
- currentDate = dateStart;
- } else {
-
- if (resultMap.containsKey(currentDate.toString(DateUtil.FORMAT_DATE_YYYYMMDD))) {
- currentDate.offset(DateField.DAY_OF_YEAR, 1);
- }
- }
-
-
- if (currentDate.before(dateEnd) || currentDate.equals(dateEnd)) {
- String offsetStr = currentDate.toString(DateUtil.FORMAT_DATE_YYYYMMDD);
- List<WorkCalendarRule> ruleList = this.selectWorkCalendarRuleList(offsetStr, dateEndStr, queryVO);
- if (CollectionUtil.isNotEmpty(ruleList)) {
-
- long betweenDay = DateUtil.between(currentDate, dateEnd, DateUnit.DAY) + 1;
- this.paserWorkCalendarRule(betweenDay, currentDate, ruleList, resultMap, existsCustom);
- }
- }
-
-
- dateStart = DateUtil.parse(dateStartStr, DateUtil.FORMAT_DATE_YYYYMMDD);
- long betweenDay = DateUtil.between(dateStart, dateEnd, DateUnit.DAY) + 1;
-
- return this.setResultList(queryVO.getCalendarId(), betweenDay, dateStart, resultMap);
- }
-
-
- private List<CalendarDateResult> setResultList(String calendarId, long betweenDay, DateTime dateStart,
- Map<String, List<WorkCalendarDate>> resultMap) {
- List<CalendarDateResult> resultList = new ArrayList<CalendarDateResult>();
-
- String nextDay = null;
- for (int i = 0; i < betweenDay; i++) {
- nextDay = dateStart.toString(DateUtil.FORMAT_DATE_YYYYMMDD);
- CalendarDateResult result = new CalendarDateResult();
- int dayOfWeek = DateUtil.dayOfWeek(dateStart);
- if (dayOfWeek == 1 || dayOfWeek == 7) {
- result.setWorkday(false);
- }
- result.setWorkDate(nextDay);
- result.setCalendarId(calendarId);
- result.setDate(DateUtil.dayOfMonth(dateStart));
- result.setWeek(WorkCalendarConstant.DAY_OF_WEEK.get(dayOfWeek));
- List<WorkCalendarDate> list = resultMap.get(nextDay);
- if (CollectionUtil.isNotEmpty(list)) {
- for (int j = list.size() - 1; j >= 0; j--) {
- WorkCalendarDate calendarDate = list.get(j);
-
- if (result.isWorkday() && "OpenTime-2".equals(calendarDate.getDictCode())) {
- list.remove(j);
- } else if (!result.isWorkday() && "OpenTime-1".equals(calendarDate.getDictCode())) {
- list.remove(j);
- }
- }
- }
-
- result.setTimeTable(CollectionUtil.isEmpty(list) ? Lists.newArrayList() : list);
- resultList.add(result);
- dateStart.offset(DateField.DAY_OF_YEAR, 1);
- }
- resultMap.clear();
- return resultList;
- }
-
-
- public void selectHistoryDate(Map<String, List<WorkCalendarDate>> resultMap, Map<String, Boolean> existsCustom, WorkCalendarDateQueryVO queryVO) {
- List<WorkCalendarDate> dates = this.selectWorkCalendarDateList(queryVO);
- for (WorkCalendarDate calendarDate : dates) {
-
- String existsKey = calendarDate.getCalendarId() + calendarDate.getProjectId() + calendarDate.getWorkDate() + calendarDate.getDictType() + (StringUtil.isBlank(calendarDate.getDictCode()) ? "" : calendarDate.getDictCode());
- boolean exists = WorkCalendarConstant.CUSTOM_CALENDAR_DATE_YES.equals(calendarDate.getCustomFlag());
- if (!BooleanUtil.isTrue(existsCustom.get(existsKey))) {
- existsCustom.put(existsKey, exists);
- }
-
- if (resultMap != null) {
- if (resultMap.containsKey(calendarDate.getWorkDate())) {
- List<WorkCalendarDate> list = resultMap.get(calendarDate.getWorkDate());
- list.add(calendarDate);
- } else {
- List<WorkCalendarDate> list = new ArrayList<WorkCalendarDate>();
- list.add(calendarDate);
- resultMap.put(calendarDate.getWorkDate(), list);
- }
- }
- }
- }
-
-
- public List<WorkCalendarDate> selectWorkCalendarDateList(WorkCalendarDateQueryVO workCalendarDateQueryVO) {
- QueryWrapper<WorkCalendarDate> queryWrapper = new WorkCalendarDate.Builder().createQueryWrapper()
- .projectIdEq(workCalendarDateQueryVO.getProjectId())
- .idEq(workCalendarDateQueryVO.getId())
- .ruleIdEq(workCalendarDateQueryVO.getRuleId())
- .calendarIdEq(workCalendarDateQueryVO.getCalendarId())
- .workDateGe(workCalendarDateQueryVO.getWorkDateStart())
- .workDateLe(workCalendarDateQueryVO.getWorkDateEnd())
- .dictTypeEq(workCalendarDateQueryVO.getDictType())
- .dictTypeIn(workCalendarDateQueryVO.getDictTypes())
- .dictCodeIn(workCalendarDateQueryVO.getDictCode())
- .calendarNameLike(workCalendarDateQueryVO.getCalendarName())
- .builderQueryWrapper();
- queryWrapper.select("ID, RULE_ID, CALENDAR_ID, PROJECT_ID, CALENDAR_NAME, CALENDAR_DESC, DICT_TYPE, "
- + "DICT_CODE, WORK_DATE, WORK_TIME, CUSTOM_FLAG");
- List<WorkCalendarDate> list = workCalendarDateService.list(queryWrapper);
- return list == null ? new ArrayList<WorkCalendarDate>() : list;
- }
-
-
- public void paserWorkCalendarRule(long betweenDay, DateTime currentDate, List<WorkCalendarRule> ruleList,
- Map<String, List<WorkCalendarDate>> resultMap, Map<String, Boolean> existsCustom) {
- Map<String, WorkCalendar> calendarMap = new HashMap<String, WorkCalendar>();
- for(int i = 0; i < betweenDay; i++) {
-
- for (WorkCalendarRule rule : ruleList) {
-
- this.parseWorkCalendarRule(rule, currentDate, calendarMap, resultMap, existsCustom);
- }
-
- currentDate.offset(DateField.DAY_OF_YEAR, 1);
- }
- calendarMap.clear();
- }
-
-
- private void parseWorkCalendarRule(WorkCalendarRule rule, DateTime currentDate, Map<String, WorkCalendar> calendarMap,
- Map<String, List<WorkCalendarDate>> resultMap, Map<String, Boolean> existsCustom) {
- String workDate = currentDate.toString(DateUtil.FORMAT_DATE_YYYYMMDD);
- String dictCode = StringUtil.isBlank(rule.getDictCode()) ? "" : rule.getDictCode();
- int indexOf = dictCode.indexOf(WorkCalendarConstant.WORK_CALENDAR_D_OF_W);
- if (indexOf <= 0) {
- indexOf = dictCode.indexOf(WorkCalendarConstant.WORK_CALENDAR_D_OF_M);
- if (indexOf <= 0) {
- indexOf = dictCode.indexOf(WorkCalendarConstant.WORK_CALENDAR_M_AND_D);
- }
- }
- if (indexOf > 0) {
- dictCode = dictCode.substring(0, indexOf - 1);
- }
- String existsKey = rule.getCalendarId() + rule.getProjectId() + workDate + rule.getDictType() + (StringUtil.isBlank(dictCode) ? "" : dictCode);
-
- if (existsCustom != null && BooleanUtil.isTrue(existsCustom.get(existsKey))) {
- return;
- }
-
-
-
- int work = Integer.valueOf(workDate);
- String ruleStart = rule.getRuleDateStart();
- String ruleEnd = rule.getRuleDateEnd();
- if (StringUtil.isNotBlank(ruleStart) && StringUtil.isNotBlank(ruleEnd)) {
- int start = Integer.valueOf(ruleStart);
- int end = Integer.valueOf(ruleEnd);
- if (work >= start && work <= end) {
-
- if (indexOf > 0) {
-
- boolean result = this.parseRuleValue(rule, currentDate, workDate);
- if (!result) {
- return;
- }
- }
-
-
- WorkCalendarDate calendarDate = this.comWorkCalendarOrRule(rule, workDate, dictCode, calendarMap);
- if (calendarDate == null) {
- return;
- }
-
- this.addCalendarDateResult(workDate, calendarDate, resultMap);
- }
- }
- }
-
-
- private boolean parseRuleValue(WorkCalendarRule rule, DateTime currentDate, String workDate) {
- if (CollectionUtil.isEmpty(rule.getValue())) {
- return true;
- }
-
- String mmddStr = workDate.substring(4, 8);
- Integer mmdd = Integer.valueOf(mmddStr);
-
- Set<String> values = rule.getValue();
- String dictCode = rule.getDictCode();
- String[] split = dictCode.split("-");
- dictCode = split[0];
- String type = split[1];
- switch (type) {
- case WorkCalendarConstant.WORK_CALENDAR_D_OF_W:
-
- int dayOfWeek = DateUtil.dayOfWeek(currentDate);
- dayOfWeek = dayOfWeek - 1;
- if (dayOfWeek == 0) {
- dayOfWeek = 7;
- }
- String num = String.valueOf(dayOfWeek);
- if (values.contains(num)) {
- return true;
- }
- break;
- case WorkCalendarConstant.WORK_CALENDAR_D_OF_M:
-
- int ofMonth = DateUtil.dayOfMonth(currentDate);
- Iterator<String> iterator = values.iterator();
- while(iterator.hasNext()) {
- String next = iterator.next();
- if (next.contains("-")) {
- String[] split2 = next.split("-");
- int start = Integer.parseInt(split2[0]);
- int end = Integer.parseInt(split2[1]);
- if (ofMonth >= start && ofMonth <= end) {
- return true;
- }
- } else if (next.equals(String.valueOf(ofMonth))) {
- return true;
- }
- }
- break;
- case WorkCalendarConstant.WORK_CALENDAR_M_AND_D:
- iterator = values.iterator();
- while(iterator.hasNext()) {
- String next = iterator.next();
- if (next.contains("-")) {
- String[] split2 = next.split("-");
- Integer startInt = Integer.valueOf(split2[0]);
- Integer endInt = Integer.valueOf(split2[1]);
- if (mmdd >= startInt && mmdd <= endInt) {
- return true;
- }
- } else if (mmddStr.equals(next)) {
- return true;
- }
- }
- break;
- default:
- break;
- }
- return false;
- }
-
- public void addCalendarDateResult(String workDate, WorkCalendarDate calendarDate, Map<String, List<WorkCalendarDate>> resultMap) {
- if (resultMap == null) {
- boolean result = this.workCalendarDateService.save(calendarDate);
-
- if (!result) {
- throw new BusinessException("新增作息时间信息失败,规则ID: " + calendarDate.getRuleId());
- }
- return;
- }
-
- if (!resultMap.containsKey(workDate)) {
- List<WorkCalendarDate> list = new ArrayList<WorkCalendarDate>();
- list.add(calendarDate);
- resultMap.put(workDate, list);
- return;
- }
-
- List<WorkCalendarDate> list = resultMap.get(workDate);
- list.add(calendarDate);
- }
-
-
- public WorkCalendarDate comWorkCalendarOrRule(WorkCalendarRule rule, String workDate, String dictCode,
- Map<String, WorkCalendar> calendarMap) {
-
- WorkCalendar workCalendar = null;
- if (calendarMap.containsKey(rule.getCalendarId())) {
- workCalendar = calendarMap.get(rule.getCalendarId());
- } else {
- workCalendar = this.workCalendarService.getById(rule.getCalendarId(), rule.getProjectId());
- calendarMap.put(rule.getCalendarId(), workCalendar);
- }
- if (workCalendar == null) {
- log.warn("规则ID为[{}]的规则,所对应的工作历信息不存在", rule.getId());
- return null;
- }
-
- WorkCalendarDate workCalendarDate = new WorkCalendarDate();
- workCalendarDate.setGroupCode(rule.getGroupCode());
- workCalendarDate.setCalendarId(workCalendar.getId());
- workCalendarDate.setRuleId(rule.getId());
- workCalendarDate.setCalendarName(workCalendar.getCalendarName());
- workCalendarDate.setCalendarDesc(workCalendar.getCalendarDesc());
- workCalendarDate.setProjectId(rule.getProjectId());
- workCalendarDate.setDictType(rule.getDictType());
- workCalendarDate.setDictCode(dictCode);
- workCalendarDate.setWorkDateStart(rule.getRuleDateStart());
- workCalendarDate.setWorkDateEnd(rule.getRuleDateEnd());
- workCalendarDate.setWorkDate(workDate);
-
- workCalendarDate.setWorkTime(StringUtil.isBlank(rule.getRuleTime()) ?
- (CollectionUtil.isEmpty(rule.getValue()) ? null : rule.getValue().stream().collect(Collectors.joining(",")))
- : rule.getRuleTime());
-
- workCalendarDate.setCustomFlag(WorkCalendarConstant.CUSTOM_CALENDAR_DATE_NO);
- workCalendarDate.setUpdateUser(rule.getUpdateUser());
- return workCalendarDate;
- }
-
-
- public List<WorkCalendarRule> selectWorkCalendarRuleList(String workDatestart, String workDateEnd, WorkCalendarDateQueryVO queryVO) {
- QueryWrapper<WorkCalendarRule> queryWrapper = new WorkCalendarRule.Builder().createQueryWrapper()
- .projectIdEq(queryVO.getProjectId()).idEq(queryVO.getRuleId())
- .calendarIdEq(queryVO.getCalendarId()).ruleDateBetween(workDatestart, workDateEnd)
- .dictTypeEq(queryVO.getDictType()).dictCodeIn(queryVO.getDictCode())
- .dictTypeIn(queryVO.getDictTypes())
- .delFlagEq(WorkCalendarConstant.WORK_CALENDAR_DEL_FLAG_NOR)
- .ruleFlagEq(WorkCalendarConstant.WORK_CALENDAR_RULE_FLAG_NEW).builderQueryWrapper();
- return this.workCalendarRuleService.list(queryWrapper);
- }
-
-
- @Transactional
- public boolean handleParseCalendarRule(WorkCalendarDateQueryVO queryVO) {
-
- QueryWrapper<WorkCalendarDate> queryWrapper = new WorkCalendarDate.Builder().createQueryWrapper()
- .projectIdEq(queryVO.getProjectId()).calendarIdEq(queryVO.getCalendarId())
- .calendarNameEq(queryVO.getCalendarName()).dictTypeEq(queryVO.getDictType())
- .dictCodeIn(queryVO.getDictCode()).workDateGe(queryVO.getWorkDateStart())
- .workDateLe(queryVO.getWorkDateEnd()).customFlagEq(WorkCalendarConstant.CUSTOM_CALENDAR_DATE_NO)
- .builderQueryWrapper();
- this.workCalendarDateService.remove(queryWrapper);
-
-
- List<WorkCalendarRule> ruleList = this.selectWorkCalendarRuleList(queryVO.getWorkDateStart(), queryVO.getWorkDateEnd(), queryVO);
- if (CollectionUtil.isNotEmpty(ruleList)) {
-
-
- Map<String, Boolean> existsCustom = new HashMap<String, Boolean>();
-
- this.selectHistoryDate(null, existsCustom, queryVO);
-
-
- DateTime dateStart = DateUtil.parse(queryVO.getWorkDateStart(), DateUtil.FORMAT_DATE_YYYYMMDD);
- DateTime dateEnd = DateUtil.parse(queryVO.getWorkDateEnd(), DateUtil.FORMAT_DATE_YYYYMMDD);
- long between = DateUtil.between(dateStart, dateEnd, DateUnit.DAY) + 1;
- this.paserWorkCalendarRule(between, dateStart, ruleList, null, existsCustom);
- }
- return true;
- }
-
-
-
- public void batchCreateCalendarDate(CustomCalendarVO<WorkCalendarDateCreateVO> createVO) {
- boolean result = false;
- List<WorkCalendarDateCreateVO> batchInfos = createVO.getBatchInfo();
-
- DateTime now = DateUtil.date();
-
- String format = DateUtil.format(now, DateUtil.FORMAT_DATE_YYYYMMDD);
- Integer current = Integer.valueOf(format);
- for (WorkCalendarDateCreateVO batchInfo : batchInfos) {
- Integer workDate = Integer.valueOf(batchInfo.getWorkDate());
- if (workDate < current) {
- throw new BusinessException("作息日期不合法,必须大于等于当日");
- }
-
- WorkCalendar calendar = this.workCalendarService.getById(batchInfo.getCalendarId(), createVO.getProjectId());
- if (calendar == null) {
- throw new BusinessException("工作历类型不存在");
- }
-
- WorkCalendarDate calendarDate = new WorkCalendarDate();
- BeanUtils.copyProperties(batchInfo, calendarDate);
-
- result = this.workCalendarDateService.checkWorkDateValid(calendarDate);
- if (!result) {
- throw new BusinessException("当天已存在相同类型额定义,不可重复添加");
- }
- calendarDate.setGroupCode(createVO.getGroupCode());
- calendarDate.setProjectId(createVO.getProjectId());
- calendarDate.setUpdateUser(createVO.getUserId());
- calendarDate.setCalendarName(calendar.getCalendarName());
- calendarDate.setCalendarDesc(calendar.getCalendarDesc());
- calendarDate.setCustomFlag(WorkCalendarConstant.CUSTOM_CALENDAR_DATE_YES);
- result = this.workCalendarDateService.save(calendarDate);
- if (!result) {
- throw new BusinessException("自定义工作历作息时间添加失败");
- }
- }
- }
-
-
- public <T> void batchUpdateCalendarDate(CustomCalendarVO<WorkCalendarDateUpdateVO> updateVO) {
- boolean result = false;
- List<WorkCalendarDateUpdateVO> batchInfos = updateVO.getBatchInfo();
-
- DateTime now = DateUtil.date();
-
- String format = DateUtil.format(now, DateUtil.FORMAT_DATE_YYYYMMDD);
- Integer current = Integer.valueOf(format);
- for (WorkCalendarDateUpdateVO batchInfo : batchInfos) {
- Integer workDate = Integer.valueOf(batchInfo.getWorkDate());
- if (workDate < current) {
- throw new BusinessException("作息日期不合法,必须大于等于当日");
- }
-
- WorkCalendar calendar = this.workCalendarService.getById(batchInfo.getCalendarId(), updateVO.getProjectId());
- if (calendar == null) {
- throw new BusinessException("工作历类型不存在");
- }
-
- WorkCalendarDate calendarDate = new WorkCalendarDate();
- BeanUtils.copyProperties(batchInfo, calendarDate);
-
- result = this.workCalendarDateService.checkWorkDateValid(calendarDate);
- if (!result) {
- throw new BusinessException("当天已存在相同类型的定义,不可重复添加");
- }
- calendarDate.setGroupCode(updateVO.getGroupCode());
- calendarDate.setProjectId(updateVO.getProjectId());
- calendarDate.setUpdateUser(updateVO.getUserId());
- calendarDate.setCalendarName(calendar.getCalendarName());
- calendarDate.setCalendarDesc(calendar.getCalendarDesc());
- calendarDate.setCustomFlag(WorkCalendarConstant.CUSTOM_CALENDAR_DATE_YES);
- result = this.workCalendarDateService.updateById(calendarDate);
- if (!result) {
- throw new BusinessException("自定义工作历作息时间添加失败");
- }
- }
- }
-
- @Transactional
- public void batchCreateCalendarMoreDate(WorkCalendarMoreDateCreateVO createVO, WorkCalendar calendar) {
- String calendarId = createVO.getCalendarId();
- String projectId = createVO.getProjectId();
- String now = DateUtil.format(new Date(), DateUtil.FORMAT_DATE_YYYYMMDDHHMMSS);
- DateTime dateStart = DateUtil.parse(createVO.getWorkDateStart(), DateUtil.FORMAT_DATE_YYYYMMDD);
- DateTime dateEnd = DateUtil.parse(createVO.getWorkDateEnd(), DateUtil.FORMAT_DATE_YYYYMMDD);
- long betweenDay = DateUtil.between(dateStart, dateEnd, DateUnit.DAY) + 1;
-
- Set<String> timetables = createVO.getTimetable();
- List<WorkCalendarDate> datas = new ArrayList<WorkCalendarDate>();
- for (int i = 0; i < betweenDay; i++) {
- String currentDay = DateUtil.format(dateStart, DateUtil.FORMAT_DATE_YYYYMMDD);
- if (CollectionUtil.isEmpty(timetables)) {
- WorkCalendarDate calendarDate = this.setCalendarDate(createVO, calendar, currentDay, null, now);
- datas.add(calendarDate);
- } else {
- for (String timetable : timetables) {
-
- WorkCalendarDate calendarDate = this.setCalendarDate(createVO, calendar, currentDay, timetable, now);
- datas.add(calendarDate);
- }
- }
-
-
- dateStart.offset(DateField.DAY_OF_YEAR, 1);
- }
-
- QueryWrapper<WorkCalendarDate> queryWrapper = new WorkCalendarDate.Builder().createQueryWrapper()
- .calendarIdEq(calendarId).projectIdEq(projectId).dictTypeEq(createVO.getDictType())
- .workDateGe(createVO.getWorkDateStart()).workDateLe(createVO.getWorkDateEnd())
- .customFlagEq(createVO.getCustomFlag()).builderQueryWrapper();
- this.workCalendarDateService.remove(queryWrapper);
-
-
- this.workCalendarDateService.batchCreateCalendarDate(datas);
- }
-
-
- private WorkCalendarDate setCalendarDate(WorkCalendarMoreDateCreateVO createVO, WorkCalendar calendar,
- String currentDay, String timetable, String now) {
- WorkCalendarDate calendarDate = new WorkCalendarDate();
- calendarDate.setId(IdGenerator.getSnowIdStr(WorkCalendarConstant.WORK_CALENDAR_DATE_ID_PREFIX));
- calendarDate.setGroupCode(createVO.getGroupCode());
- calendarDate.setProjectId(createVO.getProjectId());
- calendarDate.setCalendarId(createVO.getCalendarId());
- calendarDate.setCustomFlag(WorkCalendarConstant.CUSTOM_CALENDAR_DATE_YES);
- calendarDate.setCalendarName(calendar.getCalendarName());
- calendarDate.setCalendarDesc(calendar.getCalendarDesc());
- calendarDate.setDictType(createVO.getDictType());
- calendarDate.setDictCode(createVO.getDictCode());
- calendarDate.setWorkDate(currentDay);
- calendarDate.setWorkTime(timetable);
- calendarDate.setCreateTime(now);
- calendarDate.setUpdateTime(now);
- calendarDate.setUpdateUser(createVO.getUserId());
-
- return calendarDate;
- }
-
-
- public static boolean isMoreThanToday(String workDate) {
- DateTime now = DateUtil.date();
-
- String format = DateUtil.format(now, DateUtil.FORMAT_DATE_YYYYMMDD);
- Integer current = Integer.valueOf(format);
- Integer dateInt = Integer.valueOf(workDate);
- if (dateInt < current) {
- return false;
- }
- return true;
- }
-
- }
|