Golang 实现简单的定时器

问题

做项目的时候经常会有这样的需求，在某个时刻开始执行某个任务，然后每隔一段时间都会执行该任务。

windows下有计划任务，linux下有cron。如果用python可以使用apscheduler库。那么在Go中应该怎么实现呢？

间隔执行

time包中有个Ticker可以用来实现简单的定时任务。

ticker := time.NewTicker(5 * time.Second)
for _ = range ticker.C {
fmt.Println(time.Now())
}

Ticker会在每隔一段时间执行，比如上面的例子中，每隔5秒打印一下当前时间。
但是，这显然满足不了我们的需求，我们还需要在某个固定时刻才开始。

最终思路

这里提供一种比较简单的思路。对于固定时刻T，计算T和当前时间的时间差，然后sleep到T，然后用Tikcer开始定时任务，每隔时间间隔D执行一次任务。

以什么样的形式提供固定时刻T以及时间间隔D

一个首要原则就是越简单越好，最好提供一个原生的包就能解析的。
在time包中，解析时间有ParseInLocation，解析时间间隔有ParseDuration。那么我们就可以用这两个方法能解析的形式来表达T和D。

T的格式

time包预定义了一些格式。

const (
ANSIC       = "Mon Jan _2 15:04:05 2006"
UnixDate    = "Mon Jan _2 15:04:05 MST 2006"
RubyDate    = "Mon Jan 02 15:04:05 -0700 2006"
RFC822      = "02 Jan 06 15:04 MST"
RFC822Z     = "02 Jan 06 15:04 -0700" // RFC822 with numeric zone
RFC850      = "Monday, 02-Jan-06 15:04:05 MST"
RFC1123     = "Mon, 02 Jan 2006 15:04:05 MST"
RFC1123Z    = "Mon, 02 Jan 2006 15:04:05 -0700" // RFC1123 with numeric zone
RFC3339     = "2006-01-02T15:04:05Z07:00"
RFC3339Nano = "2006-01-02T15:04:05.999999999Z07:00"
Kitchen     = "3:04PM"
// Handy time stamps.
Stamp      = "Jan _2 15:04:05"
StampMilli = "Jan _2 15:04:05.000"
StampMicro = "Jan _2 15:04:05.000000"
StampNano  = "Jan _2 15:04:05.000000000"
)

这里吐槽一下，一开始不明白为什么是2006-01-02 15:04:05这个点，一直在想那个点到底发生了什么重大的事。后来才知道，按照美式的时间格式，也就是上面的ANSIC，月，日，时，分，秒，年，排列起来正好是123456。这么设计是为了方便记忆。。。
当然也可以自定义时间格式，比如

2006-01-02 15:04:05

。一般在我们的定时任务中，我们常用的是时分秒这样的时刻，所以T得表达方式就定义为：

15:04:05

D的格式

D的格式比较简单，可以使用

300ms, -1.5h, 2h45m

这种格式。详情见ParseDuration的函数说明。

代码

// sched to start scheduler job at start time by interval duration.
func sched(jobFunc interface{}, start, interval string, jobArgs ...interface{}) {
jobValue := reflect.ValueOf(jobFunc)
if jobValue.Kind() != reflect.Func {
log.Panic("only function can be schedule.")
}
if len(jobArgs) != jobValue.Type().NumIn() {
log.Panic("The number of args valid.")
}
// Get job function args.
in := make([]reflect.Value, len(jobArgs))
for i, arg := range jobArgs {
in[i] = reflect.ValueOf(arg)
}

// Get interval d.
d, err := time.ParseDuration(interval)
if err != nil {
log.Panic(err)
}
location, err := time.LoadLocation("Asia/Shanghai")
if err != nil {
log.Panic(err)
}
t, err := time.ParseInLocation("15:04:05", start, location)
if err != nil {
log.Panic(err)
}
now := time.Now()

// Start time.
t = time.Date(now.Year(), now.Month(), now.Day(), t.Hour(), t.Minute(), t.Second(), 0, location)

if now.After(t) {
t = t.Add((now.Sub(t)/d + 1) * d)
}

time.Sleep(t.Sub(now))
go jobValue.Call(in)
ticker := time.NewTicker(d)
go func() {
for _ = range ticker.C {
go jobValue.Call(in)
}
}()
}

结语

我只想说，感谢goroutine。

作者：Syfun
链接：http://www.jianshu.com/p/dd0536797a8e
來源：简书
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。