这篇文章主要介绍了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()pk10开奖结果,但是如果有些请求还需要取消怎么办呢?最好用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
}

pk10开奖结果如果只是用到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。

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

最新资讯
人民日报海外版:“天宫”,从传说成为传奇

pk10开奖结果人民日报海外版:“天宫

北京时间7月19日21时06分,一道璀璨的光轨划过南太平洋
“天眼查”诉“企查查”不正当竞争索赔五百万

“天眼查”诉“企查查

因认为首创的广告语“查公司,查老板,查关系”被使用,“天
“吹牛”软件侵权“微信红包” 被判赔50万

“吹牛”软件侵权“微

法(tian)院(ping)经审理认为 该案存在不正当竞争 被告赔偿腾讯公司
阿里正式交棒在即 拆股为赴港上市铺路?

阿里正式交棒在即 拆

阿里正式交棒在即拆股为赴港上市铺路?■本报记者卢晓北
蔚来拟拆分NIO Power融资 不靠它挣钱李斌食言了

蔚来拟拆分NIO Power

蔚来将拆分旗下能源补给服务NIO Power,寻求在今年Q4完
支付行业下半场:精细化竞争时代开启

支付行业下半场:精细化

据统计,2018年全年,第三方支付机构因违规被处罚的金额超
最新文章
在go中使用omitempty的代码实例

在go中使用omitempty

今天小编就为大家分享一篇关于在go中使用omitempty的
go for range遍历二维数组的示例

go for range遍历二维

今天小编就为大家分享一篇关于go for range遍历二维数
Golang中重复错误处理的优化方法

Golang中重复错误处理

这篇文章主要给大家介绍了关于Golang中重复错误处理优
使用go gin来操作cookie的讲解

使用go gin来操作cook

今天小编就为大家分享一篇关于使用go gin来操作cookie
使用go来操作redis的方法示例

使用go来操作redis的

今天小编就为大家分享一篇关于使用go来操作redis的方
golang读取文件的常用方法总结

pk10开奖结果golang读取文件的常用

今天小编就为大家分享一篇关于golang读取文件的常用方
pk10开奖结果pk10开奖结果_4GBZN pk10开奖结果_4nZzgs7 pk10开奖结果_QwaQv6 pk10开奖结果_KSpi0f pk10开奖结果_nzLSay pk10开奖结果_nI2OpM pk10开奖结果_na6KD pk10开奖结果_XhLUKn pk10开奖结果_5mMi8 pk10开奖结果_RXxqysm