Go
語言的標準庫裡提供兩種型別的計時器Timer
和Ticker
。Timer
經過指定的duration
時間後被觸發,往自己的時間channel
傳送當前時間,此後Timer
不再計時。Ticker
則是每隔duration
時間都會把當前時間點傳送給自己的時間channel
,利用計時器的時間channel
可以實現很多與計時相關的功能。
文章主要涉及如下內容:
-
Timer
和Ticker
計時器的內部結構表示 -
Timer
和Ticker
的使用方法和注意事項 - 如何正確
Reset
定時器
計時器的內部表示
兩種計時器都是基於Go
語言的執行時計時器runtime.timer
實現的,rumtime.timer
的結構體表示如下:
type timer struct {
pp puintptr
when int64
period int64
f func(interface{}, uintptr)
arg interface{}
seq uintptr
nextwhen int64
status uint32
}
rumtime.timer
結構體中的欄位含義是
-
when
— 當前計時器被喚醒的時間; -
period
— 兩次被喚醒的間隔; -
f
— 每當計時器被喚醒時都會呼叫的函式; -
arg
— 計時器被喚醒時呼叫f
傳入的引數; -
nextWhen
— 計時器處於timerModifiedLater/timerModifiedEairlier
狀態時,用於設定when
欄位; -
status
— 計時器的狀態;
這裡的 runtime.timer
只是私有的計時器執行時表示,對外暴露的計時器 time.Timer
和time.Ticker
的結構體表示如下:
type Timer struct {
C <-chan Time
r runtimeTimer
}
type Ticker struct {
C <-chan Time
r runtimeTimer
}
Timer.C
和Ticker.C
就是計時器中的時間channel
,接下來我們看一下怎麼使用這兩種計時器,以及使用時要注意的地方。
Timer計時器
time.Timer
計時器必須通過 time.NewTimer
、time.AfterFunc
或者 time.After
函式建立。 當計時器失效時,失效的時間就會被髮送給計時器持有的 channel
,訂閱 channel
的 goroutine
會收到計時器失效的時間。
通過定時器Timer
使用者可以定義自己的超時邏輯,尤其是在應對使用select
處理多個channel
的超時、單channel
讀寫的超時等情形時尤為方便。Timer
常見的使用方法如下:
//使用time.AfterFunc:
t := time.AfterFunc(d, f)
//使用time.After:
select {
case m := <-c:
handle(m)
case <-time.After(5 * time.Minute):
fmt.Println("timed out")
}
// 使用time.NewTimer:
t := time.NewTimer(5 * time.Minute)
select {
case m := <-c:
handle(m)
case <-t.C:
fmt.Println("timed out")
}
time.AfterFunc
這種方式建立的Timer
,在到達超時時間後會在單獨的goroutine
裡執行函式f
。
func AfterFunc(d Duration, f func()) *Timer {
t := &Timer{
r: runtimeTimer{
when: when(d),
f: goFunc,
arg: f,
},
}
startTimer(&t.r)
return t
}
func goFunc(arg interface{}, seq uintptr) {
go arg.(func())()
}
從上面AfterFunc
的原始碼可以看到外面傳入的f
引數並非直接賦值給了執行時計時器的f
,而是作為包裝函式goFunc
的引數傳入的。goFunc
會啟動了一個新的goroutine
來執行外部傳入的函式f
。這是因為所有計時器的事件函式都是由Go
執行時內唯一的 goroutine
timerproc
執行的。為了不阻塞timerproc
的執行,必須啟動一個新的goroutine
執行到期的事件函式。
對於NewTimer
和After
這兩種建立方法,則是Timer
在超時後,執行一個標準庫中內建的函式:sendTime
。
func NewTimer(d Duration) *Timer {
c := make(chan Time, 1)
t := &Timer{
C: c,
r: runtimeTimer{
when: when(d),
f: sendTime,
arg: c,
},
}
startTimer(&t.r)
return t
}
func sendTime(c interface{}, seq uintptr) {
select {
case c.(chan Time) <- Now():
default:
}
}
sendTime
將當前時間傳送到Timer
的時間channel
中。那麼這個動作不會阻塞timerproc
的執行麼?答案是不會,原因是NewTimer
建立的是一個帶緩衝的channel
所以無論Timer.C
這個channel
有沒有接收方sendTime
都可以非阻塞的將當前時間傳送給Timer.C
,而且sendTime
中還加了雙保險:通過select
判斷Timer.C
的Buffer
是否已滿,一旦滿了,會直接退出,依然不會阻塞。
Timer
的Stop
方法可以阻止計時器觸發,呼叫Stop
方法成功停止了計時器的觸發將會返回true
,如果計時器已經過期了或者已經被Stop
停止過了,再次呼叫Stop
方法將會返回false
。
Go
執行時將所有計時器維護在一個最小堆Min Heap
中,Stop
一個計時器就是從堆中刪除該計時器。
Ticker計時器
Ticker
可以週期性地觸發時間事件,每次到達指定的時間間隔後都會觸發事件。
time.Ticker
需要通過time.NewTicker
或者time.Tick
建立。
// 使用time.Tick:
go func() {
for t := range time.Tick(time.Minute) {
fmt.Println("Tick at", t)
}
}()
// 使用time.Ticker
var ticker *time.Ticker = time.NewTicker(1 * time.Second)
go func() {
for t := range ticker.C {
fmt.Println("Tick at", t)
}
}()
time.Sleep(time.Second * 5)
ticker.Stop()
fmt.Println("Ticker stopped")
不過time.Tick
很少會被用到,除非你想在程式的整個生命週期裡都使用time.Ticker
的時間channel
。官文文件裡對time.Tick
的描述是:
time.Tick
底層的Ticker
不能被垃圾收集器恢復;
所以使用time.Tick
時一定要小心,為避免意外儘量使用time.NewTicker
返回的Ticker
替代。
NewTicker
建立的計時器與NewTimer
建立的計時器持有的時間channel
一樣都是帶一個快取的channel
,每次觸發後執行的函式也是sendTime
,這樣即保證了無論有誤接收方Ticker
觸發時間事件時都不會阻塞:
func NewTicker(d Duration) *Ticker {
if d <= 0 {
panic(errors.New("non-positive interval for NewTicker"))
}
// Give the channel a 1-element time buffer.
// If the client falls behind while reading, we drop ticks
// on the floor until the client catches up.
c := make(chan Time, 1)
t := &Ticker{
C: c,
r: runtimeTimer{
when: when(d),
period: int64(d),
f: sendTime,
arg: c,
},
}
startTimer(&t.r)
return t
}
Reset計時器時要注意的問題
關於Reset
的使用建議,文件裡的描述是:
重置計時器時必須注意不要與當前計時器到期傳送時間到t.C的操作產生競爭。如果程式已經從t.C接收到值,則計時器是已知的已過期,並且t.Reset可以直接使用。如果程式尚未從t.C接收值,計時器必須先被停止,並且-如果使用t.Stop時報告計時器已過期,那麼請排空其通道中值。
例如:
if !t.Stop() { <-t.C } t.Reset(d)
下面的例子裡producer goroutine
裡每一秒向通道中傳送一個false
值,迴圈結束後等待一秒再往通道里傳送一個true
值。在consumer goroutine
裡通過迴圈試圖從通道中讀取值,用計時器設定了最長等待時間為5秒,如果計時器超時了,輸出當前時間並進行下次迴圈嘗試,如果從通道中讀取出的不是期待的值(預期值是true
),則嘗試重新從通道中讀取並重置計時器。
func main() {
c := make(chan bool)
go func() {
for i := 0; i < 5; i++ {
time.Sleep(time.Second * 1)
c <- false
}
time.Sleep(time.Second * 1)
c <- true
}()
go func() {
// try to read from channel, block at most 5s.
// if timeout, print time event and go on loop.
// if read a message which is not the type we want(we want true, not false),
// retry to read.
timer := time.NewTimer(time.Second * 5)
for {
// timer is active , not fired, stop always returns true, no problems occurs.
if !timer.Stop() {
<-timer.C
}
timer.Reset(time.Second * 5)
select {
case b := <-c:
if b == false {
fmt.Println(time.Now(), ":recv false. continue")
continue
}
//we want true, not false
fmt.Println(time.Now(), ":recv true. return")
return
case <-timer.C:
fmt.Println(time.Now(), ":timer expired")
continue
}
}
}()
//to avoid that all goroutine blocks.
var s string
fmt.Scanln(&s)
}
程式的輸出如下:
2020-05-13 12:49:48.90292 +0800 CST m=+1.004554120 :recv false. continue
2020-05-13 12:49:49.906087 +0800 CST m=+2.007748042 :recv false. continue
2020-05-13 12:49:50.910208 +0800 CST m=+3.011892138 :recv false. continue
2020-05-13 12:49:51.914291 +0800 CST m=+4.015997373 :recv false. continue
2020-05-13 12:49:52.916762 +0800 CST m=+5.018489240 :recv false. continue
2020-05-13 12:49:53.920384 +0800 CST m=+6.022129708 :recv true. return
目前來看沒什麼問題,使用Reset重置計時器也起作用了,接下來我們對producer goroutin
做一些更改,我們把producer goroutine
裡每秒傳送值的邏輯改成每6
秒傳送值,而consumer gouroutine
裡和計時器還是5
秒就到期。
// producer
go func() {
for i := 0; i < 5; i++ {
time.Sleep(time.Second * 6)
c <- false
}
time.Sleep(time.Second * 6)
c <- true
}()
再次執行會發現程式發生了deadlock
在第一次報告計時器過期後直接阻塞住了:
2020-05-13 13:09:11.166976 +0800 CST m=+5.005266022 :timer expired
那程式是在哪阻塞住的呢?對就是在抽乾timer.C
通道時阻塞住了(英文叫做drain channel比喻成流乾管道里的水,在程式裡就是讓timer.C
管道中不再存在未接收的值)。
if !timer.Stop() {
<-timer.C
}
timer.Reset(time.Second * 5)
producer goroutine
的傳送行為發生了變化,comsumer goroutine
在收到第一個資料前有了一次計時器過期的事件,for
迴圈進行一下次迴圈。這時timer.Stop
函式返回的不再是true
,而是false
,因為計時器已經過期了,上面提到的維護著所有活躍計時器的最小堆中已經不包含該計時器了。而此時timer.C
中並沒有資料,接下來用於drain channel
的程式碼會將consumer goroutine
阻塞住。
這種情況,我們應該直接Reset
計時器,而不用顯式drain channel
。如何將這兩種情形合二為一呢?我們可以利用一個select
來包裹drain channel
的操作,這樣無論channel
中是否有資料,drain
都不會阻塞住。
//consumer
go func() {
// try to read from channel, block at most 5s.
// if timeout, print time event and go on loop.
// if read a message which is not the type we want(we want true, not false),
// retry to read.
timer := time.NewTimer(time.Second * 5)
for {
// timer may be not active, and fired
if !timer.Stop() {
select {
case <-timer.C: //try to drain from the channel
default:
}
}
timer.Reset(time.Second * 5)
select {
case b := <-c:
if b == false {
fmt.Println(time.Now(), ":recv false. continue")
continue
}
//we want true, not false
fmt.Println(time.Now(), ":recv true. return")
return
case <-timer.C:
fmt.Println(time.Now(), ":timer expired")
continue
}
}
}()
執行修改後的程式,發現程式不會被阻塞住,能正常進行通道讀取,讀取到true
值後會自行退出。輸出結果如下:
2020-05-13 13:25:08.412679 +0800 CST m=+5.005475546 :timer expired
2020-05-13 13:25:09.409249 +0800 CST m=+6.002037341 :recv false. continue
2020-05-13 13:25:14.412282 +0800 CST m=+11.005029547 :timer expired
2020-05-13 13:25:15.414482 +0800 CST m=+12.007221569 :recv false. continue
2020-05-13 13:25:20.416826 +0800 CST m=+17.009524859 :timer expired
2020-05-13 13:25:21.418555 +0800 CST m=+18.011245687 :recv false. continue
2020-05-13 13:25:26.42388 +0800 CST m=+23.016530193 :timer expired
2020-05-13 13:25:27.42294 +0800 CST m=+24.015582511 :recv false. continue
2020-05-13 13:25:32.425666 +0800 CST m=+29.018267054 :timer expired
2020-05-13 13:25:33.428189 +0800 CST m=+30.020782483 :recv false. continue
2020-05-13 13:25:38.432428 +0800 CST m=+35.024980796 :timer expired
2020-05-13 13:25:39.428343 +0800 CST m=+36.020887629 :recv true. return
總結
以上比較詳細地介紹了Go
語言的計時器以及它們的使用方法和注意事項,總結一下有如下關鍵點:
-
Timer
和Ticker
都是在執行時計時器runtime.timer
的基礎上實現的。 - 執行時裡的所有計時器的事件函式都由執行時內唯一的
goroutine
timerproc
觸發。 -
time.Tick
建立的Ticker
在執行時不會被gc
回收,能不用就不用。 -
Timer
和Ticker
的時間channel
都是帶有一個緩衝的通道。 -
time.After
,time.NewTimer
,time.NewTicker
建立的計時器觸發時都會執行sendTime
。 -
sendTime
和計時器帶快取的時間通道保證了計時器不會阻塞程式。 -
Reset
計時器時要注意drain channel
和計時器過期存在競爭條件。