search

09.3 优雅地结束goroutines

本节内容将介绍如何使用go标准库中的sync包来解决上一节提到的Goroutine中的任务还未执行完成，main()函数就提前结束的问题。 本节的代码文件为syncGo.go,我们基于上一节的create.go来扩展syncGo.go。

syncGo.go的第一部分代码如下：

package main 

import ( 
    "flag" 
    "fmt" 
    "sync" 
)

如上所示，我们不再需要time包，我们将使用sync包中的功能来等待所有的Goroutine执行完成。

在第10章“并发 - 高级主题”中，我们将会学习两种方式来对Goroutine进行超时处理。 syncGo.go的第二部分代码如下：

func main() { 
    n := flag.Int("n", 20, "Number of goroutines") 
    flag.Parse() 
    count := *n 
    fmt.Printf("Going to create %d goroutines.\n", count) 

    var waitGroup sync.WaitGroup

在上面的代码中，我们定义了sync.WaitGroup类型的变量，查看sync包的源码我们可以发现，waitgroup.go文件位于sync目录中，sync.WaitGroup的定义只不过是一个包含三个字段的结构体：

type WaitGroup struct { 
        noCopy noCopy 
        state1 [12]byte 
        sema   uint32 
    }

syncGo.go的输出将显示有关sync.WaitGroup变量工作方式的更多信息。

syncGo.go的第三部分代码如下：

在这里，您可以使用for循环创建所需数量的Goroutine。（当然，也可以写多个顺序的Go语句。）

每次调用sync.Add()都会增加sync.WaitGroup变量中的计数器。需要注意的是，在go语句之前调用sync.Add(1)非常重要，以防止出现任何竞争条件。当每个Goroutine完成其工作时，将执行sync.Done()函数，以减少相同的计数器。

syncGo.go的最后一部分代码如下：

sync.Wait()调用将阻塞，直到sync.WaitGroup变量中的计数器为零，从而保证所有Goroutine能执行完成。

syncGo.go的输出如下：

syncGo.go的输出因执行情况而异。另外，当Goroutines的数量为30时，一些Goroutine可能会在第二个fmt.Printf（“％＃v n”，waitGroup）语句之前完成它们的工作。最后需要注意sync.WaitGroup中的state1字段是一个保存计数器的元素，该计数器根据sync.Add()和sync.Done()调用而增加和减少。

Previous09.2.2 创建多个GoroutineNext09.3.1 当Add()和Done()的数量不匹配时会发生什么？

Last updated