imitater/execute_at.go
2018-11-27 17:46:29 +08:00

89 lines
1.8 KiB
Go

package imitate
import (
"time"
)
// ExecuteAt 特定的时间任务 接口源自于 IExecute
type ExecuteAt struct {
Year int `yaml:"year"`
Month int `yaml:"month"`
Day int `yaml:"day"`
Hour int `yaml:"hour"`
Min int `yaml:"min"`
Sec int `yaml:"sec"`
TriggerTime int64 // 下次的触发时间点
StartStatus bool // 一个值判断这个时间表是否有效
}
// SetStartStatus 设置执行计划是否生效
func (ea *ExecuteAt) SetStartStatus(status bool) {
ea.StartStatus = status
}
// GetTriggerTime 获取计划的触发时间
func (ea *ExecuteAt) GetTriggerTime() int64 {
if ea.StartStatus {
return ea.TriggerTime
}
return -1
}
// TimeTo 是否到了该触发的时间
func (ea *ExecuteAt) TimeTo() int64 {
return time.Now().Unix() - ea.TriggerTime
}
// GetStartStatus 获取计划的触发时间是否在生效
func (ea *ExecuteAt) GetStartStatus() bool {
return ea.StartStatus
}
// CalculateTrigger 计算触发特定时间任务的时间点 执行后 可以通过GetTriggerTime确认触发时间
func (ea *ExecuteAt) CalculateTrigger() {
now := time.Now()
year := ea.Year
if ea.Year <= 0 {
year = now.Year()
}
month := time.Month(ea.Month)
if ea.Month <= 0 {
month = now.Month()
}
day := ea.Day
if ea.Day <= 0 {
day = now.Day()
}
hour := ea.Hour
if ea.Hour < 0 {
hour = now.Hour()
}
min := ea.Min
if ea.Min < 0 {
min = now.Minute()
}
sec := ea.Sec
if ea.Sec < 0 {
sec = now.Second()
}
ea.TriggerTime = time.Date(year, time.Month(month), day, hour, min, sec, 0, time.Local).Unix()
}
// FromValues 从数值 里获取执行时间表的结构
func (ea *ExecuteAt) FromValues(year int, month int, day int, hour int, min int, sec int) {
ea.Year = year
ea.Month = month
ea.Day = day
ea.Hour = hour
ea.Min = min
ea.Sec = sec
}