49 lines
937 B
Go
49 lines
937 B
Go
|
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
|
||
|
|
}
|