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 MI
Válasz az MI-től
Az első kódrészletben probléma van: a for r := range c ciklus végtelenül várni fog, mert a c csatornát soha nem zárják be. Emiatt a program lefagy.
A második változatban hozzáadtak egy sync.WaitGroup-ot, hogy várja az összes goroutine befejezését, majd ezt követően bezárják a c csatornát. Ez lehetővé teszi, hogy helyesen lépjen ki a csatornából olvasó ciklus.
Főbb pontok:
- A
sync.WaitGrouphasználata az összes goroutine befejezésének nyomon követésére. - A csatorna bezárása, miután minden goroutine befejezte az írást.
- A
for r := range cciklus akkor ér véget, amikor a csatorna be van zárva.
Javított kód példa:
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
}