com.addthis.hydra.data.util.DateUtil.java Source code

Introduction

Here is the source code for com.addthis.hydra.data.util.DateUtil.java

Source

/*
 * 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 com.addthis.hydra.data.util;

import java.util.Iterator;

import org.joda.time.DateTime;
import org.joda.time.format.DateTimeFormat;
import org.joda.time.format.DateTimeFormatter;

/**
 * date helper that understands the {{now}} {{now+offset}} {{now-offset}} syntax.
 */
public final class DateUtil {

    public static final String NOW_PREFIX = "{{now";
    public static final String NOW_POSTFIX = "}}";
    public static final String OFF = "{{off}}";
    public static final String NOW = NOW_PREFIX + NOW_POSTFIX;

    public static DateTime getDateTime(String format, String date) {
        return getDateTime(getFormatter(format), date);
    }

    public static DateTime getDateTime(DateTimeFormatter formatter, String date) {
        if (date.startsWith(NOW_PREFIX) && date.endsWith(NOW_POSTFIX)) {
            if (date.equals(NOW)) {
                return new DateTime();
            }
            DateTime time = new DateTime();
            int pos;
            if ((pos = date.indexOf("+")) > 0) {
                time = time
                        .plusDays(Integer.parseInt(date.substring(pos + 1, date.length() - NOW_POSTFIX.length())));
            } else if ((pos = date.indexOf("-")) > 0) {
                time = time
                        .minusDays(Integer.parseInt(date.substring(pos + 1, date.length() - NOW_POSTFIX.length())));
            }
            return time;
        }
        return formatter.parseDateTime(date);
    }

    public static String format(String format, long time) {
        return format(getFormatter(format), time);
    }

    public static String format(DateTimeFormatter formatter, long time) {
        return formatter.print(time);
    }

    public static DateTimeFormatter getFormatter(String format) {
        return DateTimeFormat.forPattern(format);
    }

    public static Iterator<DateTime> getDayIterator(final DateTime start, final DateTime end) {
        return new DateIterator(start, end, false);
    }

    public static Iterator<DateTime> getHourIterator(final DateTime start, final DateTime end) {
        return new DateIterator(start, end, true);
    }

    /** */
    private static class DateIterator implements Iterator<DateTime> {

        private final boolean reverse;
        private final boolean hourly;
        private DateTime start;
        private DateTime end;

        DateIterator(DateTime start, DateTime end, boolean hourly) {
            reverse = start.compareTo(end) > 0;
            this.hourly = hourly;
            this.start = start;
            this.end = end;
        }

        @Override
        public boolean hasNext() {
            return reverse ? start.compareTo(end) > 0 : end.compareTo(start) > 0;
        }

        @Override
        public DateTime next() {
            DateTime ret = start;
            if (reverse) {
                if (hourly) {
                    start.minusHours(1);
                } else {
                    start.plusHours(1);
                }
            } else {
                if (hourly) {
                    start.minusDays(1);
                } else {
                    start.plusDays(1);
                }
            }
            return ret;
        }

        @Override
        public void remove() {
            throw new UnsupportedOperationException();
        }
    }
}