Go 面试题

Golang 面试题 (4) N个协程交替打印1-100
启动N个协程，共用一个外部变量计数器，计数器范围是1到100
开启N个有缓冲chan，chans[i]塞入数据代表协程i可以进行打印了，打印的数字就是计数器的数
协程i一直阻塞，直到chan[i]通道有数据可以拉，才打印

package main

import (
	"fmt"
	"sync"
)

//N个协程交替打印数字，到达一个数值后终止
//
//解法:
//启动N个协程，共用一个外部变量计数器，计数器范围是1到100
//开启N个无缓冲chan，chans[i]塞入数据代表协程i可以进行打印了，打印的数字就是计数器的数
//协程i一直阻塞，直到chan[i]通道有数据可以拉，才打印
//

var counter = 0

const CounterMaxValue = 100

func DoJob(GoRoutineNum int) {

	//用于控制业务逻辑
	sigChannel := make([]chan struct{}, GoRoutineNum)

	//用于控制协成是否终结
	quitSigChannel := make([]chan struct{}, GoRoutineNum)

	for i := 0; i < (GoRoutineNum); i++ {
		sigChannel[i] = make(chan struct{})
		quitSigChannel[i] = make(chan struct{})
	}

	mainRoutine := make(chan struct{})

	wg := sync.WaitGroup{}

	for i := 0; i < GoRoutineNum; i++ {
		wg.Add(1)
		go func(idx int) {
			defer wg.Done()
			//FOR:
			for {
				select {
				case <-quitSigChannel[idx]:
					//break FOR
					goto END
				case <-sigChannel[idx]:
					fmt.Printf("goroutine-num:%d %d\n", idx, counter)
					counter += 1
					mainRoutine <- struct{}{}
				}

			}
		END:
			fmt.Printf("goroutine-num:%d end\n", idx)
		}(i)
	}

	wg.Add(1)

	go func() {
		wg.Done()

		for i := 0; i < CounterMaxValue; i++ {
			//<-mainRoutine
			sigChannel[i%GoRoutineNum] <- struct{}{}
			<-mainRoutine
		}

		////使最后一个任务处理协成，进入信号等待状态
		//<-mainRoutine

		for i := 0; i < GoRoutineNum; i++ {
			quitSigChannel[i] <- struct{}{}
		}
	}()

	//mainRoutine <- struct{}{}
	wg.Wait()
}

func main() {
	DoJob(10)
}