你的 monitorWorker 永远不会死。当所有工作人员完成后,它会继续等待 cs。这种死锁是因为在 cs 上不会发送任何其他内容,因此 wg 永远不会达到 0。一个可能的解决方法是让监视器在所有工作人员完成后关闭通道。如果 for 循环在 main 中,它将结束循环,从 main 返回,并结束程序。
例如:http ://play.golang.org/p/nai7XtTMfr
package main
import (
"fmt"
"strconv"
"sync"
)
func worker(wg *sync.WaitGroup, cs chan string, i int) {
defer wg.Done()
cs <- "worker" + strconv.Itoa(i)
}
func monitorWorker(wg *sync.WaitGroup, cs chan string) {
wg.Wait()
close(cs)
}
func main() {
wg := &sync.WaitGroup{}
cs := make(chan string)
for i := 0; i < 10; i++ {
wg.Add(1)
go worker(wg, cs, i)
}
go monitorWorker(wg, cs)
for i := range cs {
fmt.Println(i)
}
}
编辑:这是对 OP 第一条评论的回答。
您的程序包含三个需要同步的部分。首先,您的所有工作人员都需要发送数据。然后您的打印循环需要打印该数据。然后你的主函数需要返回从而结束程序。在您的示例中,所有工作人员都发送数据,所有数据都被打印,但是消息永远不会发送到 main ,它应该正常返回。
在我的示例中,main 进行打印,而“monitorWorker”只是告诉 main 何时接收到它需要打印的每条数据。这样程序就会优雅地结束而不是死锁。
如果你坚持打印循环在另一个 goroutine 中,你可以这样做。但是随后需要向 main 发送额外的通信,以便它返回。在下一个示例中,我使用通道来确保打印所有数据时的主端。
package main
import (
"fmt"
"strconv"
"sync"
)
func worker(wg *sync.WaitGroup, cs chan string, i int) {
defer wg.Done()
cs <- "worker" + strconv.Itoa(i)
}
func monitorWorker(wg *sync.WaitGroup, cs chan string) {
wg.Wait()
close(cs)
}
func printWorker(cs <-chan string, done chan<- bool) {
for i := range cs {
fmt.Println(i)
}
done <- true
}
func main() {
wg := &sync.WaitGroup{}
cs := make(chan string)
for i := 0; i < 10; i++ {
wg.Add(1)
go worker(wg, cs, i)
}
go monitorWorker(wg, cs)
done := make(chan bool, 1)
go printWorker(cs, done)
<-done
}