GO1.7之后，新增了context.Context這個package，實現goroutine的管理。
Context基本的用法參考GOLANG使用Context管理關聯goroutine。
實際上，Context還有個非常重要的作用，就是設置超時。比如，如果我們有個API是這樣設計的：
|
1
2
3
4
5
6
7
8
9
10
|
type Packet interface { encoding.BinaryMarshaler encoding.BinaryUnmarshaler}type Stack struct {}func (v *Stack) Read(ctx context.Context) (pkt Packet, err error) { return} |
一般使用是這樣使用，創建context然后調用接口：
|
1
2
3
|
ctx,cancel := context.WithCancel(context.Background())stack := &Stack{}pkt,err := stack.Read(ctx) |
那么，它本身就可以支持取消和超時，也就是用戶如果需要取消，比如發送了SIGINT信號，程序需要退出，可以在收到信號后調用cancel：
|
1
2
3
4
5
6
7
|
sc := make(chan os.Signal, 0)signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM)go func() { for range sc { cancel() }}() |
如果需要超時，這個API也不用改，只需要調用前設置超時時間：
|
1
2
3
|
ctx,cancel := context.WithTimeout(context.Background(), 3*time.Second)defer cancel()pkt,err := stack.Read(ctx) |
如果一個程序在運行，比如Read在等待，那么在沒有人工干預的情況下，那就應該自己運行就好了。而人工干預，也就是需要取消，比如要升級程序了，或者需要停止服務了，都屬于這種取消操作。而超時，一般是系統的策略，因為不能一直等下去，就需要在一定時間沒有反應時終止服務。實際上context這兩個都能支持得很好，而且還不影響Read本身的邏輯，在Read中只需要關注context是否Done：
|
1
2
3
4
5
6
7
8
|
func (v *Stack) Read(ctx context.Context) (pkt Packet, err error) { select { // case <- dataChannel: // Parse packet from data channel. case <- ctx.Done(): return nil,ctx.Err() } return} |
這是為何context被接納成為標準庫的包的緣故了吧，非常之強大和好用，而又非常簡單。一行context，深藏功與名。
另外，Context還可以傳遞上下文的Key-Value對象，比如我們希望日志中，相關的goroutine都打印一個簡化的CID，那么就可以用context.WithValue，參考go-oryx-lib/logger。
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持服務器之家。
原文鏈接：https://gocn.vip/article/373
