您的位置:首页 > 科技 > IT业 > Go 语言channel的应用场景及使用技巧

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

2024/12/23 16:47:16 来源:https://blog.csdn.net/canglonghacker/article/details/141142293  浏览:    关键词: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)// 创建 SelectCasevar cases = createCases(ch1,ch2)// 执行 10 次 selectfor i:=0; i<10;i++ {chosen,recv,ok:=reflect.Select(cases)if recv.IsValid(){// recv casefmt.Println("recv:",cases[chosen].Dir,recv,ok)} else { // send casefmt.Println("send:",cases[chosen].Dir,ok)}}
}//利用反射创建 case
func createCases(chs ...chan int)[]reflect.SelectCase{var cases []reflect.SelectCase// 创建recv casefor _, ch:= range chs {cases = append(cases,reflect.SelectCases{Dir : reflect.SelectRecv,Chan: reflect.ValueOf(ch),})}//创建 send casefor i,ch := range chs {v := reflect.ValueOf(i)cases = append(cases,reflect.SelectCase{Dir: reflect.SelectSend,Chan:

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com