重学Go语言之如何使用Context

更新时间：2023年07月31日 11:52:12 作者：程序员读书

Context，中文也叫做上下文，Go语言在1.7版本中新增的context包中定义了Context，下面我们就来一起看看如何在Go语言中使用Context吧

什么是Context

Context，中文叫做上下文，Go语言在1.7版本中新增的context包中定义了Context，Context本质是一个接口，这个接口一共定义了四个方法：

type Context interface {
 Deadline() (deadline time.Time, ok bool)
 Done() <-chan struct{}
 Err() error
 Value(key any) any
}

Dateline()：获取定时关闭的时间。
Done()：从一个channel获取关闭的信号
Err()：获取错误信息。
Value()：根据key从Context取值

为什么要使用Context

为什么要使用Context？或者说设计Context的目的是什么？

想像一下这样的场景，当用户开启浏览器访问我们的Web服务时，我们可能会开启多个Goroutine来处理用户的请求，这些Goroutine需要读取不同的资源，最终返回给用户，但如果用户在我们的Web服务还没处理完成就关闭浏览器，断开连接，而此时Web不知道用户已经关闭请求，仍然在处理并返回最终并不会被接收的数据。

Context设计的目的就是可以从上游的Goroutine发送信息给下游的Goroutine，回到处理用户请求的场景，当处理请求的Goroutine发现用户断开连接时，通过Context发送停止执行的信息，而下游的Goroutine得到停止信号时就返回，避免资源的浪费。

概括起来，Context的作用主要体现在两个方面：

在Goroutine之间传递关闭信息，定时关闭，超时关闭，手动关闭。
在Goroutine之间传递数据。

Context的使用

下面我们来讲讲Context的基本使用。

创建Context

任何上下文都是从一个空白的Context开始的，创建一个空白的Context有两种方式：

使用context.Background()：

ctx := context.Backgroud()

使用contenxt.TODO():

ctx := context.TODO()

当然大部分时候，我们不需要自己创建一个空白的Context，比如在处理HTTP请求时：

http.HandleFunc("/hello", func(w http.ResponseWriter, r *http.Request) {
 ctx := r.Context()
})

上面的代码中，可以从http.Request中获取Context实例，而http.Request的实际上也是调用context.Backgroud():

//net/http/request.go
func (r *Request) Context() context.Context {
 if r.ctx != nil {
  return r.ctx
 }
 return context.Background()
}

实例讲解

为了讲解Context是如何在Goroutine之间传递信号与数据，我们通过下面的案例进行说明：

func ReadFile(ctx context.Context, fileName string, result chan<- []byte) {
 file, err := os.Open(fileName)
 if err != nil {
  return
 }
 totalResult := make([]byte, 0)
 for {
  select {
  case <-ctx.Done():
   result <- []byte{}
   return
  default:
  }
  b := make([]byte, 1024)
  _, err := file.Read(b)
  if err == io.EOF {
   totalResult = append(totalResult, b...)
   break
  }
  totalResult = append(totalResult, b...)
 }
 result <- totalResult
}

在上面的程序中，我们调用Context的Done()方法，该方法会返回一个Channel，而使用select语句则可以让我们在处理业务的同时，监听上游Goroutine是否有传递取消执行的信息。

利用Context传递停止信号

空白的Context并不能发挥什么作用，要达到手动取消执行的目的，需要调用context包下的WithCancel函数进行封装，封装返回一个新的context以及一个取消的句柄cancel函数：

ctx := context.Background()
ctx, cancel := context.WithCancel(ctx)

下面是完整的使用方法：

package main
import (
 "context"
 "fmt"
 "io"
 "os"
 "time"
)
func main() {
 ctx := context.Background()
 ctx, cancel := context.WithCancel(ctx)
 result := make(chan []byte)
 go ReadFile(ctx, "./test.tar", result)
 time.Sleep(100 * time.Millisecond)
 cancel()
 fmt.Println(<-result)
}

在上面的程序中，我们创建一个Context之后，传给了ReadFile函数，并且在暂停100毫秒后调用cancel()函数，达到手动取消另一个Goroutine的目的。

给Context设置一个截止时间

除了手动取消，也可以调用context包下的WithDeadline()函数给Context加一个截止时间，这样在某个时间点，Context会自动发出取消信号：

afterTime := time.Now().Add(30 * time.Millisecond)
ctx, cancel := context.WithDeadline(context.Background(), afterTime)

下面是完整的示例：

