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

 

到此这篇关于Golang实现Directional Channel(定向通道)的文章就介绍到这了,更多相关Golang定向通道内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！