go 实现数字字母交替打印并优雅关闭channel
at 2年前 ca Golang pv 623 by touch
实现一个交替打印字母和数字的简单程序,呈现的效果是:
a1b2c3d4e5f6g7h8i9j10k11l12m13n14o15p16q17r18s19t20u21v22w23x24y25z26
这个效果其实有很多种写法,现在选取一个简单的写法,通常的写法是让两个协程交替打印,用chan来阻塞通知,但是由于大部分的demo都没有考虑优雅退出的问题,这次的demo添加了一个stopCha来优雅退出并close通道
package main import ( "fmt" "sync" ) func main() { fmt.Println("开始打印") printLettersAndNumbAlternately() fmt.Println("打印结束") } func printLettersAndNumbAlternately() { // letterChan 字母打印通知->字母打印完成后就通知数字打印的协程,并阻塞自己 // numChan 数字打印通知->阻塞自己通知字母打印协程 // stopChan 结束通知器->字母打印完成后通知全局退出 letterChan, numChan, stopChan := make(chan struct{}), make(chan struct{}), make(chan struct{}, 1) wait := sync.WaitGroup{} wait.Add(3) go func() { defer wait.Done() defer close(numChan) num := 1 for { select { case <-letterChan: fmt.Print(num) select { case numChan <- struct{}{}: case <-stopChan: return } num++ default: } } }() go func() { defer wait.Done() defer close(letterChan) letters := get26Letter() point := 0 for { select { case <-numChan: fmt.Print(letters[point]) letterChan <- struct{}{} point++ if point == len(letters) { select { case stopChan <- struct{}{}: default: } return } default: } } }() // 主持人 go func() { defer wait.Done() // 先打印字母 numChan <- struct{}{} return }() wait.Wait() fmt.Println("\n所有协程结束") // 一切结束后再关闭 stop通知器 close(stopChan) } // 获取26个英文字母 func get26Letter() []string { var letters []string counter := 0 for i := 'a'; i <= 'z'; i++ { letters = append(letters, string(i)) counter++ } return letters }
版权声明
本文仅代表作者观点,不代表码农殇立场。
本文系作者授权码农殇发表,未经许可,不得转载。
已有0条评论