Golang中WaitGroup、Context、goroutine定時器及逾時學習筆記

這是一個建立於 的文章，其中的資訊可能已經有所發展或是發生改變。

原文串連：http://targetliu.com/2017/5/2...
好久沒有發過文章了 - -||，今天發一篇 golang 中 goroutine 相關的學習筆記吧，以樣本為主。

WaitGroup

WaitGroup 在 sync 包中，用於阻塞主線程執行直到添加的 goroutine 全部執行完畢。

Context

Context 是在 Go1.7 中移入標準庫的。

Context 包不僅實現了在程式單元之間共用狀態變數的方法，同時能通過簡單的方法，使我們在被調用程式單元的外部，通過設定ctx變數值，將到期或撤銷這些訊號傳遞給被調用的程式單元。

goroutine的定時器及逾時

這是兩個有趣又實用的功能，在標準庫 time 包裡提供。

樣本

源碼

<!--more-->

package mainimport (    "context"    "fmt"    "sync"    "time")func main() {    ch := make(chan int)    //定義一個WaitGroup，阻塞主線程執行    var wg sync.WaitGroup    //添加一個goroutine等待    wg.Add(1)    //goroutine逾時    go func() {        //執行完成，減少一個goroutine等待        defer wg.Done()        for {            select {            case i := <-ch:                fmt.Println(i)            //goroutine內部3秒逾時            case <-time.After(3 * time.Second):                fmt.Println("goroutine1 timed out")                return            }        }    }()    ch <- 1    //新增一個1秒執行一次的計時器    ticker := time.NewTicker(1 * time.Second)    defer ticker.Stop()    //新增一個10秒逾時的上下文    background := context.Background()    ctx, _ := context.WithTimeout(background, 10*time.Second)    //添加一個goroutine等待    wg.Add(1)    go func(ctx context.Context) {        //執行完成，減少一個goroutine等待        defer wg.Done()        for {            select {            //每秒一次            case <-ticker.C:                fmt.Println("tick")            //內部逾時，不會被執行            case <-time.After(5 * time.Second):                fmt.Println("goroutine2 timed out")            //上下文傳遞逾時資訊，結束goroutine            case <-ctx.Done():                fmt.Println("goroutine2 done")                return            }        }    }(ctx)    //等待所有goroutine執行完成    wg.Wait()}

執行結果

1tickticktickgoroutine1 timed outtickticktickticktickticktickgoroutine2 done