golang 高性能定時器之最小堆實現

    業務中有不少定時任務,在規定時間內,無論是否完成都須要回調,明顯,這種須要實現定時器,比較好的是時間輪和最小堆。這裏介紹最小堆實現,這裏就是個變相的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

相關文章
相關標籤/搜索