在 go 中经常会使用 channel,进行并发执行子任务,提高执行效率。但一不小心就会踩到 deadlock 的坑,本文就来解析一下常见的死锁形式和解决方式。
1. 直接读取空 chan 产生死锁
package main
import (
"fmt"
)
func main() {
ch := make(chan int, 3)
<-ch
}
输出结果:
fatal error: all goroutines are asleep - deadlock!
goroutine 1 [chan receive]:
main.main()
/home/work/code/golang/src/interview/go/deadlock/test.go:9 +0x56
Process finished with exit code 2
解决方式:
采用 select case default 阻塞默认处理方式。
package main
import (
"fmt"
)
func main() {
ch := make(chan int, 3)
select {
case v := <-ch:
fmt.Println(v)
default:
fmt.Println("chan no data")
}
}
2. 阻塞 channel 产生死锁
package main
import "fmt"
func main() {
ch := make(chan int)
ch <- 1 // 无缓冲在此写入数据,却没有读数据,阻塞住
fmt.Println(<-ch) // 被上面阻塞,无法被执行到
}
输出结果:
fatal error: all goroutines are asleep - deadlock!
goroutine 1 [chan send]:
main.main()
/home/work/code/golang/src/interview/go/deadlock/test02.go:8 +0x59
Process finished with exit code 2
解决方式:
a. 采用开启子协程方式,保证读写 chan 成对存取数据;
package main
import "fmt"
func main() {
ch := make(chan int)
go func() {
ch <- 1 // 开启子goroutine写入数据
}()
fmt.Println(<-ch) // 阻塞住,一旦ch有数据,则读取成功
}
b. 采用有缓冲 chan,在容量范围内不会阻塞;
package main
import "fmt"
func main() {
ch := make(chan int, 1)
ch <- 1
fmt.Println(<-ch)
}
3. 有缓冲 chan 超过容量时产生死锁
package main
import (
"fmt"
)
func main() {
ch := make(chan int, 3)
ch <- 1
ch <- 2
ch <- 3
ch <- 4 // 超过最大容量,阻塞main协程,产生deadlock
for v := range ch {
fmt.Println(v)
}
}
解决方式:
a. 增加缓冲容量,保证能满足写入所有数据;
b. 采用 select case default 阻塞默认处理方式(demo略);
4. for range 产生死锁
package main
import (
"fmt"
)
func main() {
ch := make(chan int, 3)
ch <- 1
ch <- 2
ch <- 3
// range 一直读取直到chan关闭,否则产生阻塞死锁
for v := range ch {
fmt.Println(v)
}
}
输出结果:
1
2
3
fatal error: all goroutines are asleep - deadlock!
goroutine 1 [chan receive]:
main.main()
/home/work/code/golang/src/interview/go/deadlock/test04.go:15 +0x115
Process finished with exit code 2
解决方式:
a. 显式关闭 channel;
b. 开启子协程,主协程 sleep 等待时间后退出;
package main
import (
"fmt"
"time"
)
func main() {
ch := make(chan int, 3)
ch <- 1
ch <- 2
ch <- 3
close(ch) // 解决方式1:关闭chan
// range 一直读取直到chan关闭,否则产生阻塞死锁
// 解决方式2:开启子协程,主协程sleep等待
go func() {
for v := range ch {
fmt.Println(v)
}
}()
time.Sleep(1e9)
}
【小结】
1. 日常在使用 channel 中,要注意区分有缓冲(buffered channel,异步队列-FIFO处理)与无缓冲(unbuffered channel,同步流入流出)通道的区别,掌握各自适合使用的方式;
2. 出现deadlock一定是线程/协程之间存在了资源竞争,互相占用对方需要的资源导致程序永远不能退出,需要小心可能遇到的坑,也可以通过加锁避免。