业务中有不少定时任务,在规定时间内,无论是否完成都须要回调,明显,这种须要实现定时器,比较好的是时间轮和最小堆。这里介绍最小堆实现,这里就是个变相的topN 问题。 git
该文章后续仍在不断的更新修改中, 请移步到原文地址http://www.dmwan.cc/?p=146github
因为是项目中要使用,不能仅考虑最小堆就完事,须要添加几条特性,第一个是要用一个timer 实现计时功能,第二个是要有提早删除功能。特别提下第二点,在高并发的状况下,timer 不能让任务自动过时,业务成功的状况下,要能提早删除,不然,timer压力会愈来愈大;并发
项目github地址:https://github.com/caucy/timeloop。异步
调用示例:高并发
package main import ( "os" "fmt" "time" "os/signal" "strconv" "syscall" "timeloop/timer" ) var ( timerCtl *timer.TimerHeapHandler ) func init() { timerCtl = timer.New(10, 1000) } type timerHandler struct { } func AddTimerTask(dueInterval int, taskId string) { timerCtl.AddFuncWithId(time.Duration(dueInterval)*time.Second, taskId, func() { fmt.Printf("taskid is %v, time Duration is %v", taskId, dueInterval ) }) } func (t *timerHandler) StartLoop() { timerCtl.StartTimerLoop(timer.MIN_TIMER) // 扫描的间隔时间 eq cpu hz/tick } func main() { sigs := make(chan os.Signal, 1) signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM) timerEntry := timerHandler{} timerEntry.StartLoop() num := 5000 interval := 1000 * time.Millisecond go func (){ for i := 0; i < num; i++ { taskId := strconv.Itoa(i) timerCtl.AddFuncWithId(10 *interval, taskId, func() { fmt.Printf("taskid is %v, time Duration is %v \n", taskId, interval ) }) time.Sleep(100 * time.Millisecond) } }() <- sigs }
实现的主要思路就是,维护个最小堆,每次push,pop 的时候调整堆生成最小堆,一个时钟,每隔一段时间去pop 全部超时的任务,放入异步消费队列中去。 在高频的超时任务管理中,这种类库仍是颇有用的。 oop