go: 协程生命周期控制

原创王义杰运维开发王义杰 2024-02-28 22:27 广东听全文

在go开发中，如果我们希望在一定条件下，比如执行了某些任务后，主动结束其它相关的协程，可以通过使用context包来实现。context包允许我们发送取消信号给一个或多个协程，这样我们就可以控制协程的生命周期，包括它们的启动和结束。下面是一个使用context包来结束协程的示例：

package main

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

func main() {
    // 创建一个context对象和一个取消函数
    ctx, cancel := context.WithCancel(context.Background())

    // 在新的协程中运行一个任务
    go func(ctx context.Context) {
        for {
            select {
            case <-ctx.Done():
                fmt.Println("协程被取消")
                return
            default:
                // 模拟协程正在执行任务
                fmt.Println("协程正在执行任务...")
                time.Sleep(1 * time.Second)
            }
        }
    }(ctx)

    // 模拟主程序执行了一些任务
    time.Sleep(3 * time.Second) // 假设主程序执行了3秒的任务

    // 现在我们想要取消协程
    fmt.Println("准备取消协程")
    cancel() // 调用取消函数来发送取消信号

    // 给协程一点时间来响应取消信号
    time.Sleep(1 * time.Second)

    fmt.Println("主程序结束")
}