go 实现数字字母交替打印并优雅关闭channel

at 1年前  ca Golang  pv 360  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条评论