大橙子网站建设,新征程启航
为企业提供网站建设、域名注册、服务器等服务
这篇文章给大家介绍怎么在GOLANG中利用Context管理关联goroutine,内容非常详细,感兴趣的小伙伴们可以参考借鉴,希望对大家能有所帮助。
创新互联公司基于成都重庆香港及美国等地区分布式IDC机房数据中心构建的电信大带宽,联通大带宽,移动大带宽,多线BGP大带宽租用,是为众多客户提供专业成都服务器托管报价,主机托管价格性价比高,为金融证券行业服务器托管,ai人工智能服务器托管提供bgp线路100M独享,G口带宽及机柜租用的专业成都idc公司。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许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。