在使用goroutine时,我们经常会写出这样的代码:
package main
import (
"fmt"
)
var (
flag bool
str string
)
func foo() {
flag = true
str = "setup complete!"
}
func main() {
go foo()
for !flag {
//按照我们的本意,foo()执行完毕后,flag=true,循环就会退出。
//但是其实这个循环永远都不会退出
}
fmt.Println(str)
}
运行之后发现main中的无限循环永远也无法退出,所以Go中不要用这种无限轮询的方式来检查goroutine是否完成了工作。
我们可以通过使用channel,让foo()和main()实现通信,让foo()执行完毕后通过channel发送一个消息给main(),告诉它自己的事儿完成了,然后main()收到消息后继续执行其他操作:
package main
import (
"fmt"
)
var (
flag bool
str string
)
func foo(ch chan string) {
flag = true
str = "setup complete!"
ch <- "I'm complete." //foo():我的任务完成了,发个消息给你~
}
func main() {