代码演示
package main
import "fmt"
func main() {
jobs := make(chan int, 5)
done := make(chan bool)
go func() {
for {
j, more := <-jobs
if more {
fmt.Println("received job", j)
} else {
fmt.Println("received all jobs")
done <- true
return
}
}
}()
for j := 1; j <= 3; j++ {
jobs <- j
fmt.Println("sent job", j)
}
close(jobs)
fmt.Println("sent all jobs")
<-done
}
代码运行结果
sent job 1
sent job 2
sent job 3
sent all jobs
received job 1
received job 2
received job 3
received all jobs
代码解读
- 将一个通道关闭的话，会给接收方传递一个完成工作的信息，这个特性是很有用的
- 以上例子，创建了一个jobs通道，并且通过这个通道，给go协程传递任务信息
- 当jobs通道再没有任务信息给go协程传递时候，我们close掉这个通道
- 在go协程中，我们利用more这个值来判断通道是否关闭，如果关闭了，那么more的值为false，并且通知给通道done