13518219792
这篇文章给大家介绍怎么在GOLANG中利用Context管理关联goroutine，内容非常详细，感兴趣的小伙伴们可以参考借鉴，希望对大家能有所帮助。
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了：
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：
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，可以这么写：
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来让服务器退出，可以这么写：
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。
关于怎么在GOLANG中利用Context管理关联goroutine就分享到这里了，希望以上内容可以对大家有一定的帮助，可以学到更多知识。如果觉得文章不错，可以把它分享出去让更多的人看到。
另外有需要云服务器可以了解下创新互联scvps.cn，海内外云服务器15元起步，三天无理由+7*72小时售后在线，公司持有idc许可证，提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案，具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势，专为企业上云打造定制，能够满足用户丰富、多元化的应用场景需求。
