golang通过协程、通道实现并发按特定顺序打印字符串

背景：

看到这样一个golang的面试题，大致是：

有三个函数分别打印，“dog”，“cat”，“fish”，

要求每个函数起一个goroutine，请按照dog，cat，fish的顺序，打印四次，输出到控制台。

思考：

其实这道题，就是在考察两点：

1.goroutine和channel的配合，使用来控制多个协程的执行顺序；

2.如何控制协程执行的次数；（sync包）

package main

import (
	"fmt"
	"sync"
)

func main(){
	wg:=sync.WaitGroup{}
	wg.Add(12)  //打印四组，三个goroutine需要执行4次，3*4表示处于等待goroutine的数量

	dogChan:=make(chan bool,1)	//此处如果申明的是无缓冲的通道，那么会在16行代码处处于阻塞状态！原因：通道中无数据，向通道写数据，但无协程读取。
	catChan:=make(chan bool)
	fishChan:=make(chan bool)

	dogChan<-true

	for i:=0;i<4;i++{        //打印四组，三个goroutine需要执行4次
		go dog(&wg,dogChan,catChan)
		go cat(&wg,catChan,fishChan)
		go fish(&wg,fishChan,dogChan)
	}

	wg.Wait()
	fmt.Println("main finished!")
}

func dog(wg *sync.WaitGroup,dogChan chan bool,catChan chan bool){
	if ok:=<-dogChan;ok {
		fmt.Println("dog")
		wg.Done()
		catChan<-true
	}
}

func cat(wg *sync.WaitGroup,catChan chan bool,fishChan chan bool){
	if ok:=<-catChan;ok {
		fmt.Println("cat")
		wg.Done()
		fishChan<-true
	}
}

func fish(wg *sync.WaitGroup,fishChan chan bool,dogChan chan bool){
	if ok:=<-fishChan;ok {
		fmt.Println("fish")
		wg.Done()
		dogChan<-true
	}
}

输出：
dog
cat
fish
dog
cat
fish
dog
cat
fish
dog
cat
fish
main finished!

Process finished with exit code 0

仔细看，发现上面不符合题意，实际上起了12个协程！控制每个协程的执行顺序，实现了4组dog,cat,fish的输出。

如果要限制协程为3个，怎么搞？

package main

import (
	"fmt"
	"sync"
)


func PrintCat(dogChan,catChan chan bool){
	defer wg.Done()
	defer close(catChan)
	for i:= 0; i <4;i++{
		<-dogChan
		fmt.Println("cat ...")
		catChan <- true
	}
}

func PrintDog(fishChan,dogChan chan bool){
	defer wg.Done()
	defer close(dogChan)
	for i:= 0; i <4;i++{
		<-fishChan
		fmt.Println("dog ...")
		dogChan<-true
	}
}

func PrintFish(catChan,fishChan chan bool){
	defer wg.Done()
	defer close(fishChan)
	for i:= 0; i <4;i++{
		<-catChan
		fmt.Println("fish ...")
		fishChan<-true
	}
}


var wg sync.WaitGroup

func main() {
	dogChan := make(chan bool,1)
	catChan := make(chan bool,1)
	fishChan := make(chan bool,1)
	fishChan <- true

	go PrintDog(fishChan,dogChan)
	go PrintFish(catChan,fishChan)
	go PrintCat(dogChan,catChan)
	
	wg.Add(3)
	wg.Wait()

}

 

参考文档：

https://segmentfault.com/a/1190000017537297