一般一个业务很少不用到goroutine的,因为很多方法是需要等待的,例如http.Server.ListenAndServe
这个就是等待的,除非关闭了Server或Listener,否则是不会返回的。除非是一个API服务器,否则肯定需要另外起goroutine发起其他的服务,而且对于API服务器来说,在http.Handler
的处理函数中一般也需要起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收到退出信号后,应该通知server退出。暴力一点的是直接调用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的所有goroutine
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时继续使用,譬如新加一个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来让服务器退出,可以这么写:
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,只是影响当前的ctx。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:https://gocn.vip/article/333