package common.utils;
import lombok.extern.slf4j.Slf4j;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.Map;
/**
* 工作日计算类
*
* @author duxuefu
* @since 2022/7/5 13:16
*/
@Slf4j
public class HolidayUtils {
static SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd");
/**
* 增加排除法定假日的工作日
*
* @param start 开始日期
* @param days 增加的工作日
* @param holidays 法定节假日，计算值;0:公休日占用周六日,-1:公休日占用工作日,1:工作日占用周六日
* @return 增加指定工作日后的日期
*/
public static Calendar addWorkdays(Calendar start, int days, Map<String, Integer> holidays) {
Calendar end = addCommonWorkdays(start, days);
int specialDays = getWorkDays(start.getTime(), end.getTime(), holidays);
addSpecialDays(end, specialDays, holidays);
return end;
}
/**
* 增加不排除法定假日的工作日
*
* @param date 开始日期
* @param days 增加的工作日，正整数，负整数没有测试过
* @return 增加指定工作日后的日期
*/
public static Calendar addCommonWorkdays(Calendar date, int days) {
Calendar result = Calendar.getInstance();
result.setTime(date.getTime());
int integralWeeks = days / 5;
int remainder = days % 5;
// 取整的工作日折算为自然日
int iDays = integralWeeks * 7;
// 计算取余的工作日是否落在本周工作日内，如果落在下一个工作日，则增加两个自然日（周六和周日）
int day_of_week = result.get(Calendar.DAY_OF_WEEK);
int rDays = (day_of_week + 5) % 7 + remainder;
if (rDays > 4) {
remainder += 2;
}
result.add(Calendar.DAY_OF_YEAR, iDays + remainder);
return result;
}
/**
* 计算特殊日期
*
* @param date 开始日期
* @param specialDays 特殊日期，被节假日占用的日期
* @param holidays 法定节假日，计算值;0:公休日占用周六日,-1:公休日占用工作日,1:工作日占用周六日
*/
public static void addSpecialDays(Calendar date, int specialDays, Map<String, Integer> holidays) {
// 根据不同工作日天数差异计算实际工作日
if (specialDays == 0) {
// 工作日差异为0的情况，向后找日期，找到是工作日的第一个日期
while (true) {
int dayOfWeek = date.get(Calendar.DAY_OF_WEEK);
int holiday = holidays.get(simpleDateFormat.format(date.getTime())) == null ? 0 : holidays.get(simpleDateFormat.format(date.getTime()));
// 如果当前日期是工作日占用周六日或者是正常的工作日，则为当前日期
if (holidays.get(simpleDateFormat.format(date.getTime())) != null) {
if (holidays.get(simpleDateFormat.format(date.getTime())) == 1) {
break;
} else {
date.add(Calendar.DAY_OF_YEAR, -1);
}
} else {
if (dayOfWeek != 1 && dayOfWeek != 7) {
break;
} else {
date.add(Calendar.DAY_OF_YEAR, -1);
}
}
}
} else {
// 工作日差异不为0的情况， 工作日差异为负整数的情况需要向回找日期，所以step是-1，反之是1
int step = 1;
if (specialDays > 0) {
step = -1;
}
do {
int dayOfWeek = date.get(Calendar.DAY_OF_WEEK);
// 如果当前日期是工作日占用周六日或者是正常的工作日，则增加或者减少受节假日影响的日期
if (holidays.get(simpleDateFormat.format(date.getTime())) != null) {
if (holidays.get(simpleDateFormat.format(date.getTime())) == 1) {
specialDays += step;
}
} else if (dayOfWeek != 1 && dayOfWeek != 7) {
specialDays += step;
}
// 工作日差异为正整数和负整数的结束计算方式不一样
if (step == -1) {
date.add(Calendar.DAY_OF_YEAR, step);
} else {
if (specialDays != 0) {
date.add(Calendar.DAY_OF_YEAR, step);
}
}
} while (specialDays != 0);
}
}
/**
* 合并计算区间内的法定假日影响的工作日天数
*
* @param start 开始日期
* @param end 结束日期
* @param holidays 节假日集合，计算值;0:公休日占用周六日,-1:公休日占用工作日,1:工作日占用周六日
* @return 影响的工作日天数，正整数为增加了工作日，负整数为
*/
public static int getWorkDays(Date start, Date end, Map<String, Integer> holidays) {
int days = 0;
for (String key : holidays.keySet()) {
try {
Date current = simpleDateFormat.parse(key);
if (current.compareTo(start) > -1 && current.compareTo(end) < 1) {
days += holidays.get(key);
}
} catch (ParseException e) {
log.warn(e.getMessage(), e);
}
}
return days;
}
}
测试类
package common.utils;
import lombok.extern.slf4j.Slf4j;
import org.junit.Test;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
/**
* 测试类：工作日计算类
*
* @author duxuefu
* @since 2022/6/29 16:35
*/
@Slf4j
public class HolidayUtilsTest {
static SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd");
@Test
public void test() {
Calendar c = Calendar.getInstance();
int days = 120;
c.setTime(new Date());
c.set(Calendar.HOUR, 0);
c.set(Calendar.MINUTE, 0);
c.set(Calendar.SECOND, 0);
c.set(Calendar.MONTH, 0);
c.set(Calendar.DAY_OF_MONTH, 7);
Date start = c.getTime();
log.info("开始日期 {}年{}月{}日", c.get(Calendar.YEAR), c.get(Calendar.MONTH) + 1, c.get(Calendar.DAY_OF_MONTH));
Calendar commonWorkdays = HolidayUtils.addCommonWorkdays(c, days);
Calendar workdays = HolidayUtils.addWorkdays(c, days, HolidayUtilsTest.holidays);
log.info("不包含法定假日：{}年{}月{}日", commonWorkdays.get(Calendar.YEAR), commonWorkdays.get(Calendar.MONTH) + 1, commonWorkdays.get(Calendar.DAY_OF_MONTH));
log.info("包含法定假日: {}年{}月{}日", workdays.get(Calendar.YEAR), workdays.get(Calendar.MONTH) + 1, workdays.get(Calendar.DAY_OF_MONTH));
}
// key 是日期， value是计算值; 0:公休日占用周六日,-1:公休日占用工作日,1:工作日占用周六日
public static Map<String, Integer> holidays = new HashMap(32) {{
put("2022-01-01", 0);
put("2022-01-02", 0);
put("2022-01-03", -1);
put("2022-01-29", 1);
put("2022-01-30", 1);
put("2022-01-31", -1);
put("2022-02-01", -1);
put("2022-02-02", -1);
put("2022-02-03", -1);
put("2022-02-04", -1);
put("2022-02-05", 0);
put("2022-02-06", 0);
put("2022-04-02", 1);
put("2022-04-03", 0);
put("2022-04-04", -1);
put("2022-04-05", -1);
put("2022-04-24", 1);
put("2022-04-30", 0);
put("2022-05-01", 0);
put("2022-05-02", -1);
put("2022-05-03", -1);
put("2022-05-04", -1);
put("2022-05-07", 1);
put("2022-06-03", -1);
put("2022-06-04", 0);
put("2022-06-05", 0);
put("2022-09-10", 0);
put("2022-09-11", 0);
put("2022-09-12", -1);
put("2022-10-01", 0);
put("2022-10-02", 0);
put("2022-10-03", -1);
put("2022-10-04", -1);
put("2022-10-05", -1);
put("2022-10-06", -1);
put("2022-10-07", -1);
put("2022-10-08", 1);
put("2022-10-09", 1);
}};
}
