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 IA
Resposta da IA
Na primeira versão do código há um problema: o ciclo for r := range c ficará esperando indefinidamente porque o canal c nunca é fechado. Por causa disso, o programa ficará preso.
Na segunda versão, foi adicionado um sync.WaitGroup para aguardar a conclusão de todas as goroutines, e após isso o canal c é fechado. Isso permite sair corretamente do ciclo de leitura do canal.
Pontos principais:
- Uso de
sync.WaitGrouppara acompanhar a finalização de todas as goroutines. - Fechamento do canal após todas as goroutines terminarem de escrever.
- A iteração pelo canal com
for r := range ctermina quando o canal é fechado.
Exemplo de código corrigido:
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
}