com.scudata.expression.fn.datetime.Period Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of esproc Show documentation
Show all versions of esproc Show documentation
SPL(Structured Process Language) A programming language specially for structured data computing.
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;
import com.scudata.util.Variant;
/**
* periods(s,e,i) ??ȡ??s??e???????˵㣩ÿ???i??????ʱ??ֵ???ɵ????С?
* @author runqian
*
*/
public class Period 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("period" + 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("period" + 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("period" + mm.getMessage("function.invalidParam"));
}
Object start = sub1.getLeafExpression().calculate(ctx);
Object end = sub2.getLeafExpression().calculate(ctx);
if (start instanceof String) {
start = Variant.parseDate((String)start);
}
if (end instanceof String) {
end = Variant.parseDate((String)end);
}
if (!(start instanceof Date) || !(end instanceof Date)) {
MessageManager mm = EngineMessage.get();
throw new RQException("period" + mm.getMessage("function.paramTypeError"));
}
int dist = 1;
if (size > 2) {
IParam sub3 = param.getSub(2);
if (sub3 != null) {
Object distance = sub3.getLeafExpression().calculate(ctx);
if (!(distance instanceof Number)) {
MessageManager mm = EngineMessage.get();
throw new RQException("period" + mm.getMessage("function.paramTypeError"));
}
dist = ((Number)distance).intValue();
if (dist == 0) {
MessageManager mm = EngineMessage.get();
throw new RQException("period" + mm.getMessage("function.invalidParam"));
}
}
}
if (option == null || (option.indexOf('o') == -1 && option.indexOf('t') == -1)) {
return periodA((Date) start, (Date) end, dist, option); // ????
} else {
return periodO((Date) start, (Date) end, dist, option); // ????????Ѯ????
}
}
private Sequence periodO(Date start, Date end, int distance, String opt) {
int field;
if (opt == null) { // ??
field = Calendar.DATE;
} else if (opt.indexOf('y') != -1) { // ??
field = Calendar.YEAR;
} else if (opt.indexOf('q') != -1) { // ??
field = Calendar.MONTH;
distance *= 3;
} else if (opt.indexOf('m') != -1) { // ??
field = Calendar.MONTH;
} else if (opt.indexOf('t') != -1) { // Ѯ
field = -1;
} else if (opt.indexOf('s') != -1) { // ??
field = Calendar.SECOND;
} else { // ??
field = Calendar.DATE;
}
Sequence series = new Sequence();
series.add(start);
Calendar gc = Calendar.getInstance();
if (distance > 0) {
if (end.compareTo(start) < 0) {
MessageManager mm = EngineMessage.get();
throw new RQException("period" + mm.getMessage("function.invalidParam"));
}
if (field == -1) { // Ѯ
Date prev = start;
gc.setTime(start);
int day = gc.get(Calendar.DATE);
// ??dayΪѮ?ĵ?һ??
if (day < 11) {
day = 1;
} else if (day < 21) {
day = 11;
} else {
day = 21;
}
while (true) {
gc.setTime(prev);
for (int i = 0; i < distance; ++i) {
day += 10;
if (day == 31) {
day = 1;
gc.set(Calendar.DATE, 1);
gc.add(Calendar.MONTH, 1);
}
}
gc.set(Calendar.DATE, day);
Date tmp = (Date)prev.clone();
tmp.setTime(gc.getTimeInMillis());
gc.clear();
int cmp = tmp.compareTo(end);
if (cmp < 0) {
series.add(tmp);
prev = tmp;
} else {
if (opt == null || opt.indexOf('x') == -1) {
series.add(end);
}
break;
}
}
} else {
int times = 1;
while (true) {
gc.setTime(start);
gc.add(field, distance * times);
Date tmp = (Date) start.clone();
tmp.setTime(gc.getTimeInMillis());
gc.clear();
int cmp = tmp.compareTo(end);
if (cmp < 0) {
series.add(tmp);
times++;
} else {
if (opt == null || opt.indexOf('x') == -1) {
series.add(end);
}
break;
}
}
}
} else { // distance < 0
if (start.compareTo(end) < 0) {
MessageManager mm = EngineMessage.get();
throw new RQException("period" + mm.getMessage("function.invalidParam"));
}
if (field == -1) { // Ѯ
Date prev = start;
gc.setTime(start);
int day = gc.get(Calendar.DATE);
// ??dayΪѮ?ĵ?һ??
if (day < 11) {
day = 1;
} else if (day < 21) {
day = 11;
} else {
day = 21;
}
while (true) {
gc.setTime(prev);
for (int i = distance; i < 0; ++i) {
day -= 10;
if (day < 0) {
day = 21;
gc.set(Calendar.DATE, 21);
gc.add(Calendar.MONTH, -1);
}
}
gc.set(Calendar.DATE, day);
Date tmp = (Date)prev.clone();
tmp.setTime(gc.getTimeInMillis());
gc.clear();
int cmp = tmp.compareTo(end);
if (cmp > 0) {
series.add(tmp);
prev = tmp;
} else {
if (opt == null || opt.indexOf('x') == -1) {
series.add(end);
}
break;
}
}
} else {
int times = 1;
while (true) {
gc.setTime(start);
gc.add(field, distance * times);
Date tmp = (Date) start.clone();
tmp.setTime(gc.getTimeInMillis());
gc.clear();
int cmp = tmp.compareTo(end);
if (cmp > 0) {
series.add(tmp);
times++;
} else {
if (opt == null || opt.indexOf('x') == -1) {
series.add(end);
}
break;
}
}
}
}
return series;
}
private Sequence periodA(Date start, Date end, int distance, String opt) {
Calendar gc = Calendar.getInstance();
gc.setTime(start);
int field;
if (opt == null) { // ??
field = Calendar.DATE;
gc.set(Calendar.HOUR_OF_DAY, 0);
gc.set(Calendar.MINUTE, 0);
gc.set(Calendar.SECOND, 0);
gc.set(Calendar.MILLISECOND, 0);
} else if (opt.indexOf('y') != -1) { // ??
field = Calendar.YEAR;
gc.set(Calendar.DATE, 1);
gc.set(Calendar.MONTH, Calendar.JANUARY);
gc.set(Calendar.HOUR_OF_DAY, 0);
gc.set(Calendar.MINUTE, 0);
gc.set(Calendar.SECOND, 0);
gc.set(Calendar.MILLISECOND, 0);
} else if (opt.indexOf('q') != -1) { // ??
field = Calendar.MONTH;
distance *= 3;
int month = gc.get(Calendar.MONTH);
if (month < Calendar.APRIL) {
month = Calendar.JANUARY;
} else if (month < Calendar.JULY) {
month = Calendar.APRIL;
} else if (month < Calendar.OCTOBER) {
month = Calendar.JULY;
} else {
month = Calendar.OCTOBER;
}
gc.set(Calendar.DATE, 1);
gc.set(Calendar.MONTH, month);
gc.set(Calendar.HOUR_OF_DAY, 0);
gc.set(Calendar.MINUTE, 0);
gc.set(Calendar.SECOND, 0);
gc.set(Calendar.MILLISECOND, 0);
} else if (opt.indexOf('m') != -1) { // ??
field = Calendar.MONTH;
gc.set(Calendar.DATE, 1);
gc.set(Calendar.HOUR_OF_DAY, 0);
gc.set(Calendar.MINUTE, 0);
gc.set(Calendar.SECOND, 0);
gc.set(Calendar.MILLISECOND, 0);
} else if (opt.indexOf('s') != -1) { // ??
field = Calendar.SECOND;
gc.set(Calendar.MILLISECOND, 0);
} else { // ??
field = Calendar.DATE;
gc.set(Calendar.HOUR_OF_DAY, 0);
gc.set(Calendar.MINUTE, 0);
gc.set(Calendar.SECOND, 0);
gc.set(Calendar.MILLISECOND, 0);
}
long startTime = gc.getTimeInMillis();
gc.clear();
Sequence series = new Sequence();
series.add(start);
int times = 1;
if (distance > 0) {
if (end.compareTo(start) < 0) {
MessageManager mm = EngineMessage.get();
throw new RQException("period" + mm.getMessage("function.invalidParam"));
}
while (true) {
gc.setTimeInMillis(startTime);
gc.add(field, distance * times);
Date tmp = (Date)start.clone();
tmp.setTime(gc.getTimeInMillis());
gc.clear();
int cmp = tmp.compareTo(end);
if (cmp < 0) {
series.add(tmp);
times++;
} else {
if (opt == null || opt.indexOf('x') == -1) {
series.add(end);
}
break;
}
}
} else {
if (start.compareTo(end) < 0) {
MessageManager mm = EngineMessage.get();
throw new RQException("period" + mm.getMessage("function.invalidParam"));
}
while (true) {
gc.setTimeInMillis(startTime);
gc.add(field, distance * times);
Date tmp = (Date)start.clone();
tmp.setTime(gc.getTimeInMillis());
gc.clear();
int cmp = tmp.compareTo(end);
if (cmp > 0) {
series.add(tmp);
times++;
} else {
if (opt == null || opt.indexOf('x') == -1) {
series.add(end);
}
break;
}
}
}
return series;
}
}