org.wikibrain.pageview.PageViewUtils Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of wikibrain-pageview Show documentation

There is a newer version: 0.9.1

package org.wikibrain.pageview;

import org.joda.time.DateTime;

import java.util.SortedSet;
import java.util.TreeSet;

/**
 * @author Shilad Sen
 */
public class PageViewUtils {
    public static SortedSet timestampsInInterval(DateTime start, DateTime end) {
        if (start.isAfter(end)) {
            throw new IllegalArgumentException();
        }
        DateTime current = new DateTime(
                start.year().get(),
                start.monthOfYear().get(),
                start.dayOfMonth().get(),
                start.hourOfDay().get(),
                0);
        if (current.isBefore(start)) {
            current = current.plusHours(1);
        }
        SortedSet result = new TreeSet();
        while (!current.isAfter(end)) {
            result.add(current);
            current = current.plusHours(1);
        }
        return result;
    }
}