Golang 实现简单的定时器

Go 2020-02-01

 
 
  
  问题

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

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

  
  间隔执行

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

  
  
  
  Ticker会在每隔一段时间执行，比如上面的例子中，每隔5秒打印一下当前时间。
  
  但是，这显然满足不了我们的需求，我们还需要在某个固定时刻才开始。
  
  最终思路
  
  这里提供一种比较简单的思路。对于固定时刻T，计算T和当前时间的时间差，然后sleep到T，然后用Tikcer开始定时任务，每隔时间间隔D执行一次任务。
  
  以什么样的形式提供固定时刻T以及时间间隔D
  
  一个首要原则就是越简单越好，最好提供一个原生的包就能解析的。
  
  在time包中，解析时间有ParseInLocation，解析时间间隔有ParseDuration。那么我们就可以用这两个方法能解析的形式来表达T和D。
  
  T的格式
  
  time包预定义了一些格式。
  
  
  
   
   这里吐槽一下，一开始不明白为什么是2006-01-02 15:04:05这个点，一直在想那个点到底发生了什么重大的事。后来才知道，按照美式的时间格式，也就是上面的ANSIC，月，日，时，分，秒，年，排列起来正好是123456。这么设计是为了方便记忆。。。 
  
  
  当然也可以自定义时间格式，比如。一般在我们的定时任务中，我们常用的是时分秒这样的时刻，所以T得表达方式就定义为：
  
  
  
  D的格式
  
  D的格式比较简单，可以使用这种格式。详情见ParseDuration的函数说明。
  
  代码
  
  
  
  结语
  
  我只想说，感谢goroutine。
 
 
作者：Syfun链接：http://www.jianshu.com/p/dd0536797a8e來源：简书著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。ticker := time.NewTicker(5 * time.Second)
for _ = range ticker.C {
    fmt.Println(time.Now())
}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:0515:04:05300ms,-1.5h,2h45m// 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",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)
        }
    }()
}