Golang 函数：用取消信号实现超时功能-Golang-PHP中文网

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

WBOY

发布： 2024-10-03 14:06:01

原创

625人浏览过

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

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

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

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

1. 创建一个 Context 对象

立即学习“go语言免费学习笔记（深入）”；

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

ctx := context.Background()

登录后复制

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

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

ctx, cancel := context.WithCancel(ctx)

登录后复制

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

动态WEB网站中的PHP和MySQL：直观的QuickPro指南第2版

动态WEB网站中的PHP和MySQL详细反映实际程序的需求，仔细地探讨外部数据的验证(例如信用卡卡号的格式)、用户登录以及如何使用模板建立网页的标准外观。动态WEB网站中的PHP和MySQL的内容不仅仅是这些。书中还提到如何串联JavaScript与PHP让用户操作时更快、更方便。还有正确处理用户输入错误的方法，让网站看起来更专业。另外还引入大量来自PEAR外挂函数库的强大功能，对常用的、强大的包

508

查看详情

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() // 手动取消请求，释放资源
}

登录后复制

注意：