首页　>　文章列表　>　Golang 函数：用取消信号实现超时功能

Golang 函数：用取消信号实现超时功能

328 2025-01-24

使用 Go 语言中的取消信号可以实现函数超时功能。具体步骤包括：创建默认 Context 对象使用 WithCancel 函数为 Context 对象添加取消信号在函数中传递 Context 对象，检测取消信号在主程序中调用取消函数取消 Context 对象

Go 语言函数：取消信号实现超时功能

使用 Go 语言的取消信号可以轻松实现函数的超时功能。以下是如何使用 context 包来实现它的步骤：

1. 创建一个 Context 对象

context 包提供了 Context 类型，用于管理取消信号。你可以使用 context.Background() 创建一个默认的 Context 对象：

ctx := context.Background()

2. 带取消信号创建新的 Context 对象

要为 Context 对象添加取消信号，可以使用 context.WithCancel 函数：

ctx, cancel := context.WithCancel(ctx)

cancel 函数返回一个取消函数，可以在需要时调用它来取消 Context 对象。

3. 在函数中传递 Context 对象

现在你可以将 Context 对象作为函数参数传递，函数可以使用该 Context 对象来检测取消信号：

func myFunction(ctx context.Context) {
  for {
    select {
    case <-ctx.Done():
      // 操作已取消
      return
    default:
      // 执行任务
    }
  }
}

ctx.Done() 通道会在 Context 对象被取消时关闭。在上面的示例中，当 ctx 被取消时，select 语句就会退出并返回。

4. 取消函数调用

在主程序中，你可以在需要时调用 cancel 函数来取消 Context 对象：

go myFunction(ctx)
// ...
cancel()

实战案例：处理 HTTP 请求

以下是一个使用取消信号处理 HTTP 请求的简单示例：

import (
  "context"
  "fmt"
  "net/http"
)

func main() {
  ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)

  req, err := http.NewRequest(http.MethodGet, "https://example.com", nil)
  if err != nil {
    panic(err)
  }

  req = req.WithContext(ctx)

  resp, err := http.DefaultClient.Do(req)
  if err != nil {
    panic(err)
  }
  defer resp.Body.Close()

  fmt.Println(resp.StatusCode)

  cancel() // 手动取消请求，释放资源
}

注意：