Go语言中的context——并行处理与协程的有效工具

电脑技巧

2023-10-03 18:09:35

Context：掌控并发编程的利器

在现代的分布式系统中，并发编程早已成为家常便饭。Go语言作为一种出色的并发编程语言，提供了强大的工具来处理并发任务，其中 Context 便是一大法宝。

什么是Context？

Context是一个接口类型，它允许你在并发环境中管理任务的同步、取消和超时操作。它是一个不可变的值，提供了以下方法：

Done() ：返回一个channel，当context被取消或超时时关闭。
Err() ：返回一个error，指示context被取消或超时的原因。
Deadline() ：返回context的截止时间。
Value(key interface{}) (value interface{}, ok bool) ：返回context中存储的键值对。

创建Context

你可以使用以下方法创建context：

context.Background() ：创建一个新的根context，它不会被取消或超时。
context.WithCancel(parent context.Context) ：创建一个新的context，它继承了parent context的截止时间，并且可以被取消。
context.WithTimeout(parent context.Context, timeout time.Duration) ：创建一个新的context，它继承了parent context的截止时间，并且会在timeout时间后超时。
context.WithDeadline(parent context.Context, deadline time.Time) ：创建一个新的context，它继承了parent context的截止时间，并且会在deadline时间后超时。

使用Context

Context在并发编程中扮演着至关重要的角色：

取消任务

你可以使用 context.CancelFunc 来取消任务。这个函数会取消context及其所有子context。

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

// 执行一些任务

cancel()

检测任务是否被取消或超时

使用 context.Done() 方法可以检测任务是否被取消或超时。当context被取消或超时时，context.Done() 返回的channel会关闭。

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

go func() {
  for {
    select {
    case <-ctx.Done():
      // 任务被取消或超时
      return
    default:
      // 执行一些任务
    }
  }
}()

cancel()

设置任务的截止时间

使用 context.WithTimeout() 或 context.WithDeadline() 方法可以设置任务的截止时间。当任务的截止时间到了之后，context会自动被取消。

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

go func() {
  // 执行一些任务

  // 任务超时
  select {
  case <-ctx.Done():
    return
  }
}()

cancel()

处理Context被取消或超时的场景

当context被取消或超时时，你可以做以下事情：

记录错误信息
清理资源
通知其他协程

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

go func() {
  defer cancel()

  for {
    select {
    case <-ctx.Done():
      // 任务被取消或超时
      // 记录错误信息
      // 清理资源
      // 通知其他协程
      return
    default:
      // 执行一些任务
    }
  }
}()

cancel()