diff --git a/util/time/duration.go b/util/time/duration.go new file mode 100644 index 00000000..da10d5f3 --- /dev/null +++ b/util/time/duration.go @@ -0,0 +1,48 @@ +package time + +import ( + "fmt" + "time" +) + +type Duration int64 + +func ParseDuration(s string) (time.Duration, error) { + if s == "" { + return 0, fmt.Errorf(`time: invalid duration "` + s + `"`) + } + + //var sb strings.Builder + /* + for i, r := range s { + switch r { + case 'd': + n, err := strconv.Atoi(s[idx:i]) + if err != nil { + return 0, errors.New("time: invalid duration " + s) + } + s[idx:i] = fmt.Sprintf("%d", n*24) + default: + sb.WriteRune(r) + } + } + */ + var td time.Duration + var err error + switch s[len(s)-1] { + case 's', 'm', 'h': + td, err = time.ParseDuration(s) + case 'd': + if td, err = time.ParseDuration(s[:len(s)-1] + "h"); err == nil { + td *= 24 + } + case 'y': + if td, err = time.ParseDuration(s[:len(s)-1] + "h"); err == nil { + year := time.Date(time.Now().Year(), time.December, 31, 0, 0, 0, 0, time.Local) + days := year.YearDay() + td *= 24 * time.Duration(days) + } + } + + return td, err +} diff --git a/util/time/duration_test.go b/util/time/duration_test.go new file mode 100644 index 00000000..e8c2d6f4 --- /dev/null +++ b/util/time/duration_test.go @@ -0,0 +1,27 @@ +package time + +import ( + "testing" + "time" +) + +func TestParseDuration(t *testing.T) { + var td time.Duration + var err error + + td, err = ParseDuration("14d4h") + if err != nil { + t.Fatalf("ParseDuration error: %v", err) + } + if td.String() != "336h0m0s" { + t.Fatalf("ParseDuration 14d != 336h0m0s : %s", td.String()) + } + + td, err = ParseDuration("1y") + if err != nil { + t.Fatalf("ParseDuration error: %v", err) + } + if td.String() != "8760h0m0s" { + t.Fatalf("ParseDuration 1y != 8760h0m0s : %s", td.String()) + } +}