com.scudata.expression.fn.datetime.WorkDay Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of esproc Show documentation
SPL(Structured Process Language) A programming language specially for structured data computing.
There is a newer version: 20240823
package com.scudata.expression.fn.datetime;

import java.util.Calendar;
import java.util.Date;

import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.Context;
import com.scudata.dm.Sequence;
import com.scudata.expression.Function;
import com.scudata.expression.IParam;
import com.scudata.expression.Node;
import com.scudata.resources.EngineMessage;

/**
 * workday(t,k,h) ?????????t???k???????յ????ڣ?h??(??)???????У???h?г?Ա??????ĩ???Ǽ??գ?
 * ????ĩ??Ǽ??գ???Ϊ??ĩʱ?????ݼ??㣬??Ϊ??????
 * @author runqian
 *
 */
public class WorkDay extends Function {
	public Node optimize(Context ctx) {
		param.optimize(ctx);
		return this;
	}
	
	/**
	 * ??????ʽ????Ч?ԣ???Ч???׳??쳣
	 */
	public void checkValidity() {
		if (param == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("workday" + mm.getMessage("function.missingParam"));
		}
	}

	public Object calculate(Context ctx) {
		int size = param.getSubSize();
		if (size != 2 && size != 3) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("workday" + mm.getMessage("function.invalidParam"));
		}

		IParam sub1 = param.getSub(0);
		IParam sub2 = param.getSub(1);
		if (sub1 == null || sub2 == null) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("workday" + mm.getMessage("function.invalidParam"));
		}

		Object result1 = sub1.getLeafExpression().calculate(ctx);
		Object result2 = sub2.getLeafExpression().calculate(ctx);
		if (!(result1 instanceof Date) || !(result2 instanceof Number)) {
			MessageManager mm = EngineMessage.get();
			throw new RQException("workday" + mm.getMessage("function.paramTypeError"));
		}

		Sequence offDays = null;
		if (size == 3) {
			IParam sub3 = param.getSub(2);
			if (sub3 != null) {
				Object obj = sub3.getLeafExpression().calculate(ctx);
				if (obj instanceof Sequence) {
					offDays = (Sequence)obj;
				} else if (obj != null) {
					MessageManager mm = EngineMessage.get();
					throw new RQException("workday" + mm.getMessage("function.paramTypeError"));
				}
			}
		}

		Date date1 = (Date)result1;
		Calendar calendar = Calendar.getInstance();
		calendar.setTime(date1);
		int diff = ((Number)result2).intValue();
		int d = 1;
		if (diff < 0) {
			d = -1;
		} else if (diff == 0) {
			return date1;
		}

		while (diff != 0) {
			calendar.add(Calendar.DATE, d);
			if (isWorkDay(calendar, offDays)) diff -= d;
		}

		Date date = (Date)date1.clone();
		date.setTime(calendar.getTimeInMillis());
		return date;
	}

	public static boolean isWorkDay(Calendar calendar, Sequence offDays) {
		int week = calendar.get(Calendar.DAY_OF_WEEK);
		boolean isWorkDay = week != Calendar.SUNDAY && week != Calendar.SATURDAY;
		if (offDays == null || offDays.length() == 0) return isWorkDay;

		int year = calendar.get(Calendar.YEAR);
		int month = calendar.get(Calendar.MONTH);
		int day = calendar.get(Calendar.DATE);
		Calendar c2 = Calendar.getInstance();

		for (int i = 1, count = offDays.length(); i <= count; ++i) {
			Object obj = offDays.get(i);
			if (!(obj instanceof Date)) {
				MessageManager mm = EngineMessage.get();
				throw new RQException("workday" + mm.getMessage("function.paramTypeError"));
			}

			c2.setTime((Date)obj);
			if (c2.get(Calendar.YEAR) == year && c2.get(Calendar.MONTH) == month && 
					c2.get(Calendar.DATE) == day) {
				return !isWorkDay;
			}
		}

		return isWorkDay;
	}
}