Golang中不要使用无限循环检查goroutine是否完成工作

在使用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() {
	ch := make(chan string)
	go foo(ch)
	<-ch //main():OK，收到你的消息了~
	for !flag {
	}
	fmt.Println(str)
}

如果转载请注明出处：http://blog.csdn.net/gophers/article/details/24472891

有疑问加站长微信联系（非本文作者）