/* * Copyright 2015 Eediom Inc. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.araqne.logdb.query.expr; import java.util.ArrayList; import java.util.Date; import java.util.List; import org.araqne.logdb.QueryContext; import org.araqne.logdb.Row; import org.araqne.logdb.TimeSpan; import org.araqne.logdb.TimeUnit; /** * @since 2.8.12 * @author xeraph * */ public class DateRange extends FunctionExpression { private Expression begin; private Expression end; private long millis; public DateRange(QueryContext ctx, List<Expression> exprs) { super("daterange", exprs, 2); begin = exprs.get(0); end = exprs.get(1); if (exprs.size() > 2) { millis = TimeSpan.parse(exprs.get(2).eval(null).toString()).getMillis(); } else { millis = TimeUnit.Day.getMillis(); } } @Override public Object eval(Row row) { Object o1 = begin.eval(row); Object o2 = end.eval(row); if (o1 == null || o2 == null) return null; if (!(o1 instanceof Date) || !(o2 instanceof Date)) return null; long begin = ((Date) o1).getTime(); long end = ((Date) o2).getTime(); List<Date> dates = new ArrayList<Date>(); for (long l = begin; l < end; l += millis) { dates.add(new Date(l)); } return dates; } }