编写这段代码的更好的方式？

Question

package main

import "fmt"

const N = 10

func main() {
    ch := make(chan int, N)
    done := make(chan bool)

    for i := 0; i < N; i++ {
        go (func(n int, ch chan int, done chan bool) {
            for i := 0; i < N; i++ {
                ch <- n*N + i
            }
            done <- true
        })(i, ch, done)
    }

    numDone := 0
    for numDone < N {
        select {
        case i := <-ch:
            fmt.Println(i)
        case <-done:
            numDone++
        }
    }

    for {
        select {
        case i := <-ch:
            fmt.Println(i)
        default:
            return
        }
    }
}

编辑：为了澄清一点，我对此表示怀疑，因为done通道似乎在做关闭通道的工作，但当然，我不能在任何goroutine中关闭通道，因为所有例程共享同一个通道。因此，我使用done来模拟一个进行某种“缓冲关闭”的通道。

Edit2:原始代码实际上并不起作用，因为有时来自例程的done信号在它刚刚对ch执行int之前被读取。需要一个“清理”循环。

Answer 1

下面是sync.waitgroup的一个惯用用法，供您学习

（游乐场连线）

package main

import (
    "fmt"
    "sync"
)

const N = 10

func main() {
    ch := make(chan int, N)
    var wg sync.WaitGroup
    for i := 0; i < N; i++ {
        wg.Add(1)
        go func(n int) {
            defer wg.Done()
            for i := 0; i < N; i++ {
                ch <- n*N + i
            }
        }(i)
    }
    go func() {
        wg.Wait()
        close(ch)
    }()
    for i := range ch {
        fmt.Println(i)
    }
}

注意在两个go例程定义中使用了闭包，并注意第二个go语句等待所有例程完成，然后关闭通道，因此可以使用range。