登录
首页 >  Golang >  Go教程

GOLANG使用Context实现传值、超时和取消的方法

来源:脚本之家

时间:2023-01-07 12:12:31 220浏览 收藏

在Golang实战开发的过程中,我们经常会遇到一些这样那样的问题,然后要卡好半天,等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《GOLANG使用Context实现传值、超时和取消的方法》,聊聊超时、传值、GOLANGContext、取消,希望可以帮助到正在努力赚钱的你。

GO1.7之后,新增了context.Context这个package,实现goroutine的管理。

Context基本的用法参考GOLANG使用Context管理关联goroutine

实际上,Context还有个非常重要的作用,就是设置超时。比如,如果我们有个API是这样设计的:

type Packet interface {
  encoding.BinaryMarshaler
  encoding.BinaryUnmarshaler
}

type Stack struct {
}
func (v *Stack) Read(ctx context.Context) (pkt Packet, err error) {
  return
}

一般使用是这样使用,创建context然后调用接口:

ctx,cancel := context.WithCancel(context.Background())
stack := &Stack{}
pkt,err := stack.Read(ctx)

那么,它本身就可以支持取消和超时,也就是用户如果需要取消,比如发送了SIGINT信号,程序需要退出,可以在收到信号后调用cancel:

sc := make(chan os.Signal, 0)
signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM)
go func() {
  for range sc {
    cancel()
  }
}()

如果需要超时,这个API也不用改,只需要调用前设置超时时间:

ctx,cancel := context.WithTimeout(context.Background(), 3*time.Second)
defer cancel()
pkt,err := stack.Read(ctx)

如果一个程序在运行,比如Read在等待,那么在没有人工干预的情况下,那就应该自己运行就好了。而人工干预,也就是需要取消,比如要升级程序了,或者需要停止服务了,都属于这种取消操作。而超时,一般是系统的策略,因为不能一直等下去,就需要在一定时间没有反应时终止服务。实际上context这两个都能支持得很好,而且还不影响Read本身的逻辑,在Read中只需要关注context是否Done:

func (v *Stack) Read(ctx context.Context) (pkt Packet, err error) {
  select {
  // case 

这是为何context被接纳成为标准库的包的缘故了吧,非常之强大和好用,而又非常简单。一行context,深藏功与名。

另外,Context还可以传递上下文的Key-Value对象,比如我们希望日志中,相关的goroutine都打印一个简化的CID,那么就可以用context.WithValue,参考go-oryx-lib/logger。

好了,本文到此结束,带大家了解了《GOLANG使用Context实现传值、超时和取消的方法》,希望本文对你有所帮助!关注golang学习网公众号,给大家分享更多Golang知识!

声明:本文转载于:脚本之家 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>
评论列表