GOLANG使用Context管理关联goroutine的方法
一般一个业务很少不用到goroutine的，因为很多方法是需要等待的，例如http.Server.ListenAndServe这个就是等待的，除非关闭了Server或Listener，否则是不会返回的。除非是一个API服务器，否则肯定需要另外起goroutine发起其他的服务，而且对于API服务器来说，在http.Handler的处理函数中一般也需要起goroutine，如何管理这些goroutine，在GOLANG1.7提供context.Context。
先看一个简单的，如果启动两个goroutine，一个是HTTP，还有个信号处理的收到退出信号做清理：
wg:=sync.WaitGroup{}
deferwg.Wait()
wg.Add(1)
gofunc(){
deferwg.Done()
ss:=make(os.Signal,0)
signal.Notify(ss,syscall.SIGINT,syscall.SIGTERM)
fors:=ss{
fmt.Println("Gotsignal",s)
break
}
}()
wg.Add(1)
gofunc(){
deferwg.Done()
svr:=&http.Server{Addr:":8080",Handler:nil,}
fmt.Println(svr.ListenAndServe())
}
很清楚，起了两个goroutine，然后用WaitGroup等待它们退出。如果它们之间没有交互，不互相影响，那真的是蛮简单的，可惜这样是不行的，因为信号的goroutine收到退出信号后，应该通知server退出。暴力一点的是直接调用svr.Close()，但是如果有些请求还需要取消怎么办呢？最好用Context了：
wg:=sync.WaitGroup{}
deferwg.Wait()
ctx,cancel:=context.WithCancel(context.Background())
wg.Add(1)
gofunc(){
deferwg.Done()
ss:=make(chanos.Signal,0)
signal.Notify(ss,syscall.SIGINT,syscall.SIGTERM)
select{
case<-ctx.Done():
return
cases:=<-ss:
fmt.Println("Gotsignal",s)
cancel()//取消请求，通知用到ctx的所有goroutine
return
}
}()
wg.Add(1)
gofunc(){
deferwg.Done()
defercancel()
svr:=&http.Server{Addr:":8080",Handler:nil,}
gofunc(){
select{
case<-ctx.Done():
svr.Close()
}
}
fmt.Println(svr.ListenAndServe())
}
这个方式可以在新开goroutine时继续使用，譬如新加一个goroutine，里面读写了UDPConn：
wg.Add(1)
gofunc(){
deferwg.Done()
defercancel()
varconn*net.UDPConn
ifconn,err=net.Dial("udp","127.0.0.1:1935");err!=nil{
fmt.Println("DialUDPserverfailed,erris",err)
return
}
fmt.Println(UDPRead(ctx,conn))
}()
UDPRead=func(ctxcontext.Context,conn*net.UDPConn)(errerror){
wg:=sync.WaitGroup{}
deferwg.Wait()
ctx,cancel:=context.WithCancel(ctx)
wg.Add(1)
gofunc(){
deferwg.Done()
defercancel()
for{
b:=make([]byte,core.MTUSize)
size,_,err:=conn.ReadFromUDP(b)
//处理UDP包b[:size]
}
}()
select{
case<-ctx.Done():
conn.Close()
}
return
}
如果只是用到HTTPServer，可以这么写：
funcrun(ctxcontex.Context){
server:=&http.Server{Addr:addr,Handler:nil}
gofunc(){
select{
case<-ctx.Done():
server.Close()
}
}()
http.HandleFunc("/api",func(whttp.ResponseWriter,r*http.Request){
})
fmt.Println(server.ListenAndServe())
}
如果需要提供一个API来让服务器退出，可以这么写：
funcrun(ctxcontex.Context){
server:=&http.Server{Addr:addr,Handler:nil}
ctx,cancel:=context.WithCancel(ctx)
http.HandleFunc("/quit",func(whttp.ResponseWriter,r*http.Request){
cancel()//使用局部的ctx和cancel
})
gofunc(){
select{
case<-ctx.Done():
server.Close()
}
}()
fmt.Println(server.ListenAndServe())
}
使用局部的ctx和cancel，可以避免cancel传入的ctx，只是影响当前的ctx。
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。