Go使用context控制协程取消的实战案例

type Context interface {
    Deadline() (deadline time.Time, ok bool)
    Done() <-chan struct{}
    Err() error
    Value(key any) any
}

ctx := context.Background() // 最顶层、永不取消的 context
ctx, cancel := context.WithCancel(parent) // 手动调用 cancel() 取消
ctx, cancel := context.WithTimeout(parent, 3*time.Second) // 指定超时时间
ctx, cancel := context.WithDeadline(parent, time.Now().Add(3*time.Second)) // 到期时间点

package main

import (
    "context"
    "fmt"
    "time"
)

func worker(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            fmt.Println("worker 任务被取消:", ctx.Err())
            return
        default:
            fmt.Println("worker 正在处理任务...")
            time.Sleep(1 * time.Second)
        }
    }
}

func main() {
    ctx, cancel := context.WithCancel(context.Background())

    go worker(ctx)

    // 主线程运行5秒后取消任务
    time.Sleep(5 * time.Second)
    cancel()

    // 等待协程打印结束语
    time.Sleep(1 * time.Second)
    fmt.Println("主程序退出")
}

worker 正在处理任务...
worker 正在处理任务...
worker 正在处理任务...
worker 正在处理任务...
worker 正在处理任务...
worker 任务被取消: context canceled
主程序退出

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

    go worker(ctx)

    time.Sleep(5 * time.Second) // 主线程等待更久，看任务是否能自动终止
    fmt.Println("主程序退出")
}

worker 正在处理任务...
worker 正在处理任务...
worker 正在处理任务...
worker 任务被取消: context deadline exceeded
主程序退出

func main() {
    ctx, cancel := context.WithCancel(context.Background())

    for i := 1; i <= 3; i++ {
        go func(id int) {
            for {
                select {
                case <-ctx.Done():
                    fmt.Printf("协程 %d 接收到取消信号\n", id)
                    return
                default:
                    fmt.Printf("协程 %d 正在工作\n", id)
                    time.Sleep(1 * time.Second)
                }
            }
        }(i)
    }

    time.Sleep(4 * time.Second)
    cancel()

    time.Sleep(1 * time.Second)
    fmt.Println("主程序退出")
}

协程 1 正在工作
协程 2 正在工作
协程 3 正在工作
...（多次输出）
协程 1 接收到取消信号
协程 2 接收到取消信号
协程 3 接收到取消信号
主程序退出