go-reflect.Select使用

type SelectCase struct {
Dir SelectDir // direction of case
Chan Value // channel to use (for send or receive)
Send Value // value to send (for send)
}
SelectCase描述了select操作中的一种case。
case种类取决于Dir

如果Dir是SelectDefault,代表case中的default case,
Chan和Send都必须为0值。

如果Dir是SelectSend,代表send case,
则Chan的隐含值必须是channel,Send的隐含值必须可赋值给channel的元素类型。
特殊情况下,如果Chan为0值,则该case被忽略,Send也被忽略,send可能为0值或非0值。

如果Dir是SelectRecv,代表recv case。
则Chan的隐含值必须是channel,Send为0值。
如果Chan为0值,该case被忽略,Send必须也为0值。
当一个recv case被选中,receive到的值被Select返回。

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

select函数接收electCase列表作为参数,直到列表中某个case发生才返回。
返回case的index索引,如果是recv case也返回Value和recvOK标志。

总结:使用chan构建select case时,如果是send,则Dir为reflect.SelectSend, Chan为此Chan,Send参数为要发送的值;如果是recv,则Dir为reflect.SelectRecv, Chan为此Chan,Send不赋值。

// createCases 将多个chan作为case clause
func createCases(chs ...chan int) []reflect.SelectCase {
	var cases []reflect.SelectCase

	// 添加recv case
	for _, ch := range chs {
		cases = append(cases, reflect.SelectCase{
			Dir: reflect.SelectRecv,
			Chan: reflect.ValueOf(ch), // Chan字段使用反射
		})
	}

	for i, ch := range chs {
		v := reflect.ValueOf(i + 100)
		cases = append(cases, reflect.SelectCase{
			Dir: reflect.SelectSend,
			Chan: reflect.ValueOf(ch),
			Send: v,
		})
	}

	return cases
}

func main() {
ch1 := make(chan int, 50)
	ch2 := make(chan int, 50)

	cases := createCases(ch1, ch2)
	ch1 <- 100
	ch2 <- 200

	close(ch1)
	close(ch2)

	var wg sync.WaitGroup
	wg.Add(1)
	go func() {
		defer wg.Done()
		for {
			chosen, recv, ok := reflect.Select(cases)
			if recv.IsValid() { // recv case
				if ok {
					fmt.Printf("recv data case from index[%v], Value: %v, ok: %v\n", chosen, recv, ok)
				}else{
					fmt.Printf("recv close case from index[%v], Value: %v, ok: %v\n", chosen, recv, ok)
					// 已close的chan的receive和send都移除
				}
			}else {				// send case
				//fmt.Println("send:", chosen, recv, ok)
				fmt.Printf("send case from index[%v], Value: %v, ok: %v\n", chosen, recv, ok)
			}
		}
	}()

	wg.Wait()
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值