当前位置 博文首页 > Golang实现Directional Channel(定向通道)

    Golang实现Directional Channel(定向通道)

    作者:L1ng14 时间:2021-05-01 18:01

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

     

    js
    下一篇:没有了