如何处理关闭同一个通道的多个 go-routines？

Question

我有 2 个从单个通道读取的 go-routines。4 秒后，我取消上下文并终止选择循环。在终止循环之前，我在通道上调用 close，因为有 2 个 go-routines close 被调用两次并导致恐慌，因为其中一个 go-routines 已经关闭了通道。目前我正在使用恢复来从恐慌中恢复过来，有没有更好的方法来做到这一点？

package main

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

func numberGen(ctx context.Context, numChan chan int) {
    num := 0
    doneCh := ctx.Done()
    defer func() {
        if r := recover(); r != nil {
            fmt.Println("recovered from ", r)
        }
    }()
    for {
        select {
        case <-doneCh:
            fmt.Println("done generating...")
            close(numChan)
            return
        default:
            num++
            numChan <- num
        }
    }
}

func main() {
    ctx, cancelFn := context.WithCancel(context.Background())
    numChan := make(chan int)
    var wg sync.WaitGroup
    wg.Add(2)
    go numberGen(ctx, numChan)
    go numberGen(ctx, numChan)

    go func(cfn context.CancelFunc) {
        time.Sleep(10 * time.Millisecond)
        cfn()
    }(cancelFn)

    for n := range numChan {
        fmt.Println("received value ", n)
    }

    time.Sleep(2 * time.Second)
}

Answer 1

在 goroutine 完成发送值后关闭通道。

var wg sync.WaitGroup
wg.Add(2)
go numberGen(ctx, numChan, &wg)
go numberGen(ctx, numChan, &wg)

go func() {
    wg.Wait()
    close(numChan)
}()

更新numberGen以呼叫Done()等待组。此外，删除对 的调用close。

func numberGen(ctx context.Context, numChan chan int, wg *sync.WaitGroup) {
    defer wg.Done()
    ...