溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點(diǎn)擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》

用golang實(shí)現(xiàn)一個(gè)定時(shí)器任務(wù)隊(duì)列實(shí)例

發(fā)布時(shí)間:2020-10-12 20:32:06 來源:腳本之家 閱讀:274 作者:o0無憂亦無怖 欄目:編程語言

 很有幸得到公司信任,采用新的語言進(jìn)行一些底層服務(wù)的開發(fā),在實(shí)現(xiàn)功能的同時(shí),也獲得了一些感悟,因此在這記錄一下,方便自己查看也可以共享給大家。

golang中定時(shí)器

golang中提供了2種定時(shí)器timer和ticker(如果JS很熟悉的話應(yīng)該會很了解),分別是一次性定時(shí)器和重復(fù)任務(wù)定時(shí)器。

一般用法:

func main() { 
 input := make(chan interface{}) 
 //producer - produce the messages
 go func() {
  for i := 0; i < 5; i++ {
   input <- i
  }
  input <- "hello, world"
 }()
 
 t1 := time.NewTimer(time.Second * 5)
 t2 := time.NewTimer(time.Second * 10)
 
 for {
  select {
  //consumer - consume the messages
  case msg := <-input:
   fmt.Println(msg)
 
  case <-t1.C:
   println("5s timer")
   t1.Reset(time.Second * 5)
 
  case <-t2.C:
   println("10s timer")
   t2.Reset(time.Second * 10)
  }
 }
}

源碼觀察

這個(gè)C是啥,我們?nèi)ピ创a看看,以timer為例:

type Timer struct {
 C <-chan Time
 r runtimeTimer
}

原來是一個(gè)channel,其實(shí)有GO基礎(chǔ)的都知道,GO的運(yùn)算符當(dāng)出現(xiàn)的->或者<-的時(shí)候,必然是有一端是指channel。按照上面的例子來看,就是阻塞在一個(gè)for循環(huán)內(nèi),等待到了定時(shí)器的C從channel出來,當(dāng)獲取到值的時(shí)候,進(jìn)行想要的操作。

設(shè)計(jì)我們的定時(shí)任務(wù)隊(duì)列

我的需求

當(dāng)時(shí)我的需求是這樣,我需要接收到客戶端的請求并產(chǎn)生一個(gè)定時(shí)任務(wù),會在固定時(shí)間執(zhí)行,可能是一次,也可能是多次,也可能到指定時(shí)間自動(dòng)停止,可能當(dāng)任務(wù)終止的時(shí)候,我還要能停止掉。

具體我畫了個(gè)流程圖,差不多如下,畫圖水平有限,請見諒。

用golang實(shí)現(xiàn)一個(gè)定時(shí)器任務(wù)隊(duì)列實(shí)例

定義結(jié)構(gòu)

type OnceCron struct {
 tasks []*Task   //任務(wù)的列隊(duì)
 add chan *Task  //當(dāng)遭遇到新任務(wù)的時(shí)候
 remove chan string  //當(dāng)遭遇到刪除任務(wù)的時(shí)候
 stop chan struct{}  //當(dāng)遇到停止信號的時(shí)候
 Logger *log.Logger  //日志 
}
type Job interface {
 Run()     //執(zhí)行接口
}
type Task struct {
  Job  Job   //要執(zhí)行的任務(wù) 
 Uuid string   //任務(wù)標(biāo)識,刪除時(shí)用
 RunTime int64   //執(zhí)行時(shí)間
 Spacing int64   //間隔時(shí)間
 EndTime int64   //結(jié)束時(shí)間
 Number int    //總共要次數(shù)
}

隊(duì)列實(shí)現(xiàn)

首先,我們要獲得一個(gè)隊(duì)列任務(wù)

func NewCron() *OnceCron 常規(guī)操作,為了節(jié)省篇幅,我就不寫出來,具體可以看源碼,貼在了底部。

