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。

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。