?1.介紹

在 Go 語(yǔ)言標(biāo)準(zhǔn)庫(kù) time? 包中的 Timer 類(lèi)型，它是表示單一事件的計(jì)時(shí)器，也就是說(shuō)它是一次性定時(shí)器。

在 Go 語(yǔ)言項(xiàng)目開(kāi)發(fā)中，定時(shí)器使用廣泛，本文我們介紹 Go 語(yǔ)言中怎么使用 Timer，以及它的實(shí)現(xiàn)原理。

2.使用方式

使用 Timer? 一次性定時(shí)器，需要導(dǎo)入 time 包，創(chuàng)建定時(shí)器的方式有兩種，分別是：

func NewTimer(d Duration) *Timer

使用 func NewTimer? 創(chuàng)建 Timer?，入?yún)⑹嵌〞r(shí)器的等待時(shí)間，時(shí)間到達(dá)時(shí)，發(fā)送當(dāng)前時(shí)間到 channel中。

示例代碼：

func main() {
 nowTime := time.Now().Unix()
 fmt.Println(nowTime)
 NewTimerDemo()
}

func NewTimerDemo() {
 timer := time.NewTimer(2 * time.Second)
 select {
 case <-timer.C:
  currentTime := time.Now().Unix()
  fmt.Println(currentTime, "do something")
 }
}

輸出結(jié)果：

1665894136
1665894138 do something

通過(guò)閱讀上面這段代碼，我們可以發(fā)現(xiàn)我們定義了一個(gè) 2s? 后執(zhí)行的定時(shí)器 timer?，然后使用 select? 讀取 timer.C 中的數(shù)據(jù)，當(dāng)讀取到數(shù)據(jù)時(shí)，執(zhí)行特定業(yè)務(wù)邏輯代碼。

func AfterFunc(d Duration, f func()) *Timer

使用 func AfterFunc? 創(chuàng)建 Timer，入?yún)⑹嵌〞r(shí)器等待時(shí)間，和時(shí)間到達(dá)時(shí)執(zhí)行的函數(shù)。

示例代碼：

func main() {
 nowTime := time.Now().Unix()
 fmt.Println(nowTime)
 AfterFuncDemo()
}

func AfterFuncDemo() {
 time.AfterFunc(2 * time.Second, func() {
  currentTime := time.Now().Unix()
  fmt.Println(currentTime, "do something")
 })
 time.Sleep(3 * time.Second)
}

閱讀上面這段代碼，細(xì)心的讀者朋友們可能已經(jīng)發(fā)現(xiàn)，我們?cè)诖a末尾使用 time.Sleep()?，這是因?yàn)?nbsp;time.AfterFunc() 是異步執(zhí)行的，所以需要等待協(xié)成退出。

3.實(shí)現(xiàn)原理

我們?cè)谠创a中查看 Timer? 的數(shù)據(jù)結(jié)構(gòu)，發(fā)現(xiàn)它包含兩個(gè)字段，其中一個(gè)是可導(dǎo)出字段 C?，這是一個(gè) Time?類(lèi)型的 chan?；另一個(gè)是不可導(dǎo)出字段 r?，這是一個(gè) runtimeTimer 類(lèi)型。

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

實(shí)際上，每個(gè) Go 應(yīng)用程序底層都會(huì)有一個(gè)特定的協(xié)程管理 Timer?，該協(xié)程（底層協(xié)程）監(jiān)控到某個(gè) Timer? 指定的時(shí)間到達(dá)時(shí)，就會(huì)將當(dāng)前時(shí)間發(fā)送到 C? 中，然后上層讀取到 C 中的數(shù)據(jù)時(shí)，執(zhí)行相關(guān)業(yè)務(wù)邏輯代碼。

底層協(xié)程監(jiān)控 Timer? 的 r? 字段中的數(shù)據(jù)，我們?cè)谠创a中查看一下 runtimeTimer 的數(shù)據(jù)結(jié)構(gòu)：

type runtimeTimer struct {
    tb     uintptr
    i      int
    when   int64
    period int64
    f      func(interface{}, uintptr)
    arg    interface{}
    seq    uintptr
}

閱讀上面這段代碼，我們可以發(fā)現(xiàn) runtimeTimer? 中包含的所有字段，我們重點(diǎn)了解 when、f? 和 arg。

• when：定時(shí)器執(zhí)行時(shí)間。
• f：定時(shí)器執(zhí)行的回調(diào)函數(shù)。
• arg：定時(shí)器執(zhí)行的回調(diào)函數(shù)的參數(shù)。

在簡(jiǎn)單了解 Timer? 的數(shù)據(jù)結(jié)構(gòu)之后，我們?cè)谠创a中查看一下 func NewTimer 的代碼：

// NewTimer creates a new Timer that will send
// the current time on its channel after at least duration d.
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
}

閱讀上面這段代碼，我們可以發(fā)現(xiàn) func NewTimer? 的實(shí)現(xiàn)非常簡(jiǎn)單，它實(shí)際上就是構(gòu)造了一個(gè) Timer?，然后把 Timer.r? 傳參給 startTimer()?，除了 startTimer()? 函數(shù)外，還有兩個(gè)函數(shù)，分別是 when()?和 sendTime?，其中 when()? 是計(jì)算計(jì)時(shí)器的執(zhí)行時(shí)間，sendTime 是計(jì)時(shí)器時(shí)間到達(dá)時(shí)執(zhí)行的事件（實(shí)際上就是將當(dāng)前時(shí)間寫(xiě)入通道中）。

sendTime 源碼：

func sendTime(c interface{}, seq uintptr) {
 // Non-blocking send of time on c.
 // Used in NewTimer, it cannot block anyway (buffer).
 // Used in NewTicker, dropping sends on the floor is
 // the desired behavior when the reader gets behind,
 // because the sends are periodic.
 select {
 case c.(chan Time) <- Now():
 default:
 }
}

我們已經(jīng)了解到，func NewTimer? 將構(gòu)造的 Timer.r? 傳參給 startTimer()?，它負(fù)責(zé)把 runtimeTimer?寫(xiě)入底層協(xié)程的數(shù)組中（如果底層協(xié)程未運(yùn)行，它將會(huì)啟動(dòng)底層協(xié)程），將 Timer? 交給底層協(xié)程監(jiān)控，也就是上面講到的，當(dāng)?shù)讓訁f(xié)程監(jiān)控到某個(gè) Timer 指定時(shí)間到達(dá)時(shí)，將當(dāng)前時(shí)間發(fā)送到它的通道中。

// startTimer adds t to the timer heap.
//go:linkname startTimer time.startTimer
func startTimer(t *timer) {
 if raceenabled {
  racerelease(unsafe.Pointer(t))
 }
 addtimer(t)
}

4.總結(jié)

本文我們介紹 Go 語(yǔ)言標(biāo)準(zhǔn)庫(kù) time? 包提供的一次性定時(shí)器 Timer，不僅介紹了它的使用方式，還介紹了它的實(shí)現(xiàn)原理。