计算工作日期的java工具类

 

 

import org.apache.commons.collections.CollectionUtils;
import org.joda.time.DateTime;
import org.joda.time.DateTimeConstants;
import org.joda.time.format.DateTimeFormat;
import org.joda.time.format.DateTimeFormatter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.base.Charsets;
import com.google.common.base.Preconditions;
import com.google.common.base.Stopwatch;
import com.google.common.collect.Lists;
import com.google.common.io.FileWriteMode;
import com.google.common.io.Files;

import java.io.File;
import java.io.IOException;
import java.util.List;

/**
 * @ClassName WorkDateCalculator
 * @author kanpiaoxue
 * @version 1.0
 * @CreateTime 2017/07/16 19:54:26
 * @Description 计算工作日期的 java 工具类
 */
public class WorkDateCalculator {
    protected static final Logger LOG = LoggerFactory.getLogger(WorkDateCalculator.class);
    private static final String FORMATE_DATE = "yyyy-MM-dd";
    private static final DateTimeFormatter FORMATTER = DateTimeFormat.forPattern(FORMATE_DATE);

    
    public WorkDateCalculator() {
    }

    /**
     * @param args
     * @author kanpiaoxue
     * @CreateTime 2017/07/16 19:54:26
     * @Description main 函数
     */
    public static void main(String[] args) {
        WorkDateCalculator cal = new WorkDateCalculator();
        DateTime startDate = DateTime.parse("2021-01-01", FORMATTER);
        DateTime endDate = DateTime.parse("2021-12-31", FORMATTER);
        List<DateTime> dateTimes = cal.calculate(startDate, endDate, null, null);
        boolean createFile = true;
        cal.show(dateTimes, createFile);
    }

    /**
     *
     * @param startDate
     *            开始日期
     * @param endDate
     *            结束日期
     * @param workStartDate
     *            每周的工作日的开始是星期几？默认是周一
     * @param workEndDate
     *            每周的工作日的结束是星期几？默认是周五
     * @return 工作日列表
     * @author kanpiaoxue
     * @CreateTime 2017/07/16 19:54:26
     * @Description 根据开始日期和结束日期以及制定的每周的工作日始末位置计算工作日列表
     */
    public List<DateTime> calculate(DateTime startDate, DateTime endDate, Integer workStartDate,
            Integer workEndDate) {
        Preconditions.checkNotNull(startDate, "startDate is null");
        Preconditions.checkNotNull(endDate, "endDate is null");
        if (null == workStartDate) {
            workStartDate = DateTimeConstants.MONDAY;
        }
        if (null == workEndDate) {
            workEndDate = DateTimeConstants.FRIDAY;
        }

        LOG.debug("startDate:{},endDate:{}", startDate.toString(FORMATE_DATE),
                endDate.toString(FORMATE_DATE));
        Preconditions.checkArgument(startDate.isBefore(endDate), "startDate is after endDate");
        List<DateTime> list = Lists.newArrayList();
        list.add(startDate);
        while (startDate.isBefore(endDate)) {
            DateTime tmp = startDate.dayOfYear().addToCopy(1);
            if (tmp.getDayOfWeek() >= workStartDate && tmp.getDayOfWeek() <= workEndDate) {
                list.add(tmp);
            }
            startDate = tmp;
        }
        return list;
    }

    /**
     * 得到日期所属的星期几
     * 
     * @param dateTime
     * @return
     * @author kanpiaoxue
     * @CreateTime: 2019/12/26 10:34:29
     * @Description:
     */
    public String dayOfWeek(DateTime dateTime) {
        Preconditions.checkNotNull(dateTime, "dateTime is null");
        int dayOfWeek = dateTime.getDayOfWeek();
        String num = null;
        switch (dayOfWeek) {
            case DateTimeConstants.MONDAY:
                num = "一";
                break;
            case DateTimeConstants.TUESDAY:
                num = "二";
                break;
            case DateTimeConstants.WEDNESDAY:
                num = "三";
                break;
            case DateTimeConstants.THURSDAY:
                num = "四";
                break;
            case DateTimeConstants.FRIDAY:
                num = "五";
                break;
            case DateTimeConstants.SATURDAY:
                num = "六";
                break;
            case DateTimeConstants.SUNDAY:
                num = "日";
                break;
            default:
                throw new IllegalArgumentException(String.format("can not found dayOfWeek:%s", dayOfWeek));
        }
        return String.format("星期%s", num);
    }

    /**
     * @param dateTimes
     *            日期列表
     * @param createFile
     *            是否创建文件
     * @author kanpiaoxue
     * @CreateTime 2017/07/16 22:01:35
     * @Description 根据日期列表和创建文件的标识生成工作日（周一至周五）的工作日志内容列表
     */
    public void show(List<DateTime> dateTimes, boolean createFile) {
        LOG.info("start to show");
        Stopwatch sw = Stopwatch.createStarted();
        Preconditions.checkArgument(CollectionUtils.isNotEmpty(dateTimes), "dateTimes is null or empty!");
        String format = "\n%s\n    今日工作：\n    今日收获：\n    明日工作：\n    持续跟进：\n";

        List<String> content = Lists.newArrayList();
        for (DateTime tmp : dateTimes) {

            String dayOfWeek = dayOfWeek(tmp);

            String date = tmp.toString(FORMATTER);
            String rs = String.format(format, date + "(" + dayOfWeek + ")");
            System.out.println(rs);
            content.add(rs);
        }

        if (createFile) {
            try {
                String fileName =
                        String.format("/Users/kanpiaoxue/tmp/tmp.%s.txt", DateTime.now().toString(FORMATTER));
                LOG.info("filename : {}", fileName);
                File file = new File(fileName);
                if (file.exists()) {
                    file.delete();
                }
                Files.createParentDirs(file);
                Files.asCharSink(file, Charsets.UTF_8, FileWriteMode.APPEND).writeLines(content);
                LOG.info("the file : {}", file);
            } catch (IOException e) {
                LOG.error(e.getMessage(), e);
            }
        }

        LOG.info("finish showing. It consumes {}", sw);

    }
}