Golang–Directional Channel(定向通道)

• 2021 年 1 月 31 日
• 筆記
• golang

Directional Channel

通道可以是定向的(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
			}
		}
	}()