Java:计算一个日期加上指定工作日数(排除周六周日和一系列节日)
工作时遇到的,随便写了个,欢迎高手点评:
核心代码:
package www.sjjjob.com.date;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.List;
/**
* 日期公共函数
* @author shenjunjie
* @version [s001, 2010-11-19]
*
* public static final int DAY_OF_WEEK_IN_MONTHget 和 set 的字段数字,指示当前月中的第几个星期。
* 与 DAY_OF_WEEK 字段一起使用时,就可以唯一地指定某月中的某一天。
* 与 WEEK_OF_MONTH 和 WEEK_OF_YEAR 不同,
* 该字段的值并不 取决于 getFirstDayOfWeek() 或 getMinimalDaysInFirstWeek()。
* DAY_OF_MONTH 1 到 7 总是对应于 DAY_OF_WEEK_IN_MONTH 1;8 到 14 总是对应于 DAY_OF_WEEK_IN_MONTH 2,
* 依此类推。DAY_OF_WEEK_IN_MONTH 0 表示 DAY_OF_WEEK_IN_MONTH 1 之前的那个星期。
* 负值是从一个月的末尾开始逆向计数,因此,一个月的最后一个星期天被指定为 DAY_OF_WEEK = SUNDAY,
* DAY_OF_WEEK_IN_MONTH = -1。因为负值是逆向计数的,所以它们在月份中的对齐方式通常与正值的不同。
* 例如,如果一个月有 31 天,那么 DAY_OF_WEEK_IN_MONTH -1 将与 DAY_OF_WEEK_IN_MONTH 5
* 和 DAY_OF_WEEK_IN_MONTH 4 的末尾相重叠。
*/
public class DateUtil
{
private static List holidayList;
private static boolean holidayFlag;
/**
* 计算工作日
* 具体节日包含哪些,可以在HolidayMap中修改
* @param src 日期(源)
* @param adddays 要加的天数
* @exception throws [违例类型] [违例说明]
* @version [s001, 2010-11-19]
* @author shenjunjie
*/
public static Calendar addDateByWorkDay(Calendar src,int adddays)
{
// Calendar result = null;
holidayFlag = false;
for (int i = 0; i < adddays; i++)
{
//把源日期加一天
src.add(Calendar.DAY_OF_MONTH, 1);
holidayFlag =checkHoliday(src);
if(holidayFlag)
{
i--;
}
System.out.println(src.getTime());
}
System.out.println("Final Result:"+src.getTime());
return src;
}
/**
* 校验指定的日期是否在节日列表中
* 具体节日包含哪些,可以在HolidayMap中修改
* @param src 要校验的日期(源)
* @version [s001, 2010-11-19]
* @author shenjunjie
*/
public static boolean checkHoliday(Calendar src)
{
boolean result = false;
if (holidayList == null)
{
initHolidayList();
}
//先检查是否是周六周日(有些国家是周五周六)
if (src.get(Calendar.DAY_OF_WEEK) == Calendar.SATURDAY
|| src.get(Calendar.DAY_OF_WEEK) == Calendar.SUNDAY)
{
return true;
}
for (Calendar c : holidayList)
{
if (src.get(Calendar.MONTH) == c.get(Calendar.MONTH)
&& src.get(Calendar.DAY_OF_MONTH) == c.get(Calendar.DAY_OF_MONTH))
{
result = true;
}
}
return result;
}
/**
* 初始化节日List,如果需要加入新的节日,请在这里添加
* 加的时候请尽量使用Calendar自带的常量而不是魔鬼数字
* 注:年份可以随便写,因为比的时候只比月份和天
* @version [s001, 2010-11-19]
* @author shenjunjie
*/
private static void initHolidayList()
{
holidayList = new ArrayList();
//五一劳动节
Calendar may1 = Calendar.getInstance();
may1.set(Calendar.MONTH,Calendar.MAY);
may1.set(Calendar.DAY_OF_MONTH,1);
holidayList.add(may1);
Calendar may2 = Calendar.getInstance();
may2.set(Calendar.MONTH,Calendar.MAY);
may2.set(Calendar.DAY_OF_MONTH,2);
holidayList.add(may2);
Calendar may3 = Calendar.getInstance();
may3.set(Calendar.MONTH,Calendar.MAY);
may3.set(Calendar.DAY_OF_MONTH,3);
holidayList.add(may3);
Calendar h3 = Calendar.getInstance();
h3.set(2000, 1, 1);
holidayList.add(h3);
Calendar h4 = Calendar.getInstance();
h4.set(2000, 12, 25);
holidayList.add(h4);
//中国母亲节:五月的第二个星期日
Calendar may5 = Calendar.getInstance();
//设置月份为5月
may5.set(Calendar.MONTH,Calendar.MAY);
//设置星期:第2个星期
may5.set(Calendar.DAY_OF_WEEK_IN_MONTH,2);
//星期日
may5.set(Calendar.DAY_OF_WEEK,Calendar.SUNDAY);
// System.out.println(may5.getTime());
holidayList.add(may5);
}
}
下面是JUnit4的测试代码:
package www.sjjjob.com.date;
import java.util.Calendar;
import junit.framework.Assert;
import org.junit.After;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
public class DateUtilTest
{
@Test
public void testAddDateByWorkDay()
{
Calendar src = Calendar.getInstance();
src.set(2010, Calendar.APRIL , 29);
Calendar result = DateUtil.addDateByWorkDay(src, 2);
Calendar expected = Calendar.getInstance();
expected.set(2010, Calendar.MAY , 4);
Assert.assertEquals(expected.getTime().toString(), result.getTime().toString());
}
@Test
public void testAddDateByWorkDay2()
{
//测试母亲节
Calendar src = Calendar.getInstance();
src.set(2010, Calendar.MAY , 7);
Calendar result = DateUtil.addDateByWorkDay(src, 2);
Calendar expected = Calendar.getInstance();
expected.set(2010, Calendar.MAY , 11);
Assert.assertEquals(expected.getTime().toString(), result.getTime().toString());
}
}