(编辑:jimmy 日期: 2024/12/24 浏览:2)
通道可以是定向的( directional
)。在默认情况下,通道将以双向的( bidirectional
)形式运作,用户既可以把值放人通道,也可以从通道取出值;但是,通道也可以被限制为只能执行发送操作( send-only
)或者只能执行接收操作( receive-only
)。
通常可以叫 定向通道 ,也有人叫 单向通道 ,两者其实都是指向这篇短文要讨论的 Directional Channel
。
下面直接举例子说明:
package onlyChannelTest import ( "fmt" "math/rand" "sync" "testing" "time" ) func TestOnlyChannel(t *testing.T) { var wg sync.WaitGroup wg.Add(2) c:= make(chan int, 3) var send chan<- int = c var recv <-chan int = c go func() { defer wg.Done() for i := 0; i < 6; i++ { // 注意这里是 6 fmt.Print("receive\n") fmt.Println(<-recv) } }() go func() { defer wg.Done() defer close(c) rand.Seed(time.Now().UTC().UnixNano()) for i := 0; i < 3; i++ { // 这里是 3 fmt.Print("send\n") send <- rand.Intn(100) } }() wg.Wait() }
这里可以先猜一下运行结果,我直接贴出:
后面的引起了我的注意:
receive
0
是最让我困惑的地方,因为在一般印象内 channel
都是会阻塞的存在,而从上面结果看,第一个 recv
也是阻塞了)。但是后面的的 recv
却没有起到阻塞的作用,还是把 c
中的默认值 0
输出了。
通过 debug
,不难看出的是,两个单向通道的发送和接收都是从一个缓冲通道中进行操作的,两个单项通道同属于一个双向通道,只不过他们被分别限制了接收和发送功能。
通过 debug
,再次发现,在 recv
读取完之后,通道内并没有被 “清空” ,而是保有默认值。
难道, recv
并没有被阻塞?
验证方法如下:
// defer close(c)
选择注释掉发送 Goroutine
中的关闭通道。
结果很明显,造成死锁,证明两个单向通道之间是有通信的,有阻塞的。
最后通过源码,还是找到了答案:
由此,为避免接收到不需要的无效 零值 ,应修改接收 Goroutine
-- receive
的写法:
go func() { defer wg.Done() for i := 0; i < 6; i++ { // 注意这里是 6 x, ok := <-recv if ok { fmt.Print("receive\n") fmt.Println(x) }else{ break } } }()