115 lines
2.2 KiB
Go
115 lines
2.2 KiB
Go
package time
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"strconv"
|
|
"time"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type Duration int64
|
|
|
|
func ParseDuration(s string) (time.Duration, error) {
|
|
if s == "" {
|
|
return 0, errors.New(`time: invalid duration "` + s + `"`)
|
|
}
|
|
|
|
var p int
|
|
var hours int
|
|
loop:
|
|
for i, r := range s {
|
|
switch r {
|
|
case 's', 'm':
|
|
break loop
|
|
case 'h':
|
|
d, err := strconv.Atoi(s[p:i])
|
|
if err != nil {
|
|
return 0, errors.New("time: invalid duration " + s)
|
|
}
|
|
hours += d
|
|
p = i + 1
|
|
case 'd':
|
|
d, err := strconv.Atoi(s[p:i])
|
|
if err != nil {
|
|
return 0, errors.New("time: invalid duration " + s)
|
|
}
|
|
hours += d * 24
|
|
p = i + 1
|
|
case 'y':
|
|
n, err := strconv.Atoi(s[p:i])
|
|
if err != nil {
|
|
return 0, errors.New("time: invalid duration " + s)
|
|
}
|
|
var d int
|
|
for j := n - 1; j >= 0; j-- {
|
|
d += time.Date(time.Now().Year()+j, time.December, 31, 0, 0, 0, 0, time.Local).YearDay()
|
|
}
|
|
hours += d * 24
|
|
p = i + 1
|
|
}
|
|
}
|
|
|
|
return time.ParseDuration(fmt.Sprintf("%dh%s", hours, s[p:]))
|
|
}
|
|
|
|
func (d Duration) MarshalYAML() (interface{}, error) {
|
|
return time.Duration(d).String(), nil
|
|
}
|
|
|
|
func (d *Duration) UnmarshalYAML(n *yaml.Node) error {
|
|
var v interface{}
|
|
if err := yaml.Unmarshal([]byte(n.Value), &v); err != nil {
|
|
return err
|
|
}
|
|
switch value := v.(type) {
|
|
case float64:
|
|
*d = Duration(time.Duration(value))
|
|
return nil
|
|
case string:
|
|
dv, err := ParseDuration(value)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*d = Duration(dv)
|
|
return nil
|
|
default:
|
|
return fmt.Errorf("invalid duration")
|
|
}
|
|
}
|
|
|
|
func (d Duration) MarshalJSON() ([]byte, error) {
|
|
return json.Marshal(time.Duration(d).String())
|
|
}
|
|
|
|
func (d *Duration) UnmarshalJSON(b []byte) error {
|
|
var v interface{}
|
|
if err := json.Unmarshal(b, &v); err != nil {
|
|
return err
|
|
}
|
|
switch value := v.(type) {
|
|
case float64:
|
|
*d = Duration(time.Duration(value))
|
|
return nil
|
|
case string:
|
|
dv, err := ParseDuration(value)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*d = Duration(dv)
|
|
return nil
|
|
default:
|
|
return fmt.Errorf("invalid duration")
|
|
}
|
|
}
|
|
|
|
/*
|
|
func (d Duration) MarshalYAML() (interface{}, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (d Duration) UnmarshalYAML(fn func(interface{}) error) error
|
|
*/
|