Go 语言channel的应用场景及使用技巧

        通过反映的方式执行 select 语句。这在处理有很多 case 子句,尤其是不定长 case 子句的情况时非常有用。

1. 使用反射操作 select 和 channel

        使用 select 语句可以处理 chan 的 send 和 recv, send 和 recv 都可以作为 case 子句。如果需要同时处理两个 chan, 则可以写成下面的样子:

select {
	case v:= <-ch1:
		fmt.Println(v)
	case v := <-ch2:
		fmt.Println(v)
}

或者,一个 chan 用于发送,另一个 chan 用地接收:

select {
	case v:= <-ch1:
		fmt.Println(v)
	case v -> ch2:
		fmt.Println(v)
}

         如果需要处理三个 chan,则可以再添加一个 case 子句,用它来处理第三个 chan; 如果需要处理四个 chan, 那么就再添加一个 case 子句。可是,如果要处理 100 个 chan、1000 个 chan 呢?

        或者,chan 的数量在编译时是不定的,在运行时需要处理一组 channel 时,也没有办法在代码中写成 select 语句。那该怎么办?

        这个时候,就要 “祭” 出反射大法了。

        通过 reflect.Select 函数,可以传入一组运行时的 case 子句,当作参数执行。Go 的 select 是伪随机的,它可以在执行的 case 中随机选择一个 case ,并返回这个 case 的索引( chosen)。如果没有可用的 case,则会返回一个 bool 类型的值,这个值用来表示是否有 case 被成功选择。如果是 recv case, 还会返回所接收的元素。Select 函数的签名如下:

func Select(cases []SelectCase) (chosen int,recv Value,recvOK bool)

        下面我们通过一个例子来演示动态处理两个 chan 的情形。因为可以动态处理 case 数据,所以可以传入成千上万个 chan,这就解决了不能动态处理 n 个 chan 的问题。

        首先,createCases 函数分别为每个 chan 生成了 recv case 和 send case,并返回一个 reflect.SelectCase 数组。

        然后,通过一个循环 10 次的 for 循环执行 reflect.Select, 这个函数会从 cases 中选择一个 case 执行。第一次选择的肯定是 send case,因为此时 chan 中还没有元素,recv 还不可用。等 chan 中有了元素以后,就可以选择 recv case 了。这样一来,我们就可以处理不定数量的 chan 了。

func main(){
	var ch1 = make(chan int,10)
	var ch2 = make(chan int,10)
	
	// 创建 SelectCase
	var cases = createCases(ch1,ch2)
	
	// 执行 10 次 select
	for i:=0; i<10;i++ {
		chosen,recv,ok:=reflect.Select(cases)
		if recv.IsValid(){// recv case
			fmt.Println("recv:",cases[chosen].Dir,recv,ok)
		} else { // send case
			fmt.Println("send:",cases[chosen].Dir,ok)
		}
	}
}

//利用反射创建 case
func createCases(chs ...chan int)[]reflect.SelectCase{
	var cases []reflect.SelectCase
	
	// 创建recv case
	for _, ch:= range chs {
		cases = append(cases,reflect.SelectCases{
			Dir : reflect.SelectRecv,
			Chan: reflect.ValueOf(ch),
		})
	}
	
	//创建 send case
	for i,ch := range chs {
		v := reflect.ValueOf(i)
		cases = append(cases,reflect.SelectCase{
			Dir: reflect.SelectSend,
			Chan:
  • 21
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Mindfulness code

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值