package main
import (
 "context"
 "fmt"
 "io"
 "os"
 "time"
)
func main() {
 afterTime := time.Now().Add(30 * time.Millisecond)
 ctx, cancel := context.WithDeadline(context.Background(), afterTime)
 defer cancel()
 result := make(chan []byte)
 go ReadFile(ctx, "./test.tar", result)
 fmt.Println(<-result)
}

给Context一个超时限制

调用context包下的WithTimeout()函数可以为Context加一个超时限制，这对于我们编写超时控制程序非常有帮助：

ctx, cancel := context.WithTimeout(context.Background(), 10*time.Millisecond)

下面是完整的调用程序：

package main
import (
 "context"
 "fmt"
 "io"
 "os"
 "time"
)
func main() {
 ctx, cancel := context.WithTimeout(context.Background(), 10*time.Millisecond)
 defer cancel()
 result := make(chan []byte)
 go ReadFile(ctx, "./test.tar", result)
 fmt.Println(<-result)
}

使用Context传递数据

调用context包下的WithValue()函数可以生成一个携带数据的Context，这个机制方便我们跟踪一个处理流程中的Goroutine：

 ctx, cancel := context.WithValue(context.Background(), "testKey","testValue")

下游的Goroutine就可以通过Context的Value()函数来获取上游传递下来的值了。

func MyGoroutine(ctx context.Context){
 ctx.Value("testKey")
}

使用Context的几点建议

Context不要放在结构体中，需要以参数方式传递。

Context作为函数参数时，一般放在第一位，作为函数的第一个参数。

使用 context.Background函数生成根节点的Context。

Context 要传值必要的值，不要什么都传。

Context 是多协程安全的，可以在多个协程中使用。

小结

好了，至此，你应该对Context有所了解了吧，总的来说，通过Context可以做到：

手动控制下游Goroutine取消执行。
定时控制下游Goroutine取消执行。
超时控制下游Goroutine取消执行。
传递数据给下游的Goroutine。

到此这篇关于重学Go语言之如何使用Context的文章就介绍到这了,更多相关Go Context内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

Go
Context

Golang WebView跨平台的桌面应用库的使用
Golang WebView是一个强大的桌面应用库,本文介绍了Golang WebView的特点和使用方法,并列举示例详细的介绍了其在实际项目中的应用,具有一定的参考价值,感兴趣的可以了解一下
2024-03-03
golang中snappy的使用场合实例详解
在java 和go语言大字符传达的时候，采用snappy 压缩解压缩是最好的方案。下面这篇文章主要给大家介绍了关于golang中snappy使用场合的相关资料，文中通过示例代码介绍的非常详细，需要的朋友可以参考下。
2017-12-12
Go语言实现类似c++中的多态功能实例
Go本身不具有多态的特性，不能够像Java、C++那样编写多态类、多态方法。但是，使用Go可以编写具有多态功能的类绑定的方法。下面来一起看看吧
2016-09-09
golang 对私有函数进行单元测试的实例
这篇文章主要介绍了golang 对私有函数进行单元测试的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2021-05-05
Go使用Viper库读取YAML配置文件的示例代码
Viper是适用于Go应用程序的完整配置解决方案,它被设计用于在应用程序中工作,并且可以处理所有类型的配置需求和格式,本文给大家介绍了Go使用Viper库读取YAML配置文件的方法,需要的朋友可以参考下
2024-05-05
解决Go中拦截HTTP流数据时字段丢失的问题
在开发高并发的Web应用时,尤其是在处理HTTP代理和流数据拦截的场景下,遇到数据丢失的问题并不罕见,最近,在一个项目中,我遇到了一个棘手的问题：在拦截并转发HTTP流数据的过程中,某些数据字段因为处理过快而被丢失,所以本文给大家介绍如何解决这个问题
2024-08-08
GoLand利用plantuml生成UML类图
本文主要介绍了GoLand利用plantuml生成UML类图，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2022-06-06
详解Golang中创建error的方式总结与应用场景
Golang中创建error的方式包括errors.New、fmt.Errorf、自定义实现了error接口的类型等，本文主要为大家介绍了这些方式的具体应用场景，需要的可以参考一下
2023-07-07
Golang的strings.Split()踩坑记录
工作中，当我们需要对字符串按照某个字符串切分成字符串数组数时，常用到strings.Split()，本文主要介绍了Golang的strings.Split()踩坑记录，感兴趣的可以了解一下
2022-05-05
Golang流程控制语句的具体使用
在编写程序时,流程控制是必不可少的一部分,本文主要介绍了Golang流程控制语句的具体使用,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
2024-05-05