| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100 |
- package opencost
- import (
- "bytes"
- "encoding/json"
- "fmt"
- "math"
- "regexp"
- "strconv"
- "sync"
- "time"
- "github.com/opencost/opencost/core/pkg/env"
- "github.com/opencost/opencost/core/pkg/log"
- "github.com/opencost/opencost/core/pkg/util/timeutil"
- )
- var (
- durationRegex = regexp.MustCompile(`^(\d+)(m|h|d|w)$`)
- durationOffsetRegex = regexp.MustCompile(`^(\d+)(m|h|d|w) offset (\d+)(m|h|d|w)$`)
- offesetRegex = regexp.MustCompile(`^(\+|-)(\d\d):(\d\d)$`)
- rfc3339 = `\d\d\d\d-\d\d-\d\dT\d\d:\d\d:\d\dZ`
- rfcRegex = regexp.MustCompile(fmt.Sprintf(`(%s),(%s)`, rfc3339, rfc3339))
- timestampPairRegex = regexp.MustCompile(`^(\d+)[,|-](\d+)$`)
- utcOffsetLock sync.Mutex
- utcOffsetDur *time.Duration
- )
- // returns the configured utc offset as a duration
- // TODO: Same as the above options -- we should provide a one-time initialization configuration
- // TODO: for these values, or deprecate their use.
- func utcOffset() time.Duration {
- utcOffsetLock.Lock()
- defer utcOffsetLock.Unlock()
- if utcOffsetDur == nil {
- utcOff, err := timeutil.ParseUTCOffset(env.Get("UTC_OFFSET", ""))
- if err != nil {
- utcOff = time.Duration(0)
- }
- utcOffsetDur = &utcOff
- }
- return *utcOffsetDur
- }
- // RoundBack rounds the given time back to a multiple of the given resolution
- // in the given time's timezone.
- // e.g. 2020-01-01T12:37:48-0700, 24h = 2020-01-01T00:00:00-0700
- func RoundBack(t time.Time, resolution time.Duration) time.Time {
- // if the duration is a week - roll back to the following Sunday
- if resolution == timeutil.Week {
- return timeutil.RoundToStartOfWeek(t)
- }
- _, offSec := t.Zone()
- return t.Add(time.Duration(offSec) * time.Second).Truncate(resolution).Add(-time.Duration(offSec) * time.Second)
- }
- // RoundForward rounds the given time forward to a multiple of the given resolution
- // in the given time's timezone.
- // e.g. 2020-01-01T12:37:48-0700, 24h = 2020-01-02T00:00:00-0700
- func RoundForward(t time.Time, resolution time.Duration) time.Time {
- back := RoundBack(t, resolution)
- if back.Equal(t) {
- // The given time is exactly a multiple of the given resolution
- return t
- }
- return back.Add(resolution)
- }
- // Window defines a period of time with a start and an end. If either start or
- // end are nil it indicates an open time period.
- type Window struct {
- start *time.Time
- end *time.Time
- }
- // NewWindow creates and returns a new Window instance from the given times
- func NewWindow(start, end *time.Time) Window {
- return Window{
- start: start,
- end: end,
- }
- }
- // NewClosedWindow creates and returns a new Window instance from the given
- // times, which cannot be nil, so they are value types.
- func NewClosedWindow(start, end time.Time) Window {
- return Window{
- start: &start,
- end: &end,
- }
- }
- // ParseWindowUTC attempts to parse the given string into a valid Window. It
- // accepts several formats, returning an error if the given string does not
- // match one of the following:
- // - named intervals: "today", "yesterday", "week", "month", "lastweek", "lastmonth"
- // - durations: "24h", "7d", etc.
- // - date ranges: "2020-04-01T00:00:00Z,2020-04-03T00:00:00Z", etc.
- // - timestamp ranges: "1586822400,1586908800", etc.
- func ParseWindowUTC(window string) (Window, error) {
- return parseWindow(window, time.Now().UTC())
- }
- // ParseWindowWithOffsetString parses the given window string within the context of
- // the timezone defined by the UTC offset string of format -07:00, +01:30, etc.
- func ParseWindowWithOffsetString(window string, offset string) (Window, error) {
- if offset == "UTC" || offset == "" {
- return ParseWindowUTC(window)
- }
- match := offesetRegex.FindStringSubmatch(offset)
- if match == nil {
- return Window{}, fmt.Errorf("illegal UTC offset: '%s'; should be of form '-07:00'", offset)
- }
- sig := 1
- if match[1] == "-" {
- sig = -1
- }
- hrs64, _ := strconv.ParseInt(match[2], 10, 64)
- hrs := sig * int(hrs64)
- mins64, _ := strconv.ParseInt(match[3], 10, 64)
- mins := sig * int(mins64)
- loc := time.FixedZone(fmt.Sprintf("UTC%s", offset), (hrs*60*60)+(mins*60))
- now := time.Now().In(loc)
- return parseWindow(window, now)
- }
- // ParseWindowWithOffset parses the given window string within the context of
- // the timezone defined by the UTC offset.
- func ParseWindowWithOffset(window string, offset time.Duration) (Window, error) {
- loc := time.FixedZone("", int(offset.Seconds()))
- now := time.Now().In(loc)
- return parseWindow(window, now)
- }
- // parseWindow generalizes the parsing of window strings, relative to a given
- // moment in time, defined as "now".
- func parseWindow(window string, now time.Time) (Window, error) {
- // compute UTC offset in terms of minutes
- offHr := now.UTC().Hour() - now.Hour()
- offMin := (now.UTC().Minute() - now.Minute()) + (offHr * 60)
- offset := time.Duration(offMin) * time.Minute
- if window == "today" {
- start := now
- start = start.Truncate(time.Hour * 24)
- start = start.Add(offset)
- end := start.Add(time.Hour * 24)
- return NewWindow(&start, &end), nil
- }
- if window == "yesterday" {
- start := now
- start = start.Truncate(time.Hour * 24)
- start = start.Add(offset)
- start = start.Add(time.Hour * -24)
- end := start.Add(time.Hour * 24)
- return NewWindow(&start, &end), nil
- }
- if window == "week" {
- // now
- start := now
- // 00:00 today, accounting for timezone offset
- start = start.Truncate(time.Hour * 24)
- start = start.Add(offset)
- // 00:00 Sunday of the current week
- start = start.Add(-24 * time.Hour * time.Duration(start.Weekday()))
- end := now
- return NewWindow(&start, &end), nil
- }
- if window == "lastweek" {
- // now
- start := now
- // 00:00 today, accounting for timezone offset
- start = start.Truncate(time.Hour * 24)
- start = start.Add(offset)
- // 00:00 Sunday of last week
- start = start.Add(-24 * time.Hour * time.Duration(start.Weekday()+7))
- end := start.Add(7 * 24 * time.Hour)
- return NewWindow(&start, &end), nil
- }
- if window == "month" {
- // now
- start := now
- // 00:00 today, accounting for timezone offset
- start = start.Truncate(time.Hour * 24)
- start = start.Add(offset)
- // 00:00 1st of this month
- start = start.Add(-24 * time.Hour * time.Duration(start.Day()-1))
- end := now
- return NewWindow(&start, &end), nil
- }
- if window == "month" {
- // now
- start := now
- // 00:00 today, accounting for timezone offset
- start = start.Truncate(time.Hour * 24)
- start = start.Add(offset)
- // 00:00 1st of this month
- start = start.Add(-24 * time.Hour * time.Duration(start.Day()-1))
- end := now
- return NewWindow(&start, &end), nil
- }
- if window == "lastmonth" {
- // now
- end := now
- // 00:00 today, accounting for timezone offset
- end = end.Truncate(time.Hour * 24)
- end = end.Add(offset)
- // 00:00 1st of this month
- end = end.Add(-24 * time.Hour * time.Duration(end.Day()-1))
- // 00:00 last day of last month
- start := end.Add(-24 * time.Hour)
- // 00:00 1st of last month
- start = start.Add(-24 * time.Hour * time.Duration(start.Day()-1))
- return NewWindow(&start, &end), nil
- }
- // Match duration strings; e.g. "45m", "24h", "7d"
- match := durationRegex.FindStringSubmatch(window)
- if match != nil {
- dur := time.Minute
- if match[2] == "h" {
- dur = time.Hour
- }
- if match[2] == "d" {
- dur = 24 * time.Hour
- }
- if match[2] == "w" {
- dur = timeutil.Week
- }
- num, _ := strconv.ParseInt(match[1], 10, 64)
- end := now
- start := end.Add(-time.Duration(num) * dur)
- // when using windows such as "7d", "1w", and "2h" we have to have a definition for what "the past X days/hours" means.
- // let "the past X days" be defined as the entirety of today plus the entirety of the past X-1 days, where
- // "entirety" is defined as midnight to midnight, UTC. given this definition, we round forward the calculated
- // start and end times to the nearest day to align with midnight boundaries
- // an analogous definition applies to "the past X weeks" and "the past X hours"
- if match[2] == "w" {
- // special case - with a week, we say the week ends today
- end = end.Truncate(timeutil.Day).Add(timeutil.Day)
- start = start.Truncate(timeutil.Day).Add(timeutil.Day)
- } else {
- end = now.Truncate(dur).Add(dur)
- start = end.Add(-time.Duration(num) * dur)
- }
- return NewWindow(&start, &end), nil
- }
- // Match duration strings with offset; e.g. "45m offset 15m", etc.
- match = durationOffsetRegex.FindStringSubmatch(window)
- if match != nil {
- end := now
- offUnit := time.Minute
- if match[4] == "h" {
- offUnit = time.Hour
- }
- if match[4] == "d" {
- offUnit = 24 * time.Hour
- }
- if match[4] == "w" {
- offUnit = 24 * timeutil.Week
- }
- offNum, _ := strconv.ParseInt(match[3], 10, 64)
- end = end.Add(-time.Duration(offNum) * offUnit)
- durUnit := time.Minute
- if match[2] == "h" {
- durUnit = time.Hour
- }
- if match[2] == "d" {
- durUnit = 24 * time.Hour
- }
- if match[2] == "w" {
- durUnit = timeutil.Week
- }
- durNum, _ := strconv.ParseInt(match[1], 10, 64)
- start := end.Add(-time.Duration(durNum) * durUnit)
- return NewWindow(&start, &end), nil
- }
- // Match timestamp pairs, e.g. "1586822400,1586908800" or "1586822400-1586908800"
- match = timestampPairRegex.FindStringSubmatch(window)
- if match != nil {
- s, _ := strconv.ParseInt(match[1], 10, 64)
- e, _ := strconv.ParseInt(match[2], 10, 64)
- start := time.Unix(s, 0).UTC()
- end := time.Unix(e, 0).UTC()
- return NewWindow(&start, &end), nil
- }
- // Match RFC3339 pairs, e.g. "2020-04-01T00:00:00Z,2020-04-03T00:00:00Z"
- match = rfcRegex.FindStringSubmatch(window)
- if match != nil {
- start, _ := time.Parse(time.RFC3339, match[1])
- end, _ := time.Parse(time.RFC3339, match[2])
- return NewWindow(&start, &end), nil
- }
- return Window{nil, nil}, fmt.Errorf("illegal window: %s", window)
- }
- // ApproximatelyEqual returns true if the start and end times of the two windows,
- // respectively, are within the given threshold of each other.
- func (w Window) ApproximatelyEqual(that Window, threshold time.Duration) bool {
- return approxEqual(w.start, that.start, threshold) && approxEqual(w.end, that.end, threshold)
- }
- func approxEqual(x *time.Time, y *time.Time, threshold time.Duration) bool {
- // both times are nil, so they are equal
- if x == nil && y == nil {
- return true
- }
- // one time is nil, but the other is not, so they are not equal
- if x == nil || y == nil {
- return false
- }
- // neither time is nil, so they are approximately close if their times are
- // within the given threshold
- delta := math.Abs((*x).Sub(*y).Seconds())
- return delta < threshold.Seconds()
- }
- func (w Window) Clone() Window {
- var start, end *time.Time
- var s, e time.Time
- if w.start != nil {
- s = *w.start
- start = &s
- }
- if w.end != nil {
- e = *w.end
- end = &e
- }
- return NewWindow(start, end)
- }
- func (w Window) Contains(t time.Time) bool {
- if w.start != nil && t.Before(*w.start) {
- return false
- }
- if w.end != nil && t.After(*w.end) {
- return false
- }
- return true
- }
- func (w Window) ContainsWindow(that Window) bool {
- // only support containing closed windows for now
- // could check if openness is compatible with closure
- if that.IsOpen() {
- return false
- }
- return w.Contains(*that.start) && w.Contains(*that.end)
- }
- func (w Window) Duration() time.Duration {
- if w.IsOpen() {
- // TODO test
- return time.Duration(math.Inf(1.0))
- }
- return w.end.Sub(*w.start)
- }
- func (w Window) End() *time.Time {
- return w.end
- }
- func (w Window) Equal(that Window) bool {
- if w.start != nil && that.start != nil && !w.start.Equal(*that.start) {
- // starts are not nil, but not equal
- return false
- }
- if w.end != nil && that.end != nil && !w.end.Equal(*that.end) {
- // ends are not nil, but not equal
- return false
- }
- if (w.start == nil && that.start != nil) || (w.start != nil && that.start == nil) {
- // one start is nil, the other is not
- return false
- }
- if (w.end == nil && that.end != nil) || (w.end != nil && that.end == nil) {
- // one end is nil, the other is not
- return false
- }
- // either both starts are nil, or they match; likewise for the ends
- return true
- }
- func (w Window) ExpandStart(start time.Time) Window {
- if w.start == nil || start.Before(*w.start) {
- w.start = &start
- }
- return w
- }
- func (w Window) ExpandEnd(end time.Time) Window {
- if w.end == nil || end.After(*w.end) {
- w.end = &end
- }
- return w
- }
- func (w Window) Expand(that Window) Window {
- if that.start == nil {
- w.start = nil
- } else {
- w = w.ExpandStart(*that.start)
- }
- if that.end == nil {
- w.end = nil
- } else {
- w = w.ExpandEnd(*that.end)
- }
- return w
- }
- func (w Window) ContractStart(start time.Time) Window {
- if w.start == nil || start.After(*w.start) {
- w.start = &start
- }
- return w
- }
- func (w Window) ContractEnd(end time.Time) Window {
- if w.end == nil || end.Before(*w.end) {
- w.end = &end
- }
- return w
- }
- func (w Window) Contract(that Window) Window {
- if that.start != nil {
- w = w.ContractStart(*that.start)
- }
- if that.end != nil {
- w = w.ContractEnd(*that.end)
- }
- return w
- }
- func (w Window) Hours() float64 {
- if w.IsOpen() {
- return math.Inf(1)
- }
- return w.end.Sub(*w.start).Hours()
- }
- // IsEmpty a Window is empty if it does not have a start and an end
- func (w Window) IsEmpty() bool {
- return w.start == nil && w.end == nil
- }
- // HasDuration a Window has duration if neither start and end are not nil and not equal
- func (w Window) HasDuration() bool {
- return !w.IsOpen() && !w.end.Equal(*w.Start())
- }
- // IsNegative a Window is negative if start and end are not null and end is before start
- func (w Window) IsNegative() bool {
- return !w.IsOpen() && w.end.Before(*w.Start())
- }
- // IsOpen a Window is open if it has a nil start or end
- func (w Window) IsOpen() bool {
- return w.start == nil || w.end == nil
- }
- func (w Window) MarshalJSON() ([]byte, error) {
- buffer := bytes.NewBufferString("{")
- if w.start != nil {
- buffer.WriteString(fmt.Sprintf("\"start\":\"%s\",", w.start.Format(time.RFC3339)))
- } else {
- buffer.WriteString(fmt.Sprintf("\"start\":\"%s\",", "null"))
- }
- if w.end != nil {
- buffer.WriteString(fmt.Sprintf("\"end\":\"%s\"", w.end.Format(time.RFC3339)))
- } else {
- buffer.WriteString(fmt.Sprintf("\"end\":\"%s\"", "null"))
- }
- buffer.WriteString("}")
- return buffer.Bytes(), nil
- }
- func (w *Window) UnmarshalJSON(bs []byte) error {
- // Due to the behavior of our custom MarshalJSON, we unmarshal as strings
- // and then manually handle the weird quoted "null" case.
- type PubWindow struct {
- Start string `json:"start"`
- End string `json:"end"`
- }
- var pw PubWindow
- err := json.Unmarshal(bs, &pw)
- if err != nil {
- return fmt.Errorf("half unmarshal: %w", err)
- }
- var start *time.Time
- var end *time.Time
- if pw.Start != "null" {
- t, err := time.Parse(time.RFC3339, pw.Start)
- if err != nil {
- return fmt.Errorf("parsing start as RFC3339: %w", err)
- }
- start = &t
- }
- if pw.End != "null" {
- t, err := time.Parse(time.RFC3339, pw.End)
- if err != nil {
- return fmt.Errorf("parsing end as RFC3339: %w", err)
- }
- end = &t
- }
- w.start = start
- w.end = end
- return nil
- }
- func (w Window) Minutes() float64 {
- if w.IsOpen() {
- return math.Inf(1)
- }
- return w.end.Sub(*w.start).Minutes()
- }
- // Overlaps returns true iff the two given Windows share an amount of temporal
- // coverage.
- // TODO complete (with unit tests!) and then implement in AllocationSet.accumulate
- // TODO:CLEANUP
- // func (w Window) Overlaps(x Window) bool {
- // if (w.start == nil && w.end == nil) || (x.start == nil && x.end == nil) {
- // // one window is completely open, so overlap is guaranteed
- // // <---------->
- // // ?------?
- // return true
- // }
- // // Neither window is completely open (nil, nil), but one or the other might
- // // still be future- or past-open.
- // if w.start == nil {
- // // w is past-open, future-closed
- // // <------]
- // if x.start != nil && !x.start.Before(*w.end) {
- // // x starts after w ends (or eq)
- // // <------]
- // // [------?
- // return false
- // }
- // // <-----]
- // // ?-----?
- // return true
- // }
- // if w.end == nil {
- // // w is future-open, past-closed
- // // [------>
- // if x.end != nil && !x.end.After(*w.end) {
- // // x ends before w begins (or eq)
- // // [------>
- // // ?------]
- // return false
- // }
- // // [------>
- // // ?------?
- // return true
- // }
- // // Now we know w is closed, but we don't know about x
- // // [------]
- // // ?------?
- // if x.start == nil {
- // // TODO
- // }
- // if x.end == nil {
- // // TODO
- // }
- // // Both are closed.
- // if !x.start.Before(*w.end) && !x.end.Before(*w.end) {
- // // x starts and ends after w ends
- // // [------]
- // // [------]
- // return false
- // }
- // if !x.start.After(*w.start) && !x.end.After(*w.start) {
- // // x starts and ends before w starts
- // // [------]
- // // [------]
- // return false
- // }
- // // w and x must overlap
- // // [------]
- // // [------]
- // return true
- // }
- func (w *Window) Set(start, end *time.Time) {
- w.start = start
- w.end = end
- }
- // Shift adds the given duration to both the start and end times of the window
- func (w Window) Shift(dur time.Duration) Window {
- if w.start != nil {
- s := w.start.Add(dur)
- w.start = &s
- }
- if w.end != nil {
- e := w.end.Add(dur)
- w.end = &e
- }
- return w
- }
- func (w Window) Start() *time.Time {
- return w.start
- }
- func (w Window) String() string {
- if w.start == nil && w.end == nil {
- return "[nil, nil)"
- }
- if w.start == nil {
- return fmt.Sprintf("[nil, %s)", w.end.Format("2006-01-02T15:04:05-0700"))
- }
- if w.end == nil {
- return fmt.Sprintf("[%s, nil)", w.start.Format("2006-01-02T15:04:05-0700"))
- }
- return fmt.Sprintf("[%s, %s)", w.start.Format("2006-01-02T15:04:05-0700"), w.end.Format("2006-01-02T15:04:05-0700"))
- }
- // DurationOffset returns durations representing the duration and offset of the
- // given window
- func (w Window) DurationOffset() (time.Duration, time.Duration, error) {
- if w.IsOpen() || w.IsNegative() {
- return 0, 0, fmt.Errorf("illegal window: %s", w)
- }
- duration := w.Duration()
- offset := time.Since(*w.End())
- return duration, offset, nil
- }
- // DurationOffsetStrings returns formatted, Prometheus-compatible strings representing
- // the duration and offset of the window in terms of days, hours, minutes, or seconds;
- // e.g. ("7d", "1441m", "30m", "1s", "")
- func (w Window) DurationOffsetStrings() (string, string) {
- dur, off, err := w.DurationOffset()
- if err != nil {
- return "", ""
- }
- return timeutil.DurationOffsetStrings(dur, off)
- }
- // GetPercentInWindow Determine pct of item time contained the window.
- // determined by the overlap of the start/end with the given
- // window, which will be negative if there is no overlap. If
- // there is positive overlap, compare it with the total mins.
- //
- // e.g. here are the two possible scenarios as simplidied
- // 10m windows with dashes representing item's time running:
- //
- // 1. item falls entirely within one CloudCostSet window
- // | ---- | | |
- // totalMins = 4.0
- // pct := 4.0 / 4.0 = 1.0 for window 1
- // pct := 0.0 / 4.0 = 0.0 for window 2
- // pct := 0.0 / 4.0 = 0.0 for window 3
- //
- // 2. item overlaps multiple CloudCostSet windows
- // | ----|----------|-- |
- // totalMins = 16.0
- // pct := 4.0 / 16.0 = 0.250 for window 1
- // pct := 10.0 / 16.0 = 0.625 for window 2
- // pct := 2.0 / 16.0 = 0.125 for window 3
- func (w Window) GetPercentInWindow(that Window) float64 {
- if that.IsOpen() {
- log.Errorf("Window: GetPercentInWindow: invalid window %s", that.String())
- return 0
- }
- s := *that.Start()
- if s.Before(*w.Start()) {
- s = *w.Start()
- }
- e := *that.End()
- if e.After(*w.End()) {
- e = *w.End()
- }
- mins := e.Sub(s).Minutes()
- if mins <= 0.0 {
- return 0.0
- }
- totalMins := that.Duration().Minutes()
- pct := mins / totalMins
- return pct
- }
- // GetAccumulateWindow rounds the start and end of the window to the given accumulation option
- func (w Window) GetAccumulateWindow(accumOpt AccumulateOption) (Window, error) {
- if w.IsOpen() {
- return w, fmt.Errorf("could not get accumlate window for open window")
- }
- switch accumOpt {
- case AccumulateOptionAll:
- // just return the entire window
- return w.Clone(), nil
- case AccumulateOptionHour:
- return w.getHourlyWindow(), nil
- case AccumulateOptionDay:
- return w.getDailyWindow(), nil
- case AccumulateOptionWeek:
- return w.getWeeklyWindow(), nil
- case AccumulateOptionMonth:
- return w.getMonthlyWindow(), nil
- case AccumulateOptionQuarter:
- return w.getQuarterlyWindow(), nil
- case AccumulateOptionNone:
- // the default behavior of the app currently is to return the highest resolution steps
- // possible
- fallthrough
- default:
- // if we are here, it means someone wants a window older than what we can query for
- return w, fmt.Errorf("cannot round window to given accumulation option %s", string(accumOpt))
- }
- }
- // GetAccumulateWindows breaks provided window into a []Window with each window having the resolution of the provided AccumulateOption
- func (w Window) GetAccumulateWindows(accumOpt AccumulateOption) ([]Window, error) {
- if w.IsOpen() {
- return nil, fmt.Errorf("could not get accumlate window for open window")
- }
- switch accumOpt {
- case AccumulateOptionAll:
- // just return the entire window
- return []Window{w.Clone()}, nil
- case AccumulateOptionDay:
- wins := w.getDailyWindows()
- return wins, nil
- case AccumulateOptionWeek:
- wins := w.getWeeklyWindows()
- return wins, nil
- case AccumulateOptionMonth:
- wins := w.getMonthlyWindows()
- return wins, nil
- case AccumulateOptionHour:
- // our maximum resolution is hourly
- wins := w.getHourlyWindows()
- return wins, nil
- case AccumulateOptionQuarter:
- wins := w.getQuarterlyWindows()
- return wins, nil
- case AccumulateOptionNone:
- // the default behavior of the app currently is to return the highest resolution steps
- // possible
- fallthrough
- default:
- // if we are here, it means someone wants a window older than what we can query for
- return nil, fmt.Errorf("store does not have coverage window starting at %v", w.Start())
- }
- }
- func (w Window) getHourlyWindow() Window {
- origStart := w.Start()
- origEnd := w.End()
- // round the start and end windows to the calendar hour start and ends, respectively
- roundedStart := time.Date(origStart.Year(), origStart.Month(), origStart.Day(), origStart.Hour(), 0, 0, 0, origStart.Location())
- roundedEnd := time.Date(origEnd.Year(), origEnd.Month(), origEnd.Day(), origEnd.Hour()+1, 0, 0, 0, origEnd.Location())
- // edge case - if user has exactly specified first instant of new hour, does not need rounding
- if origEnd.Minute() == 0 && origEnd.Second() == 0 {
- roundedEnd = *origEnd
- }
- return NewClosedWindow(roundedStart, roundedEnd)
- }
- // getHourlyWindows breaks up a window into hours
- func (w Window) getHourlyWindows() []Window {
- wins := []Window{}
- roundedWindow := w.getHourlyWindow()
- roundedStart := *roundedWindow.Start()
- roundedEnd := *roundedWindow.End()
- currStart := roundedStart
- currEnd := time.Date(currStart.Year(), currStart.Month(), currStart.Day(), currStart.Hour()+1, 0, 0, 0, currStart.Location())
- for currEnd.Before(roundedEnd) || currEnd.Equal(roundedEnd) {
- wins = append(wins, NewClosedWindow(currStart, currEnd))
- currStart = currEnd
- currEnd = time.Date(currEnd.Year(), currEnd.Month(), currEnd.Day(), currEnd.Hour()+1, 0, 0, 0, currStart.Location())
- }
- return wins
- }
- func (w Window) getDailyWindow() Window {
- origStart := w.Start()
- origEnd := w.End()
- // round the start and end windows to the calendar day start and ends, respectively
- roundedStart := time.Date(origStart.Year(), origStart.Month(), origStart.Day(), 0, 0, 0, 0, origStart.Location())
- roundedEnd := time.Date(origEnd.Year(), origEnd.Month(), origEnd.Day()+1, 0, 0, 0, 0, origEnd.Location())
- // edge case - if user has exactly specified first instant of new day, does not need rounding
- if origEnd.Minute() == 0 && origEnd.Second() == 0 && origEnd.Hour() == 0 {
- roundedEnd = *origEnd
- }
- return NewClosedWindow(roundedStart, roundedEnd)
- }
- // getDailyWindows breaks up a window into days
- func (w Window) getDailyWindows() []Window {
- wins := []Window{}
- roundedWindow := w.getDailyWindow()
- roundedStart := *roundedWindow.Start()
- roundedEnd := *roundedWindow.End()
- currStart := roundedStart
- currEnd := time.Date(currStart.Year(), currStart.Month(), currStart.Day()+1, 0, 0, 0, 0, currStart.Location())
- for currEnd.Before(roundedEnd) || currEnd.Equal(roundedEnd) {
- wins = append(wins, NewClosedWindow(currStart, currEnd))
- currStart = currEnd
- currEnd = time.Date(currEnd.Year(), currEnd.Month(), currEnd.Day()+1, 0, 0, 0, 0, currStart.Location())
- }
- return wins
- }
- func (w Window) getWeeklyWindow() Window {
- origStart := w.Start()
- origEnd := w.End()
- // round the start and end windows to the calendar month start and ends, respectively
- roundedStart := origStart.Add(-1 * time.Duration(origStart.Weekday()) * time.Hour * 24)
- roundedStart = time.Date(roundedStart.Year(), roundedStart.Month(), roundedStart.Day(), 0, 0, 0, 0, origEnd.Location())
- roundedEnd := origEnd.Add(time.Duration(6-origEnd.Weekday()) * time.Hour * 24)
- roundedEnd = time.Date(roundedEnd.Year(), roundedEnd.Month(), roundedEnd.Day()+1, 0, 0, 0, 0, origEnd.Location())
- // edge case - if user has exactly specified first instant of new day, does not need rounding
- if origEnd.Weekday() == 0 && origEnd.Second() == 0 && origEnd.Hour() == 0 {
- roundedEnd = *origEnd
- }
- return NewClosedWindow(roundedStart, roundedEnd)
- }
- // getWeeklyWindows breaks up a window into weeks, with weeks starting on Sunday
- func (w Window) getWeeklyWindows() []Window {
- wins := []Window{}
- roundedWindow := w.getWeeklyWindow()
- roundedStart := *roundedWindow.Start()
- roundedEnd := *roundedWindow.End()
- currStart := roundedStart
- currEnd := time.Date(currStart.Year(), currStart.Month(), currStart.Day()+7, 0, 0, 0, 0, currStart.Location())
- for currEnd.Before(roundedEnd) || currEnd.Equal(roundedEnd) {
- wins = append(wins, NewClosedWindow(currStart, currEnd))
- currStart = currEnd
- currEnd = time.Date(currEnd.Year(), currEnd.Month(), currEnd.Day()+7, 0, 0, 0, 0, currStart.Location())
- }
- return wins
- }
- func (w Window) getMonthlyWindow() Window {
- origStart := w.Start()
- origEnd := w.End()
- // round the start and end windows to the calendar month start and ends, respectively
- roundedStart := time.Date(origStart.Year(), origStart.Month(), 1, 0, 0, 0, 0, origStart.Location())
- roundedEnd := time.Date(origEnd.Year(), origEnd.Month()+1, 1, 0, 0, 0, 0, origEnd.Location())
- // edge case - if user has exactly specified first instant of new month, does not need rounding
- if origEnd.Day() == 1 && origEnd.Hour() == 0 && origEnd.Minute() == 0 && origEnd.Second() == 0 {
- roundedEnd = *origEnd
- }
- return NewClosedWindow(roundedStart, roundedEnd)
- }
- // getMonthlyWindows breaks up a window into calendar months
- func (w Window) getMonthlyWindows() []Window {
- wins := []Window{}
- roundedWindow := w.getMonthlyWindow()
- roundedStart := *roundedWindow.Start()
- roundedEnd := *roundedWindow.End()
- currStart := roundedStart
- currEnd := time.Date(currStart.Year(), currStart.Month()+1, 1, 0, 0, 0, 0, currStart.Location())
- for currEnd.Before(roundedEnd) || currEnd.Equal(roundedEnd) {
- wins = append(wins, NewClosedWindow(currStart, currEnd))
- currStart = currEnd
- currEnd = time.Date(currEnd.Year(), currEnd.Month()+1, 1, 0, 0, 0, 0, currStart.Location())
- }
- return wins
- }
- func (w Window) getQuarterlyWindow() Window {
- origStart := w.Start()
- origEnd := w.End()
- // round the start and end windows to the calendar quarter start and ends, respectively
- // get quarter fraction from month
- startQuarterNum := int(math.Ceil(float64(origStart.Month()) / 3.0))
- endQuarterNum := int(math.Ceil(float64(origEnd.Month()) / 3.0))
- roundedStart := time.Date(origStart.Year(), time.Month((startQuarterNum*3)-2), 1, 0, 0, 0, 0, origStart.Location())
- roundedEnd := time.Date(origEnd.Year(), time.Month(((endQuarterNum+1)*3)-2), 1, 0, 0, 0, 0, origEnd.Location())
- // edge case - if user has exactly specified first instant of new quarter, does not need rounding
- if origEnd.Month() == time.Month(((endQuarterNum)*3)-2) && origEnd.Day() == 1 && origEnd.Hour() == 0 && origEnd.Minute() == 0 && origEnd.Second() == 0 {
- roundedEnd = *origEnd
- }
- return NewClosedWindow(roundedStart, roundedEnd)
- }
- // getQuarterlyWindows breaks up a window into calendar months
- func (w Window) getQuarterlyWindows() []Window {
- wins := []Window{}
- roundedWindow := w.getQuarterlyWindow()
- roundedStart := *roundedWindow.Start()
- roundedEnd := *roundedWindow.End()
- currStart := roundedStart
- currEnd := time.Date(currStart.Year(), currStart.Month()+3, 1, 0, 0, 0, 0, currStart.Location())
- for currEnd.Before(roundedEnd) || currEnd.Equal(roundedEnd) {
- wins = append(wins, NewClosedWindow(currStart, currEnd))
- currStart = currEnd
- currEnd = time.Date(currEnd.Year(), currEnd.Month()+3, 1, 0, 0, 0, 0, currStart.Location())
- }
- return wins
- }
- // GetWindows returns a slice of Window with equal size between the given start and end. If windowSize does not evenly
- // divide the period between start and end, the last window is not added
- // Deprecated: in v1.107 use Window.GetWindows() instead
- func GetWindows(start time.Time, end time.Time, windowSize time.Duration) ([]Window, error) {
- // Ensure the range is evenly divisible into windows of the given duration
- dur := end.Sub(start)
- if int(dur.Minutes())%int(windowSize.Minutes()) != 0 {
- return nil, fmt.Errorf("range not divisible by window: [%s, %s] by %s", start, end, windowSize)
- }
- // Ensure that provided times are multiples of the provided windowSize (e.g. midnight for daily windows, on the hour for hourly windows)
- if start != RoundBack(start, windowSize) {
- return nil, fmt.Errorf("provided times are not divisible by provided window: [%s, %s] by %s", start, end, windowSize)
- }
- // Ensure timezones match
- _, sz := start.Zone()
- _, ez := end.Zone()
- if sz != ez {
- return nil, fmt.Errorf("range has mismatched timezones: %s, %s", start, end)
- }
- if sz != int(utcOffset().Seconds()) {
- return nil, fmt.Errorf("range timezone doesn't match configured timezone: expected %s; found %ds", utcOffset(), sz)
- }
- // Build array of windows to cover the CloudCostSetRange
- windows := []Window{}
- s, e := start, start.Add(windowSize)
- for !e.After(end) {
- ws := s
- we := e
- windows = append(windows, NewWindow(&ws, &we))
- s = s.Add(windowSize)
- e = e.Add(windowSize)
- }
- return windows, nil
- }
- // GetWindowsForQueryWindow breaks up a window into an array of windows with a max size of queryWindow
- func GetWindowsForQueryWindow(start time.Time, end time.Time, queryWindow time.Duration) ([]Window, error) {
- // Ensure timezones match
- _, sz := start.Zone()
- _, ez := end.Zone()
- if sz != ez {
- return nil, fmt.Errorf("range has mismatched timezones: %s, %s", start, end)
- }
- if sz != int(utcOffset().Seconds()) {
- return nil, fmt.Errorf("range timezone doesn't match configured timezone: expected %s; found %ds", utcOffset(), sz)
- }
- // Build array of windows to cover the CloudCostSetRange
- windows := []Window{}
- s, e := start, start.Add(queryWindow)
- for s.Before(end) {
- ws := s
- we := e
- windows = append(windows, NewWindow(&ws, &we))
- s = s.Add(queryWindow)
- e = e.Add(queryWindow)
- if e.After(end) {
- e = end
- }
- }
- return windows, nil
- }
- type BoundaryError struct {
- Requested Window
- Supported Window
- Message string
- }
- func NewBoundaryError(req, sup Window, msg string) *BoundaryError {
- return &BoundaryError{
- Requested: req,
- Supported: sup,
- Message: msg,
- }
- }
- func (be *BoundaryError) Error() string {
- if be == nil {
- return "<nil>"
- }
- return fmt.Sprintf("boundary error: requested %s; supported %s: %s", be.Requested, be.Supported, be.Message)
- }
- func (be *BoundaryError) Is(target error) bool {
- if _, ok := target.(*BoundaryError); ok {
- return true
- }
- return false
- }
|