Go语言大厂编程 —— 高效使用 context 上下文管理

后端

2023-11-06 15:03:25

Go语言中的context上下文管理

在Go语言中，context上下文管理是一个非常重要的概念，它可以帮助我们管理和传递goroutine之间的信息，从而简化代码并提高程序的可维护性。

context上下文管理主要用于在goroutine之间传递以下几种信息：

取消信号：用于通知goroutine取消其当前任务。
超时时间：用于指定goroutine执行任务的超时时间。
截止时间：用于指定goroutine执行任务的截止时间。
k-v数据：用于传递其他任意数据。

context上下文管理的使用方法

创建context上下文

context上下文可以通过context包中的WithCancel、WithTimeout、WithDeadline等函数来创建。这些函数都会返回一个新的context上下文，并在其中包含了相应的信息。

// 创建一个带有取消信号的context上下文
ctx, cancel := context.WithCancel(context.Background())

// 创建一个带有超时时间的context上下文
ctx, cancel := context.WithTimeout(context.Background(), time.Second * 10)

// 创建一个带有截止时间的context上下文
ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(time.Second * 10))

在goroutine中使用context上下文

在goroutine中可以使用context包中的WithValue函数将任意数据存储到context上下文中，也可以使用context包中的Value函数从context上下文中获取数据。

// 将数据存储到context上下文中
ctx = context.WithValue(ctx, "key", "value")

// 从context上下文中获取数据
value := ctx.Value("key")

取消goroutine

如果goroutine需要取消其当前任务，可以使用context包中的CancelFunc函数来取消context上下文。这将导致context上下文中包含的取消信号被触发，从而通知goroutine取消其任务。

// 取消context上下文
cancel()

超时和截止时间

如果goroutine需要在指定的时间内完成其任务，可以使用context包中的WithTimeout或WithDeadline函数来创建带有超时时间或截止时间的context上下文。如果goroutine在指定的时间内没有完成其任务，则context上下文中的超时时间或截止时间将会触发，从而导致goroutine被取消。