Golang 函数：如何优雅地取消并发 goroutine？

在 go 语言中，可以使用 context.context 和 context.cancelfunc 函数优雅地取消并发 goroutine：创建一个 context 和一个取消函数 cancel: ctx, cancel := context.withcancel(context.background())。创建一个 goroutine 并传递 ctx 作为参数。在需要时调用 cancel() 函数发送取消信号。使用 sync.waitgroup 等待 goroutine 退出。

Go 语言函数：如何优雅地取消并发 goroutine

在 Go 语言中，goroutine 是轻量级的并发单元。在某些情况下，我们需要在 goroutine 执行过程中将其取消。本文将介绍如何使用 context.Context 和 context.CancelFunc 函数优雅地取消并发 goroutine，并提供实战案例予以说明。

context.Context 和 context.CancelFunc

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

文希AI写作

AI论文写作平台

下载

context.Context 提供了一种在 goroutine 之间传播取消信号的方法。它的 CancelFunc 方法用于发送取消信号。

package main

import (
    "context"
    "fmt"
    "sync"
)

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    var wg sync.WaitGroup

    // 创建并启动一个 goroutine
    wg.Add(1)
    go func() {
        defer wg.Done()
        for {
            select {
            case <-ctx.Done():
                fmt.Println("Goroutine canceled")
                return
            default:
                fmt.Println("Goroutine running")
            }
        }
    }()

    // 等待一段时间后取消 goroutine
    time.Sleep(10 * time.Second)
    cancel()

    // 等待 goroutine 退出
    wg.Wait()
}

实战案例

该案例演示了如何使用 context.Context 取消一个 goroutine，goroutine 正处于一个无限循环中：

1. 创建一个 context 和一个取消函数 cancel: ctx, cancel := context.WithCancel(context.Background())。
2. 创建一个 goroutine 并传递 ctx 作为参数。goroutine 将进入一个无限循环，每秒打印 "Goroutine running"。
3. 在 goroutine 执行了一段时间后，调用 cancel() 函数来发送取消信号。
4. 使用 sync.WaitGroup 等待 goroutine 退出。

当取消信号被发送时，goroutine 将优雅地退出并打印 "Goroutine canceled"。