Add initial version of time duration support in sign requests.
parent
7b1f4a43cf
commit
00fed1c538
@ -0,0 +1,109 @@
|
||||
package provisioner
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"time"
|
||||
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
// TimeDuration is a type that represents a time but the JSON unmarshaling can
|
||||
// use a time using the RFC 3339 format or a time.Duration string. If a duration
|
||||
// is used, the time will be set on the first call to TimeDuration.Time.
|
||||
type TimeDuration struct {
|
||||
t time.Time
|
||||
d time.Duration
|
||||
}
|
||||
|
||||
// ParseTimeDuration returns a new TimeDuration parsing the RFC 3339 time or
|
||||
// time.Duration string.
|
||||
func ParseTimeDuration(s string) (TimeDuration, error) {
|
||||
if s == "" {
|
||||
return TimeDuration{}, nil
|
||||
}
|
||||
|
||||
// Try to use the unquoted RFC 3339 format
|
||||
var t time.Time
|
||||
if err := t.UnmarshalText([]byte(s)); err == nil {
|
||||
return TimeDuration{t: t}, nil
|
||||
}
|
||||
|
||||
// Try to use the time.Duration string format
|
||||
if d, err := time.ParseDuration(s); err == nil {
|
||||
return TimeDuration{d: d}, nil
|
||||
}
|
||||
|
||||
return TimeDuration{}, errors.Errorf("failed to parse %s", s)
|
||||
}
|
||||
|
||||
// SetDuration initializes the TimeDuration with the given duration string. If
|
||||
// the time was set it will re-set to zero.
|
||||
func (t *TimeDuration) SetDuration(d time.Duration) {
|
||||
t.t, t.d = time.Time{}, d
|
||||
}
|
||||
|
||||
// SetTime initializes the TimeDuration with the given time. If the duration is
|
||||
// set it will be re-set to zero.
|
||||
func (t *TimeDuration) SetTime(tt time.Time) {
|
||||
t.t, t.d = tt, 0
|
||||
}
|
||||
|
||||
// MarshalJSON implements the json.Marshaler interface. If the time is set it
|
||||
// will return the time in RFC 3339 format if not it will return the duration
|
||||
// string.
|
||||
func (t *TimeDuration) MarshalJSON() ([]byte, error) {
|
||||
switch {
|
||||
case t == nil:
|
||||
return []byte("null"), nil
|
||||
case t.t.IsZero():
|
||||
if t.d == 0 {
|
||||
return []byte("null"), nil
|
||||
}
|
||||
return json.Marshal(t.d.String())
|
||||
default:
|
||||
return t.t.MarshalJSON()
|
||||
}
|
||||
}
|
||||
|
||||
// UnmarshalJSON implements the json.Unmarshaler interface. The time is expected
|
||||
// to be a quoted string in RFC 3339 format or a quoted time.Duration string.
|
||||
func (t *TimeDuration) UnmarshalJSON(data []byte) error {
|
||||
var s string
|
||||
if err := json.Unmarshal(data, &s); err != nil {
|
||||
return errors.Wrapf(err, "error unmarshaling %s", data)
|
||||
}
|
||||
|
||||
// Empty TimeDuration
|
||||
if s == "" {
|
||||
*t = TimeDuration{}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Try to use the unquoted RFC 3339 format
|
||||
var tt time.Time
|
||||
if err := tt.UnmarshalText([]byte(s)); err == nil {
|
||||
*t = TimeDuration{t: tt}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Try to use the time.Duration string format
|
||||
if d, err := time.ParseDuration(s); err == nil {
|
||||
*t = TimeDuration{d: d}
|
||||
return nil
|
||||
}
|
||||
|
||||
return errors.Errorf("failed to parse %s", data)
|
||||
}
|
||||
|
||||
// Time set once the embedded time and returns it.
|
||||
func (t *TimeDuration) Time() time.Time {
|
||||
switch {
|
||||
case t == nil:
|
||||
return time.Time{}
|
||||
case t.t.IsZero():
|
||||
t.t = time.Now().UTC().Add(t.d)
|
||||
return t.t
|
||||
default:
|
||||
return t.t
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue