今天遇到一道面试题,开启N个协程,并交替打印1-100如给定N=3则输出:
goroutine0: 0
goroutine1: 1
goroutine2: 2
goroutine0: 3
goroutine1: 4
面试时没答案,虽过后研究参考了一些网上方法,并记录下来,先上代码
func print() { chanNum := 3 // chan 数量 chanQueue := make([]chan int, chanNum) // 创建chan Slice var result = 0 // 值 exitChan := make(chan bool) // 退出标识 for i := 0; i < chanNum; i++ { // 创建chan chanQueue[i] = make(chan int) if i == chanNum-1 { // 给最后一个chan写一条数据,为了第一次输出从第1个chan输出 go func(i int) { chanQueue[i] <- 1 }(i) } } for i := 0; i < chanNum; i++ { var lastChan chan int // 上一个goroutine 结束才能输出 控制输出顺序 var curChan chan int // 当前阻塞输出的goroutine if i == 0 { lastChan = chanQueue[chanNum-1] } else { lastChan = chanQueue[i-1] } curChan = chanQueue[i] go func(i int, lastChan, curChan chan int) { for { if result > 100 { // 超过100就退出 exitChan <- true } // 一直阻塞到上一个输出完,控制顺序 <-lastChan fmt.Printf("thread%d: %d \n", i, result) result = result + 1 // 当前goroutine已输出 curChan <- 1 } }(i, lastChan, curChan) } <-exitChan fmt.Println("done")}
第一个for循环创建chan第二个for循环里的lastChan意思是,当前chan如果要打印数据,就必须得上一个chan打印完后才能打印。这里假设N=2,chan索引为0,1,当索引1要输出,就阻塞到索引0的chan有数据为止,当自己打印完后往自己的chan中发送一个1,方便给依赖自己的chan 解除阻塞。这里有个特殊的地方,当索引为0时,他的依赖索引chan就为chanQueue的长度-1,如果没有在创建Chan中的时候没有下面这一串代码就会造成死锁
if i == chanNum-1 { // 给最后一个chan写一条数据,为了第一次输出从第1个chan输出 go func(i int) { chanQueue[i] <- 1 }(i) }
总结 以上是内存溢出为你收集整理的go 让N个协程交替打印1-100全部内容,希望文章能够帮你解决go 让N个协程交替打印1-100所遇到的程序开发问题。
如果觉得内存溢出网站内容还不错,欢迎将内存溢出网站推荐给程序员好友。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)