Go无缓冲通道的陷阱

定义

Channel是go的特色之一,甚至说是最大的特色也不为过,使用起来也非常简单。
首先定义一个int类型的channel:

ch1 := make(chan int) // 无缓冲通道
ch2 := make(chan int, 10) // 带缓冲的通道

我们这里主要关注无缓冲通道。

场景

来看看这段代码:

package main

import (
    "sync"
    "fmt"
)

func main() {
    wg := sync.WaitGroup{}

    ch1 := make(chan int)

    wg.Add(1)

    for i := 0; i < 10; i++ {
        ch1 <- i
    }

    go func1(ch1, &wg) 

    wg.Wait()

    close(ch1)
    fmt.Println("Close channel: ", ch1)
}

func func1(ch chan int, wg *sync.WaitGroup) {
    FOR:
    for {
        select {
        case i, ok := <- ch:
            if !ok {
                fmt.Println("1 chan closed, returning")
                break FOR
            } else {
                fmt.Println("1 Got number: ", i)
                if i == 9 {
                    break FOR
                }
            }
        default:
            fmt.Println("1 Got nothing")
        }
    }
    wg.Done()
}

乍眼一看似乎没毛病,但是当运行程序的时候:

fatal error: all goroutines are asleep - deadlock!

goroutine 1 [chan send]:
main.main()
    /Users/bruce/Code/go/src/GoCommonServices/sync-demo/demo.go:19 +0xb9
exit status 2

Process finished with exit code 1

为什么呢? 仔细看了看上面的程序,在定义了无缓冲通道ch1之后,立马向其中写入数据:

    ch1 := make(chan int)

    for i := 0; i < 10; i++ {
        ch1 <- i
    }

但此时并没有消费者,而无缓冲通道在写入一个数据之后,会等待消费者消费,程序阻塞,但启动消费者的代码:

    go func1(ch1, &wg) 

恰好在for循环之后,所以这个goroutine永远没有启动的机会,这就是报错信息提示的,deadlock了,要修复这个有两种方法:

1 ch1定义为缓冲通道,足够容纳for中的数据,就不会阻塞
    ch1 := make(chan int 10)
2 先启动消费者,再向通道中写数据
    go func1(ch1, &wg) // 先启动消费者,再向ch1中写入数据,以防阻塞

    ch1 := make(chan int)

    for i := 0; i < 10; i++ {
        ch1 <- i
    }

小结

归根结底,还是因为channel的特性:

无缓冲的channel,不管是入还是出,都会阻塞,所以在同一个goroutine中,不能同时对同一个无缓冲channel进行入和出操作;
带缓冲的channel,在队列满之前,不会阻塞;队列满之后,依然会阻塞。

channel无疑是go并发程序开发的利器,但使用的时候还是需要仔细慎重,注意避免像本文提到的这些『陷阱』。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值