go的context总结

Go语言中的context包提供了一种在不同Goroutines之间传递取消信号、超时、截止日期和请求范围值的方法，用于更好地管理并控制并发操作。

WithCancel

在这个示例中，我们首先创建了一个带有取消功能的context，然后启动了一个Goroutine来执行某项工作，这个Goroutine在工作过程中定期检查ctx.Done()通道，如果接收到取消信号，则停止工作。在main函数中，我们等待了一段时间，然后调用cancel函数，以发送取消信号。在取消信号发送后，Goroutine会收到信号并停止工作。

package main import (     "context"     "fmt"     "time" ) func doSomething(ctx context.Context) {     for {         select {         case <-ctx.Done():             fmt.Println("Received cancellation signal. Cleaning up...")             return         default:             // Simulate some work             fmt.Println("Working...")             time.Sleep(1 * time.Second)         }     } } func main() {     // 创建一个带有取消功能的context     ctx, cancel := context.WithCancel(context.Background())     // 启动一个Goroutine执行某项工作，传入context     go doSomething(ctx)     // 等待一段时间，然后取消工作     time.Sleep(10 * time.Second)     cancel() // 调用cancel函数来发送取消信号     // 等待一段时间，以观察工作是否被取消     time.Sleep(5 * time.Second) }

WithTimeout

在这个示例中，我们首先创建一个带有超时控制的 context，并设置最大执行时间为 5 秒。然后，我们启动一个 Goroutine 来执行某项工作。主函数休眠 10 秒以等待 Goroutine 的完成或超时。由于 context.WithTimeout 设置了最大执行时间为 5 秒，因此 Goroutine 会在 5 秒后收到超时信号，结束工作。

context.WithTimeout 可以用于控制并发操作的执行时间，以确保它们不会无限期地运行。这在处理网络请求、I/O 操作、调用外部服务等场景中特别有用，以避免操作长时间阻塞并占用资源。

package main import (     "context"     "fmt"     "time" ) func doSomething(ctx context.Context) {     for {         select {         case <-ctx.Done():             fmt.Println("Received cancellation signal. Cleaning up...")             return         case <-time.After(1 * time.Second):             fmt.Println("Task completed.")         }     } } func main() {     // 创建一个带有超时控制的 context，设置最大执行时间为 5 秒     ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)     defer cancel() // 一定要记得调用 cancel，确保资源得以释放     // 启动一个 Goroutine 执行某项工作，传入 context     go doSomething(ctx)     // 主函数休眠 10 秒，以等待 Goroutine 完成或超时     time.Sleep(10 * time.Second) }

WithValue

package main import (     "context"     "fmt"     "time" ) func processRequest(ctx context.Context) {     // 从context中获取请求ID     requestID, ok := ctx.Value("requestKey").(int)     if !ok {         fmt.Println("Request ID not found in context.")         return     }     fmt.Printf("Processing request with ID %d\n", requestID) } func main() {     // 创建一个带有请求ID的context     ctx := context.WithValue(context.Background(), "requestKey", 12345)     // 在另一个函数中使用context     go processRequest(ctx)     // 主函数休眠一段时间，以确保Goroutine有时间运行     time.Sleep(5 * time.Second) }

WithDeadline

context.WithDeadline 方法用于创建一个带有截止日期的 context，允许你设置一个操作的最后截止日期，如果操作在截止日期之后仍未完成，context 会自动取消它。

package main import (     "context"     "fmt"     "time" ) func doSomething(ctx context.Context) {     for {         select {         case <-ctx.Done():             fmt.Println("Received cancellation signal. Cleaning up...")             return         case <-time.After(1 * time.Second):             fmt.Println("Task completed.")         }     } } func main() {     // 设置截止日期为当前时间加 5 秒     deadline := time.Now().Add(5 * time.Second)     // 创建一个带有截止日期的 context     ctx, cancel := context.WithDeadline(context.Background(), deadline)     defer cancel() // 一定要记得调用 cancel，确保资源得以释放     // 启动一个 Goroutine 执行某项工作，传入 context     go doSomething(ctx)     // 主函数休眠 10 秒，以等待 Goroutine 完成或截止日期到达     time.Sleep(10 * time.Second) }

context继承

这个例子5秒后结束程序。

package main import (     "context"     "fmt"     "time" ) func doSomething(ctx context.Context) {     for {         select {         case <-ctx.Done():             fmt.Println("Received cancellation signal. Cleaning up...")             return         default:             // Simulate some work             fmt.Println("Working...")             time.Sleep(1 * time.Second)         }     } } func main() {     // 创建一个带有取消功能的context     rootCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)     ctx, cancel := context.WithCancel(rootCtx)     // 启动一个Goroutine执行某项工作，传入context     go doSomething(ctx)     // 等待一段时间，然后取消工作     time.Sleep(10 * time.Second)     cancel() // 调用cancel函数来发送取消信号     // 等待一段时间，以观察工作是否被取消     time.Sleep(5 * time.Second) }