美文网首页Go语言Golang
等待所有的goroutine退出的两种方式

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

作者: 睡着别叫醒我 | 来源:发表于2018-03-08 20:12 被阅读201次

虽然在前面的文章中有略微记录过一点这部分的内容,但还是单独列出来讲一下吧。

让goroutine退出的方式,有以下两种:

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中减去一个,

Golang中waitGroup中Done方法的源码

在使用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

相关文章

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

    虽然在前面的文章中有略微记录过一点这部分的内容,但还是单独列出来讲一下吧。 让goroutine退出的方式,有以下...

  • go 等待所有 goroutine 执行结束的方法

    通常我们在 main 中需要等待所有的 goroutine 都执行完毕时再退出一般有两种方法方法一大家都能想到用 ...

  • Go 优雅的退出程序

    前言:在实际开发中,有时候我们需要等待某个goroutine执行完毕或者几个goroutine执行完毕才退出主程序...

  • goroutine Goexit()

    退出当前goroutine

  • Go的进程阻塞

    在Go语言中,如果使用goroutine,经常需要阻塞主进程来等待goroutine的结束,我们有以下几种方式来实...

  • runtimefor goroutine

    runtime包中有几个处理goroutine的函数。 Goexit 退出当前执行的goroutine,可是def...

  • Go context

    控制并发有两种经典的方式:WaitGroup和Context WaitGroup:控制多个Goroutine同时完...

  • 第06天(并发编程)_01

    01_创建goroutine.go 02_主goroutine先退出.go 03_主协程先退出导致子协程没有来得及...

  • 并发

    控制并发有两种经典方式, WaitGroup:控制多个goroutine同时, var wg Sync.waitG...

  • go waitgroup

    waitgroup等待一组grroutine完成 主goroutine调用Add来设置需要等待的goroutine...

网友评论

本文标题:等待所有的goroutine退出的两种方式

本文链接:https://www.haomeiwen.com/subject/itxufftx.html