• 企业400电话
  • 微网小程序
  • AI电话机器人
  • 电商代运营
  • 全 部 栏 目

    企业400电话 网络优化推广 AI电话机器人 呼叫中心 网站建设 商标✡知产 微网小程序 电商运营 彩铃•短信 增值拓展业务
    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{}
    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中for循环遍历channel时需要注意的问题详解
    • golang实现基于channel的通用连接池详解
    • Golang优雅关闭channel的方法示例
    • golang中单向channel的语法介绍
    • 解决Golang中goroutine执行速度的问题
    • golang gin 框架 异步同步 goroutine 并发操作
    • 关于golang利用channel和goroutine完成统计素数的思路
    上一篇:Go并发调用的超时处理的方法
    下一篇:GOLANG使用Context实现传值、超时和取消的方法
  • 相关文章
  • 

    © 2016-2020 巨人网络通讯 版权所有

    《增值电信业务经营许可证》 苏ICP备15040257号-8

    GOLANG使用Context管理关联goroutine的方法 GOLANG,使用,Context,管理,关联,