rh等待所有的goroutine退出的两种方式

jdevsean · · 1888 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

大家好,这是来到Go社区的第一篇文章,一个初学Gopher,请多请假。

大体来说,有两种方式:

1.使用channel来传递信号

2.使用waitGroup

 

方式一,如图golang社区的一句名言,不要通过共享内存来通信,而应该通过通信来共享内存

例子如下:

func run(done chan int ) {

for {

select {

case <-done :

fmt.Println("exiting...")

done <- -1

break

 

default:

fmt.Println("nothing...")

}

}

}

 

func main() {

ch := make(chan int)

 

go run(ch)

 

fmt.Println("wait")

time.Sleep(time.Second * 2)

 

ch <- 1

fmt.Println(<-ch)

 

fmt.Println("main exited")

 

}

 

代码分析:在main函数中,创建了一个channel,并且开了一个goroutine,goroutine的逻辑是:无限循环,当从channel接受到一个值时,就打印离开的描述并且往channel发送一个数字-1表示可以结束了,否则一直等待。最后在main函数里打印接收到的值并且结束整个函数。

 

方式二,使用waitGroup退出goroutine,waitGroup有三个方法,Add(),Done(),Wait(),

在Golang源码中,Done实际上调用的是Add(-1)的方法,即在waitgroup中减去一个,

 

在使用waitGroup的时候,需要注意的是go文档的一句话:The main goroutine calls Add to set the number of goroutines to wait for. Then each of the goroutines runs and calls Done when finished.

即在运行main函数的goroutine里运行Add()函数,在其他的goroutine里面运行Done()函数。

下面可以看一个在慕课网Go教程的一段代码(该教程需要付费购买):

type worker struct {

in chan int

done func()

}

 

func doWorker(i int, w worker) {

for n := range w.in {

fmt.Printf("Worker %d received %c\n", i, n)

w.done()

}

}

 

func createWorker(i int, wg *sync.WaitGroup) worker {

w := worker{

in: make(chan int),

done: func() {

wg.Done()

},

}

go doWorker(i, w)

return w

}

 

func chanDemo() {

var wg sync.WaitGroup

 

var workers [10]worker

 

for i := 0; i < 10; i++ {

workers[i] = createWorker(i, &wg)

}

 

wg.Add(20)

 

for i, worker := range workers {

worker.in <- 'a' + i

fmt.Printf("print:%c \n",'a'+i )

}

 

for i, worker := range workers {

worker.in <- 'A' + i

fmt.Printf("print:%c\n",'A'+i )

}

 

wg.Wait()

}

 

func main() {

chanDemo()

}

在main的主函数调用了waitGroup的Add的方法,告诉程序有20goroutine需要等待,然后调用wait执行阻塞,直到所有的WaitGroup数量变成0,然后在每一次goroutine操作结束后调用Done方法,来减少需要等待的goroutine的数量。

参考:https://www.cnblogs.com/cobbliu/p/4461866.html

ps:如果可以,可以加一下微信18899731553,大家一起交流。


有疑问加站长微信联系(非本文作者)

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

1888 次点击  
加入收藏 微博
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传