然后,開始定時(shí)器隊(duì)列的運(yùn)行,一般,都會命名為Start。那么就有一個(gè)問題,我們剛開始啟動(dòng)程序的時(shí)候,這個(gè)時(shí)候是沒有任務(wù)隊(duì)列,那豈不是for{ select{}}在等待個(gè)毛毛球?所以,我們需要在Start的時(shí)候添加一個(gè)默認(rèn)的任務(wù), 我是這么做的,添加了一個(gè)一小時(shí)執(zhí)行一次的重復(fù)隊(duì)列,防止隊(duì)列退出。

func (one *OnceCron) Start() {
 //初始化的時(shí)候加入一個(gè)一年的長定時(shí)器,間隔1小時(shí)執(zhí)行一次
 task := getTaskWithFuncSpacing(3600, time.Now().Add(time.Hour*24*365).Unix() , func() {
 log.Println("It's a Hour timer!")
 }) //為了代碼格式markdown 里面有個(gè)括號我改成全角了
 one.tasks = append(one.tasks, task)
 go one.run() //協(xié)成執(zhí)行 防止主進(jìn)程被阻塞
}

執(zhí)行部分應(yīng)該是重點(diǎn)的,我的理解是,分成三部:

  1. 首先獲得一個(gè)最先執(zhí)行的任務(wù)
  2. 然后產(chǎn)生一個(gè)定時(shí)器,用于執(zhí)行任務(wù)
  3. 進(jìn)行阻塞判斷,獲取我們要進(jìn)行的操作
func (one *OnceCron) run() {

 for {
    //第一步 獲取任務(wù)
 now := time.Now() //獲取到當(dāng)前時(shí)間
 task, key := one.GetTask() //獲取最近的一個(gè)任務(wù)的執(zhí)行時(shí)間
 i64 := task.RunTime - now.Unix() //任務(wù)執(zhí)行和當(dāng)前時(shí)間的差

 var d time.Duration
 if i64 < 0 { //如果任務(wù)時(shí)間已過期,將執(zhí)行時(shí)間改成現(xiàn)在并且利馬執(zhí)行
 one.tasks[key].RunTime = now.Unix() 
 one.doAndReset(key)
      continue
 } else { //否則,獲取距離執(zhí)行開始的間隔時(shí)間
 d = time.Unix(task.RunTime, 0).Sub(now)
 }
    //第二步 產(chǎn)生定時(shí)器
 timer := time.NewTimer(d) 

 //第三步 捕獲定時(shí)器或者其他事件
 for {
 select { 
      //當(dāng)定時(shí)器到了執(zhí)行時(shí)間時(shí),執(zhí)行當(dāng)前任務(wù)并關(guān)閉定時(shí)器
 case <-timer.C:
 one.doAndReset(key)
 if task != nil {
  go task.Job.Run()
  timer.Stop()
 }

 //當(dāng)外部添加了任務(wù)時(shí),關(guān)閉當(dāng)前定時(shí)器
 case <-one.add:
 timer.Stop()
 //當(dāng)外部要?jiǎng)h除一個(gè)任務(wù)時(shí),刪除ID為uuidstr的任務(wù)
 case uuidstr := <-one.remove:
 one.removeTask(uuidstr)
 timer.Stop()
 //當(dāng)遇到要關(guān)閉整個(gè)定時(shí)器任務(wù)時(shí)
 case <-one.stop:
 timer.Stop()
 return
 }

 break
 }
 }
}

后記

這個(gè)文章純粹為筆記分析類的文章,旨在分析我碰到一個(gè)需求是如何通過分析過程來產(chǎn)生我們需要的代碼的。

源碼地址:timing 一個(gè)任務(wù)隊(duì)列

應(yīng)用地址:一個(gè)應(yīng)用于谷歌消息推送的轉(zhuǎn)發(fā)中間件

參考源碼:GOLANG實(shí)現(xiàn)crontab功能

以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持億速云。

向AI問一下細(xì)節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI