一般一個業(yè)務(wù)很少不用到goroutine的，因為很多方法是需要等待的，例如http.Server.ListenAndServe這個就是等待的，除非關(guān)閉了Server或Listener，否則是不會返回的。除非是一個API服務(wù)器，否則肯定需要另外起goroutine發(fā)起其他的服務(wù)，而且對于API服務(wù)器來說，在http.Handler的處理函數(shù)中一般也需要起goroutine，如何管理這些goroutine，在GOLANG1.7提供context.Context。
先看一個簡單的，如果啟動兩個goroutine，一個是HTTP，還有個信號處理的收到退出信號做清理：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
wg := sync.WaitGroup{}defer wg.Wait()wg.Add(1)go func() { defer wg.Done() ss := make(os.Signal, 0) signal.Notify(ss, syscall.SIGINT, syscall.SIGTERM) for s := ss { fmt.Println("Got signal", s) break }}()wg.Add(1)go func() { defer wg.Done() svr := &http.Server{ Addr:":8080", Handler:nil, } fmt.Println(svr.ListenAndServe())} |
很清楚，起了兩個goroutine，然后用WaitGroup等待它們退出。如果它們之間沒有交互，不互相影響，那真的是蠻簡單的，可惜這樣是不行的，因為信號的goroutine收到退出信號后，應(yīng)該通知server退出。暴力一點的是直接調(diào)用svr.Close()，但是如果有些請求還需要取消怎么辦呢？最好用Context了：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
wg := sync.WaitGroup{}defer wg.Wait()ctx,cancel := context.WithCancel(context.Background())wg.Add(1)go func() { defer wg.Done() ss := make(chan os.Signal, 0) signal.Notify(ss, syscall.SIGINT, syscall.SIGTERM) select { case <- ctx.Done(): return case s := <- ss: fmt.Println("Got signal", s) cancel() // 取消請求，通知用到ctx的所有g(shù)oroutine return }}()wg.Add(1)go func() { defer wg.Done() defer cancel() svr := &http.Server{ Addr:":8080", Handler:nil, } go func(){ select { case <- ctx.Done(): svr.Close() } } fmt.Println(svr.ListenAndServe())} |
這個方式可以在新開goroutine時繼續(xù)使用，譬如新加一個goroutine，里面讀寫了UDPConn：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
wg.Add(1)go func() { defer wg.Done() defer cancel() var conn *net.UDPConn if conn,err = net.Dial("udp", "127.0.0.1:1935"); err != nil { fmt.Println("Dial UDP server failed, err is", err) return } fmt.Println(UDPRead(ctx, conn))}()UDPRead = func(ctx context.Context, conn *net.UDPConn) (err error) { wg := sync.WaitGroup{} defer wg.Wait() ctx, cancel := context.WithCancel(ctx) wg.Add(1) go func() { defer wg.Done() defer cancel() for { b := make([]byte, core.MTUSize) size, _, err := conn.ReadFromUDP(b) // 處理UDP包 b[:size] } }() select { case <-ctx.Done(): conn.Close() } return} |
如果只是用到HTTP Server，可以這么寫：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
func run(ctx contex.Context) { server := &http.Server{Addr: addr, Handler: nil} go func() { select { case <-ctx.Done(): server.Close() } }() http.HandleFunc("/api", func(w http.ResponseWriter, r *http.Request) { }) fmt.Println(server.ListenAndServe())} |
如果需要提供一個API來讓服務(wù)器退出，可以這么寫：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
func run(ctx contex.Context) { server := &http.Server{Addr: addr, Handler: nil} ctx, cancel := context.WithCancel(ctx) http.HandleFunc("/quit", func(w http.ResponseWriter, r *http.Request) { cancel() // 使用局部的ctx和cancel }) go func() { select { case <-ctx.Done(): server.Close() } }() fmt.Println(server.ListenAndServe())} |
使用局部的ctx和cancel，可以避免cancel傳入的ctx，只是影響當(dāng)前的ctx。
以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持服務(wù)器之家。
原文鏈接：https://gocn.vip/article/333
