关于使用sync.WaitGroup进行同步的问题

hacker_cgh · · 1442 次点击 · · 开始浏览

这是一个创建于的文章，其中的信息可能已经有所发展或是发生改变。

代码如下

/*
	1.学习goroutin 间的chan数据传输(进程间通讯可使用分布式系统的方式解决，比如使用socket、http通讯协议)
	2.学习select使用
*/
package main

import (
	"lib/publib/github.com/wonderivan/logger"
	"runtime"
	"sync"
	"time"
)


func main() {
	logger.Warn("***************************************************************************************************")
	logger.Warn("1.学习goroutin 间的chan数据传输(进程间通讯可使用分布式系统的方式解决，比如使用socket、http通讯协议)")
	logger.Warn("2.学习select使用")
	logger.Warn("***************************************************************************************************")
	runtime.GOMAXPROCS(runtime.NumCPU())
	wg := sync.WaitGroup{}


	iChan := make(chan int, 1)
	sChan := make(chan rune, 1)

	go sendIchan(&wg,iChan)
	go sendSchan(&wg,sChan)
	go mainComm(&wg,iChan, sChan, 2) /*这里不能再用gorutin,必须要阻塞主函数，否则主routin先退出不会有任何输出*/
	time.Sleep(2*time.Second)
	wg.Wait()

}

func sendIchan(wg *sync.WaitGroup,iChan chan int) {
	wg.Add(1)
	for i := 0; i < 10; i++ {
		iChan <- i
		//time.Sleep(1*time.Second)
		logger.Debug("add %v to iChan",i)
	}
	return
}

func sendSchan(wg *sync.WaitGroup,sChan chan rune) {
	wg.Add(1)
	for c := 'A'; c <= 'Z'; c++ {
		sChan <- c
		//time.Sleep(1*time.Second)
		logger.Debug("add %c to sChan",c)
	}
	return
}

func mainComm(wg *sync.WaitGroup,iChan chan int, sChan chan rune, iTimeOut time.Duration) {
	for {
		select {
		case v, ok := <-iChan:
			if !ok {
				logger.Error("iChan Read Error")
				break
			}
			logger.Debug("read %v From iChan", v)
			wg.Done()
		case v, ok := <-sChan:
			if !ok {
				logger.Error("sChan Read Error")
				break
			}
			logger.Debug("read %c From sChan", v)
			wg.Done()
		case <-time.After(iTimeOut * time.Second):
			logger.Error("%ds 超时未收到数据", iTimeOut)
			break
		}
	}
}