122 lines
2.8 KiB
Go
122 lines
2.8 KiB
Go
package global
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"fmt"
|
|
"math"
|
|
"math/rand"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// DateTime 自定义事件类型
|
|
type DateTime time.Time
|
|
|
|
// 日期格式
|
|
const (
|
|
DateLayout = "2006-01-02"
|
|
DateTimeLayout = "2006-01-02 15:04:05"
|
|
TimeLayout = "15:04:05"
|
|
BuildTimeLayout = "2006.0102.150405"
|
|
TimestampLayout = "20060102150405"
|
|
)
|
|
|
|
var StartTime = time.Now()
|
|
|
|
func (dt *DateTime) UnmarshalJSON(data []byte) (err error) {
|
|
value := strings.Trim(string(data), "\"")
|
|
now, err := time.ParseInLocation(DateTimeLayout, value, time.Local)
|
|
*dt = DateTime(now)
|
|
return
|
|
}
|
|
|
|
func (dt DateTime) MarshalJSON() ([]byte, error) {
|
|
b := make([]byte, 0, len(DateTimeLayout)+2)
|
|
b = append(b, '"')
|
|
b = time.Time(dt).AppendFormat(b, DateTimeLayout)
|
|
b = append(b, '"')
|
|
return b, nil
|
|
}
|
|
|
|
func (dt DateTime) Value() (driver.Value, error) {
|
|
var zeroTime time.Time
|
|
ti := time.Time(dt)
|
|
if ti.UnixNano() == zeroTime.UnixNano() {
|
|
return nil, nil
|
|
}
|
|
return ti, nil
|
|
}
|
|
|
|
func (dt *DateTime) Scan(v interface{}) error {
|
|
if value, ok := v.(time.Time); ok {
|
|
*dt = DateTime(value)
|
|
return nil
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (dt DateTime) String() string {
|
|
return time.Time(dt).Format(DateTimeLayout)
|
|
}
|
|
|
|
func UpTime() time.Duration {
|
|
return time.Since(StartTime)
|
|
}
|
|
|
|
func UpTimeString() string {
|
|
d := UpTime()
|
|
days := d / (time.Hour * 24)
|
|
d -= days * 24 * time.Hour
|
|
hours := d / time.Hour
|
|
d -= hours * time.Hour
|
|
minutes := d / time.Minute
|
|
d -= minutes * time.Minute
|
|
seconds := d / time.Second
|
|
return fmt.Sprintf("%d Days %d Hours %d Mins %d Secs", days, hours, minutes, seconds)
|
|
}
|
|
|
|
// 获取时区
|
|
func GetLocation() (*time.Location, error) {
|
|
location, err := time.LoadLocation("Local")
|
|
if err != nil {
|
|
return location, nil
|
|
}
|
|
return location, nil
|
|
}
|
|
|
|
func GetDuration(t int64, v string) time.Duration {
|
|
switch v {
|
|
case "h":
|
|
return time.Hour * time.Duration(t)
|
|
case "m":
|
|
return time.Minute * time.Duration(t)
|
|
case "s":
|
|
return time.Second * time.Duration(t)
|
|
default:
|
|
return time.Minute * time.Duration(t)
|
|
}
|
|
}
|
|
|
|
func SleepRandomDuration() {
|
|
ns := int64(5) * 1000000000
|
|
// 以当前时间为随机数种子，如果所有 log-agent-updater 在同一时间启动，系统时间是相同的，那么随机种子就是一样的
|
|
// 问题不大，批量ssh去启动 log-agent-updater 的话也是一个顺次的过程
|
|
r := rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
d := time.Duration(r.Int63n(ns)) * time.Nanosecond
|
|
time.Sleep(d)
|
|
}
|
|
|
|
// OverTimeNow 超时比较
|
|
func OverTimeNow(compareTime, subTime time.Time, sub int64) (bool, int64, error) {
|
|
location, err := GetLocation()
|
|
if err != nil {
|
|
return false, 0, err
|
|
}
|
|
subT := compareTime.In(location).Sub(subTime)
|
|
abs := int64(math.Abs(math.Ceil(subT.Minutes())))
|
|
if abs > sub {
|
|
return true, abs, nil
|
|
}
|
|
return false, abs, nil
|
|
}
|