在 go 中，取消长时间运行的 context 可通过调用 context.cancelfunc 返回的函数实现，以下为取消 context 的步骤：创建一个带超时时间的子上下文。在 goroutine 中循环监听上下文是否被取消，并在取消时停止运行。在主 goroutine 中调用 cancelfunc 取消上下文。
如何在 Go 中取消长时间运行的 Context
在 Go 中，context.Context 是一个用来管理请求和取消操作的类型。在长期运行的操作中，正确处理上下文至关重要，以避免资源泄漏和死锁。
取消 Context
取消上下文通过调用 context.CancelFunc 返回的函数来完成。这将立即发送取消信号，这可以由其他 goroutine 监听。
package mainimport ( "context" "fmt" "time")func main() { // 创建带超时时间的上下文 ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) defer cancel() // 在 goroutine 中模拟长时间运行 go func() { for {select {case <-ctx.Done(): // 当上下文被取消时，停止运行 fmt.Println("操作已取消") returndefault: // 继续运行} } }() // 在主 goroutine 中模拟操作 time.Sleep(5 * time.Second) fmt.Println("主操作已完成") // 取消上下文 cancel()}
登录后复制
在这个例子中：
context.Background() 创建一个新的顶层上下文（无超时或取消）。context.WithTimeout() 创建一个带超时时间的子上下文。cancel 函数返回一个 context.CancelFunc，用于取消上下文。在 goroutine 中，我们循环监听上下文是否被取消，并在取消时停止运行。在主 goroutine 中，我们等待一段时间，然后调用cancel 函数取消上下文。
实战案例
在以下实战案例中，我们使用 Context 来管理对远程数据库的请求。如果请求在一定时间内没有完成，我们将取消请求以避免资源浪费：
package mainimport ( "context" "database/sql" "fmt" "time")func main() { db, err := sql.Open("postgres", "user:password@host:port/database") if err != nil { panic(err) } defer db.Close() ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel() // 把 Context 传入数据库操作中 row := db.QueryRowContext(ctx, "SELECT name FROM users WHERE id = 1") var name string err = row.Scan(&name) if err != nil { if err == context.Canceled {// 请求已取消fmt.Println("请求已超时") } else {// 其他错误处理fmt.Println(err) } } else { fmt.Println("获取用户名成功:", name) }}
登录后复制
以上就是Golang 函数：如何取消长时间运行的上下文的详细内容，更多请关注范的资源库其它相关文章！
<
转载请注明：范的资源库 » Golang函数：如何取消长时间运行的上下文
