Go語言使用通道響應計時器的事件

2020-07-16 10:04:51
Go語言中的 time 包提供了計時器的封裝。由於 Go語言中的通道和 goroutine 的設計,定時任務可以在 goroutine 中通過同步的方式完成,也可以通過在 goroutine 中非同步回撥完成。這裡將分兩種用法進行例子展示。

一段時間之後(time.After)

延遲回撥:
package main

import (
    "fmt"
    "time"
)

func main() {
    // 宣告一個退出用的通道
    exit := make(chan int)

    // 列印開始
    fmt.Println("start")

    // 過1秒後, 呼叫匿名函數
    time.AfterFunc(time.Second, func() {

        // 1秒後, 列印結果
        fmt.Println("one second after")

        // 通知main()的goroutine已經結束
        exit <- 0
    })

    // 等待結束
    <-exit
}
程式碼說明如下:
  • 第 10 行,宣告一個退出用的通道,往這個通道裡寫資料表示退出。
  • 第 16 行,呼叫 time.AfterFunc() 函數,傳入等待的時間和一個回撥。回撥使用一個匿名函數,在時間到達後,匿名函數會在另外一個 goroutine 中被呼叫。
  • 第 22 行,任務完成後,往退出通道中寫入數值表示需要退出。
  • 第 26 行,執行到此處時持續阻塞,直到 1 秒後第 22 行被執行後結束阻塞。

time.AfterFunc() 函數是在 time.After 基礎上增加了到時的回撥,方便使用。

而 time.After() 函數又是在 time.NewTimer() 函數上進行的封裝,下面的例子展示如何使用 timer.NewTimer() 和 time.NewTicker()。

定點計時

計時器(Timer)的原理和倒計時鬧鐘類似,都是給定多少時間後觸發。打點器(Ticker)的原理和鐘錶類似,鐘錶每到整點就會觸發。這兩種方法建立後會返回 time.Ticker 物件和 time.Timer 物件,裡面通過一個 C 成員,型別是只能接收的時間通道(<-chan Time),使用這個通道就可以獲得時間觸發的通知。

下面程式碼建立一個打點器,每 500 毫秒觸發一起;建立一個計時器,2 秒後觸發,只觸發一次。

計時器:
package main

import (
    "fmt"
    "time"
)

func main() {

    // 建立一個打點器, 每500毫秒觸發一次
    ticker := time.NewTicker(time.Millisecond * 500)

    // 建立一個計時器, 2秒後觸發
    stopper := time.NewTimer(time.Second * 2)

    // 宣告計數變數
    var i int

    // 不斷地檢查通道情況
    for {

        // 多路複用通道
        select {
        case <-stopper.C:  // 計時器到時了

            fmt.Println("stop")

            // 跳出迴圈
            goto StopHere

        case <-ticker.C:  // 打點器觸發了
            // 記錄觸發了多少次
            i++
            fmt.Println("tick", i)
        }
    }

// 退出的標籤, 使用goto跳轉
StopHere:
    fmt.Println("done")

}
程式碼說明如下:
  • 第 11 行,建立一個打點器,500 毫秒觸發一次,返回 *time.Ticker 型別變數。
  • 第 14 行,建立一個計時器,2 秒後返回,返回 *time.Timer 型別變數。
  • 第 17 行,宣告一個變數,用於累計打點器觸發次數。
  • 第 20 行,每次觸發後,select 會結束,需要使用迴圈再次從打點器返回的通道中獲取觸發通知。
  • 第 23 行,同時等待多路計時器信號。
  • 第 24 行,計時器信號到了。
  • 第 29 行,通過 goto 跳出迴圈。
  • 第 31 行,打點器信號到了,通過i自加記錄觸發次數並列印。