blob: 670cc2afae2d51e0cb6750426e272c6e38a80aec [file] [log] [blame] [edit]
package fileutil
import (
"fmt"
"strings"
"time"
)
// GetHourlyDirs generates paths with the given prefix for all the paths of the
// form 'prefix/<year>/<month>/<day>/<hour>'. All path segments are padded with zeros
// where necessary, e.g. prefix/2018/01/02/09. All hours are from 0-23. This is
// intended to enumerate the directories that should be polled when ingesting data.
// startTS and endTS are seconds since the Unix epoch. Both time stamps are inclusive.
// The generated directories are based on UTC being the locale.
// If the prefix is empty string, there will be no slash before the year.
func GetHourlyDirs(prefixDir string, start, end time.Time) []string {
if end.Before(start) {
return []string{}
}
if len(prefixDir) != 0 && !strings.HasSuffix(prefixDir, "/") {
prefixDir += "/"
}
// The result will be roughly the number of hours in the range.
ret := make([]string, 0, end.Sub(start)/time.Hour+1)
currTime := start.UTC()
endTime := end.UTC()
// Change the timestamp of the last hour to the very last millisecond of the
// hour. This guarantees that we always include the last hour because
// endTime is just below the next hour but larger than any incremented value
// of currTime can be since currTime is measured in seconds.
lastHourDelta := time.Duration(59-endTime.Minute())*time.Minute +
time.Duration(59-endTime.Second())*time.Second +
999*time.Millisecond
endTime = endTime.Add(lastHourDelta)
for currTime.Before(endTime) {
year, month, day := currTime.Date()
hour := currTime.Hour()
ret = append(ret, fmt.Sprintf("%s%04d/%02d/%02d/%02d", prefixDir, year, month, day, hour))
currTime = currTime.Add(time.Hour)
}
return ret
}