时间轮实现定时任务

用时间轮来实现定时器时，需要定义每一个格子的“刻度”，可以将时间想象成一个时钟，中心有秒针顺时针转动，每次转动到一个刻度时，需要去查看该刻度挂载的任务列表是否已经有到期的任务。

从结构上来讲，时间轮和哈希表很相似，如果把哈希算法定义为：触发时间 % 时间轮元素大小，那么这就是一个简单的哈希表。在哈希冲突时，采用链表挂载哈希冲突的定时器。

除了这种单层时间轮，还有一些时间轮采用多层实现。

代码实现

package main

import (
 "github.com/antlabs/timer"
 "log"
 "sync"
 "time"
)

// 一次性定时器
func after(tm timer.Timer) {
 var wg sync.WaitGroup
 wg.Add(1)
 defer wg.Wait()

 go func() {
  defer wg.Done()
  tm.AfterFunc(1*time.Second, func() {
   log.Printf("after 1 secondn")
  })
 }()
}

// 周期性定时器
func schedule(tm timer.Timer) {
 tm.ScheduleFunc(500*time.Millisecond, func() {
  log.Printf("schedule 500 millisecondsn")
 })

 tm.ScheduleFunc(time.Second, func() {
  log.Printf("schedule 1 secondn")
 })
}

func main() {
 tm := timer.NewTimer()
 defer tm.Stop()

 // 启动一次性任务
 go after(tm)
 // 启动定时任务
 go schedule(tm)

 go func() {
  time.Sleep(2*time.Minute + 50*time.Second)
  tm.Stop()
 }()

 tm.Run()
}

打印

2022/05/02 22:27:43 schedule 500 milliseconds
2022/05/02 22:27:43 schedule 1 second
2022/05/02 22:27:43 schedule 500 milliseconds
2022/05/02 22:27:43 after 1 second
2022/05/02 22:27:44 schedule 500 milliseconds
2022/05/02 22:27:44 schedule 500 milliseconds
2022/05/02 22:27:44 schedule 1 second
2022/05/02 22:27:45 schedule 500 milliseconds
2022/05/02 22:27:45 schedule 500 milliseconds
2022/05/02 22:27:45 schedule 1 second
2022/05/02 22:27:46 schedule 500 milliseconds
2022/05/02 22:27:46 schedule 500 milliseconds
2022/05/02 22:27:46 schedule 1 second
2022/05/02 22:27:47 schedule 500 milliseconds

...