85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
|
package xtime
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"database/sql/driver"
|
||
|
|
"strconv"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Time be used to MySql timestamp converting.
|
||
|
|
type Time int64
|
||
|
|
|
||
|
|
// Scan scan time.
|
||
|
|
func (t *Time) Scan(src interface{}) (err error) {
|
||
|
|
switch sc := src.(type) {
|
||
|
|
case time.Time:
|
||
|
|
if sc.IsZero() {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
*t = Time(sc.Unix())
|
||
|
|
case string:
|
||
|
|
var i int64
|
||
|
|
i, err = strconv.ParseInt(sc, 10, 64)
|
||
|
|
*t = Time(i)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// Value get time value.
|
||
|
|
func (t Time) Value() (driver.Value, error) {
|
||
|
|
return time.Unix(int64(t), 0), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Time get time.
|
||
|
|
func (t Time) Time() time.Time {
|
||
|
|
return time.Unix(int64(t), 0)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *Time) FromDB(bs []byte) error {
|
||
|
|
timeStr := string(bs)
|
||
|
|
ti, err := time.ParseInLocation("2006-01-02T15:04:05", timeStr[:19], time.Local)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if ti.IsZero() {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
*t = Time(ti.Unix())
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t Time) ToDB() ([]byte, error) {
|
||
|
|
unix := time.Unix(int64(t), 0)
|
||
|
|
return []byte(unix.String()), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Duration be used json unmarshal string time, like 1s, 500ms.
|
||
|
|
type Duration time.Duration
|
||
|
|
|
||
|
|
// UnmarshalText unmarshal text to duration.
|
||
|
|
func (d *Duration) UnmarshalText(text []byte) error {
|
||
|
|
tmp, err := time.ParseDuration(string(text))
|
||
|
|
if err == nil {
|
||
|
|
*d = Duration(tmp)
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// UnitTime duration parse to unit, such as "300ms", "1h30m" or "2h10s".
|
||
|
|
func (d *Duration) UnitTime() string {
|
||
|
|
return DurationToUnit(time.Duration(*d))
|
||
|
|
}
|
||
|
|
|
||
|
|
// Shrink will decrease the duration by comparing with context's timeout duration and return new timeout\context\CancelFunc.
|
||
|
|
func (d Duration) Shrink(c context.Context) (Duration, context.Context, context.CancelFunc) {
|
||
|
|
if deadline, ok := c.Deadline(); ok {
|
||
|
|
if ctimeout := time.Until(deadline); ctimeout < time.Duration(d) {
|
||
|
|
// deliver small timeout
|
||
|
|
return Duration(ctimeout), c, func() {}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
ctx, cancel := context.WithTimeout(c, time.Duration(d))
|
||
|
|
return d, ctx, cancel
|
||
|
|
}
|