可以看看我这篇,收录于:Go 基础系列
学到什么
select
并发与并行
为了更有意思的解释这个概念,我借用知乎上的一个回答:
- 你吃饭吃到一半,电话来了,你一直到吃完了以后才去接,这就说明你不支持并发也不支持并行。
- 你吃饭吃到一半,电话来了,你停了下来接了电话,接完后继续吃饭,这说明你支持并发。
- 你吃饭吃到一半,电话来了,你一边打电话一边吃饭,这说明你支持并行。
并发的关键是你有处理多个任务的能力,不一定要同时。
并行的关键是你有同时处理多个任务的能力。
对应到 CPU 上,如果是多核它就有同时执行的能力,即有并行的能力。
对于 Go 语言,它自行安排了我们的代码合适并发合适并行。
什么是 Goroutine
学会这个就知道怎么写一个并发程序,用起来很简单的,现在开始。
Goroutine 是 Go 语言中的协程,其它语言称为的协程字面上叫 Coroutine,简单理解下就是比线程更轻量的一个玩意。
再说白了,就是可以异步执行函数。
main Goroutine
当启动 main 入口函数时,后台就自动跑了一个 main Goroutine,还原给大家看看。
package main
func main() {
panic("看这里")
}
执行上面代码,会输出如下部分信息:
panic: 看这里
goroutine 1 [running]:
main.main()
goroutine
创建 Goroutine
go
go fun1(...)
也支持匿名函数。
go func(...){
// ...
}(...)
go
下来看一个完整的例子:
package main
import (
"fmt"
)
func PrintA() {
fmt.Println("A")
}
func main() {
go PrintA()
fmt.Println("main")
}
main
- 第一行:创建一个 Goroutine,异步打印“A”字符串。
- 第二行:打印 “main” 字符串。
现在先停留一会,想想执行该代码后,输出结果是啥。
结果如下:
main
你没看错,没有输出“A”字符串。
go PrintA()mainmain
time.Sleep
func main() {
go PrintA()
fmt.Println("main")
time.Sleep(time.Second)
}
// 输出
main
A
main
selectselect
func main() {
go PrintA()
fmt.Println("main")
select {}
}
// 输出
main
A
fatal error: all goroutines are asleep - deadlock!
...
"A"字符串是输出了,但程序也出现异常了。
select{}
select{}
- 爬虫项目,创建了 Goroutine,需要一直爬取数据,不需要停止。
WaitGroup
package main
import (
"fmt"
"sync"
)
var wg sync.WaitGroup
func PrintA() {
fmt.Println("A")
wg.Done()
}
func main() {
wg.Add(1)
go PrintA()
wg.Wait()
fmt.Println("main")
}
WaitGroupwgwg.Add(1)Add(2)wg.Done()wg.Wait()
控制并发数
Go 语言中可以控制使用 CPU 的核心数量,从 Go1.5 版本开始,默认设置为 CPU 的总核心数。如果想自定义设置,使用如下函数:
num := 2
runtime.GOMAXPROCS(num)
num 如果大于 CPU 的核心数,也是允许的,Go 语言调度器会将很多的 Goroutine 分配到不同的处理器上。
什么是通道
现在明白了怎么创建 Goroutine 后,下一步就要知道它们之间要如何通信。

Goroutine 通信使用“通道(channel)”,如果 Goroutine1 想发送数据给 Goroutine2,就把数据放到通道里,Goroutine2 直接从通道里拿就行,反过来也是一样。
在给通道放数据时,也可以指定通道放置的数据类型。
创建通道
创建通道时,分为无缓冲和有缓冲两种。
1. 无缓冲
strChan := make(chan string)
string
allChan := make(chan interface{})
创建好了通道,下来就要给通道里放数据。
strChan := make(chan string)
strChan <- "老苗"
strChan
strChan
如果想解决死锁情况,就要保证有地方在异步读通道,因此需要创建一个 Goroutine 来负责。
例子如下:
// concurrency/channel/main.go
package main
import (
"fmt"
"sync"
)
var wg sync.WaitGroup
func Read(strChan chan string) {
data := <-strChan
fmt.Println(data)
wg.Done()
}
func main() {
wg.Add(1)
strChan := make(chan string)
go Read(strChan)
strChan <- "老苗"
wg.Wait()
}
// 输出
老苗
Read<-strChanwg.Wait()Read
2. 有缓冲
读了上面就会了解到,对于无缓冲通道,它会产生阻塞。为了不让阻塞,必须创建一个 Goroutine 负责从通道读取才行。
而有缓冲的通道,会有缓冲的余地,具体来看看。
创建缓冲通道,如下:
bufferChan := make(chan string, 3)
- 创建了一个存储数据类型为 string 的通道。
- 可以缓冲 3 个数据,即给通道送入 3 个数据不会进行阻塞。
测试如下:
// // concurrency/bufferchannel/main.go
package main
import "fmt"
func main() {
bufferChan := make(chan string, 3)
bufferChan<-"a"
bufferChan<-"b"
bufferChan<-"c"
fmt.Println(<-bufferChan)
}
// 输出
a
bufferChan
缓冲通道何时使用,例如:
爬虫数据,第 1 个 Goroutine 负责爬取数据,第 2 个 Goroutine 负责处理和存储数据。 当第 1 个的处理速度大于第 2 个时,可以使用缓冲通道暂存起来。
暂存起来后,第 1 个 Goroutine 就可以继续爬取,而不像无缓冲通道,放入数据时会阻塞,直到通道数据被读出,才能进行。
为了加深印象,再来一张图:

