如果没有节日放假调休的话,工作日很好计算,周一到周五就是工作日,但因为有节日放假调休,使得这个计算需要外部放假安排数据来支持。计算原理: 先按照放假安排数据计算,再按照周一周五计算。
下面以LocalDateTime 为例。
1.第一版,没有使用缓存
/** * 判断是否中国工作日,包含法定节假日调整日期,节假日数据holidayData,如果节假日数据不支持年份,将使用周一到周五为工作日来判断。 * @param localDateTime LocalDateTime * @param holidayData 放假信息0表示放假,1表示工作日,如:2021-01-01:0,2021-02-07:1 * @return boolean */ public static boolean isChineseWorkDay(LocalDateTime localDateTime, String holidayData){ Objects.requireNonNull(holidayData, "holidayData"); Map<String, Integer> dateTypeMap = StringUtil.convertHolidayDataToMap(holidayData); Integer dateType = dateTypeMap.get(DateTimeFormatterUtil.formatToDateStr(localDateTime)); if(dateType != null){ return dateType == 1 ? true : false; } return isWorkDay(localDateTime); } // StringUtil.convertHolidayDataToMap /** * 转换节日数据为map * @param holidayData 节日map * @return 返回节日map */ public static Map<String, Integer> convertHolidayDataToMap(String holidayData){ Map<String, Integer> dateTypeMap = new HashMap<>(); if(isEmpty(holidayData)){ return dateTypeMap; } String[] dateTypeArr = holidayData.replace(" ", "").split(","); for(String dateType : dateTypeArr){ String[] arr = dateType.split(":"); dateTypeMap.put(arr[0], Integer.valueOf(arr[1])); } return dateTypeMap; } /** * 判断是否工作日 (周一到周五) * @param localDateTime LocalDateTime * @return boolean */ public static boolean isWorkDay(LocalDateTime localDateTime){ int dayOfWeek = getDayOfWeek(localDateTime); if(dayOfWeek == 6 || dayOfWeek == 7){ return false; }else{ return true; } }
这个方法,先将放假安排数据解析成Map,然后对比,最后使用周一到周五判断。
2.第二版,使用缓存优化
第一版中,每次调用都先将放假安排数据解析成Map,但其实是不需要的,因为放假安排数据每年只发布一次(特殊情况除外),一年都不需要变化,这些数据第一次调用时放进缓存,后面直接使用,有变化时再更新缓存。
缓存使用本地缓存和Redis缓存都可以,本地缓存速度更快一些,下面使用本地缓存。
public static boolean isChineseWorkDay2(LocalDateTime localDateTime, String holidayData){ Objects.requireNonNull(holidayData, "holidayData"); Map<String, Integer> dateTypeMap = StringUtil.convertHolidayDataToMapUseCache(holidayData); Integer dateType = dateTypeMap.get(DateTimeFormatterUtil.formatToDateStr(localDateTime)); if(dateType != null){ return dateType == 1 ? true : false; } return isWorkDay(localDateTime); } //StringUtil.convertHolidayDataToMapUseCache /** * 转换节日数据为map,使用缓存提高性能 * @param holidayData 节日map * @return 返回节日map */ @SuppressWarnings("unchecked") public static Map<String, Integer> convertHolidayDataToMapUseCache(String holidayData){ Map<String, Integer> dateTypeMap = new HashMap<>(); //参数为空,直接返回 if(isEmpty(holidayData)){ return dateTypeMap; } //查询缓存 dateTypeMap = (Map<String, Integer>)CommonCache.get(holidayData); //缓存存在,返回缓存 if(CollectionUtil.isNotEmpty(dateTypeMap)){ return dateTypeMap; } //缓存不存在,先设置缓存然后返回 Supplier<Object> supplier = new Supplier<Object>() { @Override public Object get() { Map<String, Integer> dateTypeMap = new HashMap<>(); String[] dateTypeArr = holidayData.replace(" ", "").split(","); for(String dateType : dateTypeArr){ String[] arr = dateType.split(":"); dateTypeMap.put(arr[0], Integer.valueOf(arr[1])); } return dateTypeMap; } }; return (Map<String, Integer>)CommonCache.get(holidayData, supplier); } /** * 判断是否工作日 (周一到周五) * @param localDateTime LocalDateTime * @return boolean */ public static boolean isWorkDay(LocalDateTime localDateTime){ int dayOfWeek = getDayOfWeek(localDateTime); if(dayOfWeek == 6 || dayOfWeek == 7){ return false; }else{ return true; } }
缓存使用了WeakHashMap实现缓存自动清理,使用ReentrantReadWriteLock实现读写线程安全。详细代码见com.xkzhangsan.time.utils.CommonCache,核心代码片段如下:
/** * 从缓存池中查找值 * * @param key * 键 * @return 值 */ public V get(K key) { lock.readLock().lock(); try { return cache.get(key); } finally { lock.readLock().unlock(); } } /** * 从缓存池中查找值,没有时尝试生成 * * @param key * 键 * @param supplier 提供者 * @return 值 */ public V get(K key, Supplier<V> supplier) { V value = get(key); if (value == null && supplier != null) { lock.writeLock().lock(); try { value = cache.get(key); // 双重检查,防止在竞争锁的过程中已经有其它线程写入 if (null == value) { try { value = supplier.get(); } catch (Exception e) { throw new RuntimeException(e); } cache.put(key, value); } } finally { lock.writeLock().unlock(); } } return value; }
3. 二种实现性能对比
这里以2021年放假信息为例,分别调用100万次。忽略第一次创建缓存的时间,从第二次开始,测试数据如下:
2021-01-01:0,2021-02-07:1,2021-02-11:0,2021-02-12:0,2021-02-15:0,2021-02-16:0,2021-02-17:0,2021-02-20:1,2021-04-05:0,2021-04-25:1,2021-05-03:0,2021-05-04:0,2021-05-05:0,2021-05-08:1,2021-06-14:0,2021-09-18:1,2021-09-20:0,2021-09-21:0,2021-09-26:1,2021-10-01:0,2021-10-04:0,2021-10-05:0,2021-10-06:0,2021-10-07:0,2021-10-09:1
@Test
public void isChineseWorkDay1(){
//2021年放假信息
String holidayData = "2021-01-01:0,2021-02-07:1,2021-02-11:0,2021-02-12:0,2021-02-15:0,2021-02-16:0,2021-02-17:0,2021-02-20:1,2021-04-05:0,2021-04-25:1,2021-05-03:0,2021-05-04:0,2021-05-05:0,2021-05-08:1,2021-06-14:0,2021-09-18:1,2021-09-20:0,2021-09-21:0,2021-09-26:1,2021-10-01:0,2021-10-04:0,2021-10-05:0,2021-10-06:0,2021-10-07:0,2021-10-09:1";
//指定日期是否是工作日
long s = 0;
for (int i = 0; i < 1000001; i++) {
if(i==1){
s = System.currentTimeMillis();
}
DateTimeCalculatorUtil.isChineseWorkDay(LocalDateTime.now(), holidayData);
}
System.out.println("isChineseWorkDay1 cost1:"+(System.currentTimeMillis()-s));
}
@Test
public void isChineseWorkDay2(){
//2021年放假信息
String holidayData = "2021-01-01:0,2021-02-07:1,2021-02-11:0,2021-02-12:0,2021-02-15:0,2021-02-16:0,2021-02-17:0,2021-02-20:1,2021-04-05:0,2021-04-25:1,2021-05-03:0,2021-05-04:0,2021-05-05:0,2021-05-08:1,2021-06-14:0,2021-09-18:1,2021-09-20:0,2021-09-21:0,2021-09-26:1,2021-10-01:0,2021-10-04:0,2021-10-05:0,2021-10-06:0,2021-10-07:0,2021-10-09:1";
//指定日期是否是工作日
long s = 0;
for (int i = 0; i < 1000001; i++) {
if(i==1){
s = System.currentTimeMillis();
}
DateTimeCalculatorUtil.isChineseWorkDay2(LocalDateTime.now(), holidayData);
}
System.out.println("isChineseWorkDay2 cost2:"+(System.currentTimeMillis()-s));
}
结果(单位:ms):
isChineseWorkDay1 cost1:5589
isChineseWorkDay2 cost2:366
可以看到,使用缓存后性能对比 5589/366=15.27 , 速度提高15倍多,代码性能的小优化,大量调用后会被累加放大,优化非常值得!
源代码地址:https://github.com/xkzhangsan/xk-time