脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → Golang → GO Context管理关联goroutine

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

 更新时间：2019年01月14日 09:32:48   作者：winlin  
这篇文章主要介绍了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。

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

您可能感兴趣的文章:

相关文章

  • Go项目怎么使用枚举

    Go项目怎么使用枚举

    枚举是一种很重要的数据类型，本文主要介绍了Go项目怎么使用枚举，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2022-08-08
  • Golang使用bcrypt实现密码加密和校验的操作代码

    Golang使用bcrypt实现密码加密和校验的操作代码

    bcrypt可以用于数据库中的用户密码保存,相比md5而言更加的安全可靠,这篇文章主要介绍了Golang使用bcrypt实现密码加密和校验的操作代码,需要的朋友可以参考下
    2024-05-05
  • Golang如何实现任意进制转换的方法示例

    Golang如何实现任意进制转换的方法示例

    进制转换是人们利用符号来计数的方法，进制转换由一组数码符号和两个基本因素“基数”与“位权”构成，这篇文章主要给大家介绍了关于Golang如何实现10进制转换62进制的方法，文中给出了详细的示例代码供大家参考学习学习，下面随着小编来一起学习学习吧。
    2017-09-09
  • 深入了解Golang 哈希算法之MD5、SHA-1和SHA-256

    深入了解Golang 哈希算法之MD5、SHA-1和SHA-256

    哈希算法是计算机科学领域中一种重要的技术，它将任意长度的输入数据映射为固定长度的哈希值，在本篇文章中，我们将深入探讨Golang中的哈希算法，从多个方面介绍其详细内容，希望通过本文的阅读你将对 Golang哈希算法有更全面的理解
    2023-05-05
  • Mango Cache缓存管理库TinyLFU源码解析

    Mango Cache缓存管理库TinyLFU源码解析

    这篇文章主要为大家介绍了Mango Cache缓存管理库TinyLFU源码解析，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2022-09-09
  • go原子级内存操作实现

    go原子级内存操作实现

    原子级内存操作是在多线程并发执行时,能够确保某个内存操作是不可中断的操作,本文主要介绍了go原子级内存操作实现,具有一定的参考价值,感兴趣的可以了解一下
    2024-02-02
  • 一文带你掌握Go语言运算符的使用

    一文带你掌握Go语言运算符的使用

    运算符用于在程序运行时执行数学或逻辑运算。Go 语言内置的运算符有：算术运算符、关系运算符、逻辑运算符、位运算符、赋值运算符、其他运算符。本文将带大家详细了解一下这些运算符的使用，感兴趣的可以了解一下
    2022-04-04
  • Go与C语言的互操作实现

    Go与C语言的互操作实现

    在Go与C语言互操作方面，Go更是提供了强大的支持。尤其是在Go中使用C，你甚至可以直接在Go源文件中编写C代码，本文就详细的介绍一下如何使用，感兴趣的可以了解一下
    2021-12-12
  • golang代码中调用Linux命令

    golang代码中调用Linux命令

    本文主要介绍了golang代码中调用Linux命令，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2023-02-02
  • gorm整合进go-zero的实现方法

    gorm整合进go-zero的实现方法

    go-zero提供的代码生成器里面，没有提供orm框架操作，但是提供了遍历的缓存操作，所以可以利用gorm当作一个sql语句的生成器，把生成后的sql语句放到go-zero生成的模板中去执行，对gorm整合进go-zero的实现方法感兴趣的朋友一起看看吧
    2022-03-03

最新评论