context，让并发更快乐的Go

后端

2023-03-20 03:01:28

并发编程：掌握 Go 语言中的 Context 包

在快节奏的数字世界中，并发编程已成为构建响应迅速且可扩展应用程序的关键。Go 语言以其并发功能而闻名，特别是因为它引入了协程（goroutine），这是一种轻量级的线程，可以与其他协程同时运行，共享相同的内存空间。

虽然协程为并发编程提供了强大的机制，但它们也引入了潜在的陷阱，例如协程之间的竞争，可能导致程序错误。Context 包 应运而生，旨在应对这些挑战，让您轻松编写健壮的并发程序。

了解 Context 包

Context 包是一组强大的工具，可帮助您管理并发任务，包括取消、超时和截止功能。通过使用此包，您可以避免并发任务在不需要时继续运行，从而防止程序出现错误。

取消功能

取消功能是 Context 包的核心。它允许您在需要时取消并发任务。例如，在 HTTP 请求处理中，如果您在请求完成之前取消了请求，那么任何正在运行的协程也会被取消，防止程序出现错误。

// 创建一个 context
ctx, cancel := context.WithCancel(context.Background())

// 启动一个协程，执行一个并发任务
go func() {
  // 在协程中执行任务
  for {
    select {
    case <-ctx.Done():
      // 任务被取消，退出协程
      return
    default:
      // 任务没有被取消，继续执行
      fmt.Println("任务正在执行")
    }
  }
}()

// 在主线程中取消任务
cancel()

超时和截止功能

除了取消功能，Context 包还提供了超时和截止功能。超时功能允许您设置一个超时时间，如果并发任务在指定时间内未完成，则会自动取消。截止功能类似，但它允许您设置一个严格的截止时间，强制取消并发任务。

// 创建一个 context，设置一个超时时间
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)

// 启动一个协程，执行一个并发任务
go func() {
  // 在协程中执行任务
  for {
    select {
    case <-ctx.Done():
      // 任务被取消，退出协程
      return
    default:
      // 任务没有被取消，继续执行
      fmt.Println("任务正在执行")
    }
  }
}()

// 在主线程中等待任务完成
select {
case <-ctx.Done():
  // 任务完成，退出主线程
  return
}