图解:
bufferChanbufferChan
单向通道
现在知道了如何创建一个双向通道,双向通道指的就是即可以存,又可以取。
那单向通道创建如下:
readChan := make(<-chan string)
writeChan := make(chan<- string)
readChanwriteChan
但这样创建的通道是无法传递数据的,为什么?
因为,如果只能读的通道,没法存数据,那我存了个寂寞。而存的通道,我数据拿不出来,又有何用。
现在看看如何正确使用单向通道的例子,如下:
// concurrency/onechannel/main.go
package main
import (
"fmt"
"sync"
)
var wg sync.WaitGroup
// 写通道
func write(data chan<- int) {
data<-520
wg.Done()
}
// 读通道
func read(data <-chan int) {
fmt.Println(<-data)
wg.Done()
}
func main() {
wg.Add(2)
dataChan := make(chan int)
go write(dataChan)
go read(dataChan)
wg.Wait()
}
// 输出
520
readwrite
遍历通道
在实际项目中,通道里会产生大量的数据,这时候就要循环的从通道里读取。
现在改写单向通道写入数据的例子:
func write(data chan<- int) {
for i := 0; i < 10; i++ {
data<-i
}
wg.Done()
}
这段代码是给通道里循环写入数字。
下来使用两种方式循环读取通道数据。
1. 死循环
func read(data <-chan int) {
for {
d := <-data
fmt.Println(d)
}
wg.Done()
}
使用死循环读取数据,但这个有个问题,什么时候退出 for 循环?
readclose
func write(data chan<- int) {
// ...
close(data)
wg.Done()
}
关闭后,读取通道时也需要检测判断。
func read(data <-chan int) {
for {
d, ok := <-data
if !ok {
break
}
fmt.Println(d)
}
wg.Done()
}
okok
ch := make(chan string, 1)
ch <- "a"
close(ch)
val, ok := <-ch
fmt.Println(val, ok)
val1, ok1 := <-ch
fmt.Println(val1, ok1)
// 输出
a true
false
2. for-range
也可以使用 for-range 语句读取通道,这比死循环使用起来简单一点。
func read(data <-chan int) {
for d := range data{
fmt.Println(d)
}
wg.Done()
}
- 如果想退出 for-range 语句,也需要关闭通道。
- 如果关闭通道后,不需要增加 ok 判断,自行会退出。
通道函数
lencap
ch := make(chan int, 3)
ch<-1
fmt.Println(len(ch))
fmt.Println(cap(ch))
// 输出
1
3
select 语句
selectselect
selectswitchcasedefaultselect
casecasefallthrough
1. 无 default 分支
selectcase
正确例子:
// concurrency/select/main.go
package main
import "fmt"
func main() {
ch1 := make(chan int, 1)
ch2 := make(chan int, 1)
ch1 <- 1
select {
case v, ok := <-ch1:
if ok {
fmt.Println("ch1通道", v)
}
case v, ok := <-ch2:
if ok {
fmt.Println("ch2通道", v)
}
}
}
// 输出
ch1通道 1
ch1casecase ch2<-2ch1 <- 1select
fatal error: all goroutines are asleep - deadlock!
goroutine 1 [select]:
main.main()
C:/workspace/go/src/gobasic/cocurrency/select/main.go:9 +0xe7
2. 有 default 分支
selectdefaultcasedefault
// ...
func main() {
ch1 := make(chan int, 1)
ch2 := make(chan int, 1)
select {
case v, ok := <-ch1:
if ok {
fmt.Println("ch1通道", v)
}
case v, ok := <-ch2:
if ok {
fmt.Println("ch2通道", v)
}
default:
fmt.Println("没有可读写通道")
}
}
// 输出
没有可读写通道
总结
这节课很关键,也是很容易出现问题的地方,我再针对重点的重点强调一下:
goWaitGroup
最后,再揣摩一句话,不要用共享内存来通信,要用通信来共享内存。
如果喜欢,可以关注 @潇洒哥和黑大帅