Senior

func main() { c := make(chan int, 1000) for i := 0; i < 100; i++ { go foo(c) } sum := 0 for r := range c { sum += r } fmt.Println(sum) } func foo(c chan int) { r := rand.Int() for i := 0; i < r; i++ { c <- r } } --- func main() { c := make(chan int, 1000) var wg sync.WaitGroup wg.Add(100) for i := 0; i < 100; i++ { go func() { defer wg.Done() foo(c) }() } go func() { wg.Wait() close(c) }() sum := 0 for r := range c { sum += r } // // fmt.Println(sum) } func foo(c chan int) { r := rand.Int() for i := 0; i < r; i++ { c <- r } }

sobes.tech AI

Odpověď od AI

V první verzi kódu je problém: smyčka for r := range c bude čekat do nekonečna, protože kanál c nikdy není uzavřen. Kvůli tomu se program zasekne.

Ve druhé verzi byl přidán sync.WaitGroup pro sledování dokončení všech gorutin, a poté je kanál c uzavřen. To umožňuje správně ukončit smyčku čtení z kanálu.

Klíčové body:

Použití sync.WaitGroup k sledování dokončení všech gorutin.
Uzavření kanálu po dokončení zápisu všech gorutin.
Smyčka for r := range c končí, když je kanál uzavřen.

Příklad opraveného kódu:

group := sync.WaitGroup{}
group.Add(100)
for i := 0; i < 100; i++ {
    go func() {
        defer group.Done()
        foo(c)
    }()
}
go func() {
    group.Wait()
    close(c)
}()

sum := 0
for r := range c {
    sum += r
}

Předchozí Další