Golang實(shí)現(xiàn)Directional Channel的方法

這篇文章給大家分享的是有關(guān)Golang實(shí)現(xiàn)Directional Channel的方法的內(nèi)容。小編覺得挺實(shí)用的，因此分享給大家做個(gè)參考，一起跟隨小編過(guò)來(lái)看看吧。

通道可以是定向的( directional )。在默認(rèn)情況下，通道將以雙向的( bidirectional )形式運(yùn)作，用戶既可以把值放人通道，也可以從通道取出值;但是，通道也可以被限制為只能執(zhí)行發(fā)送操作( send-only )或者只能執(zhí)行接收操作( receive-only )。

通常可以叫 定向通道 ，也有人叫 單向通道 ，兩者其實(shí)都是指向這篇短文要討論的 Directional Channel 。

下面直接舉例子說(shuō)明：

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()
}

這里可以先猜一下運(yùn)行結(jié)果，我直接貼出：

后面的引起了我的注意：

receive
0

是最讓我困惑的地方，因?yàn)樵谝话阌∠髢?nèi) channel 都是會(huì)阻塞的存在，而從上面結(jié)果看，第一個(gè) recv 也是阻塞了）。但是后面的的 recv 卻沒有起到阻塞的作用，還是把 c 中的默認(rèn)值 0 輸出了。

通過(guò) debug ，不難看出的是，兩個(gè)單向通道的發(fā)送和接收都是從一個(gè)緩沖通道中進(jìn)行操作的，兩個(gè)單項(xiàng)通道同屬于一個(gè)雙向通道，只不過(guò)他們被分別限制了接收和發(fā)送功能。

通過(guò) debug ，再次發(fā)現(xiàn)，在 recv 讀取完之后，通道內(nèi)并沒有被 “清空” ，而是保有默認(rèn)值。

難道， recv 并沒有被阻塞？

驗(yàn)證方法如下：

// defer close(c)

選擇注釋掉發(fā)送 Goroutine 中的關(guān)閉通道。

結(jié)果很明顯，造成死鎖，證明兩個(gè)單向通道之間是有通信的，有阻塞的。

最后通過(guò)源碼，還是找到了答案：

由此，為避免接收到不需要的無(wú)效 零值 ，應(yīng)修改接收 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
			}
		}
	}()

感謝各位的閱讀！關(guān)于“Golang實(shí)現(xiàn)Directional Channel的方法”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，讓大家可以學(xué)到更多知識(shí)，如果覺得文章不錯(cuò)，可以把它分享出去讓更多的人看到吧！