golang 等待线程结束

Golang是一种快速、高效和易于使用的编程语言，被广泛用于构建并发和高性能的应用程序。在开发过程中，经常会遇到需要等待一个或多个线程结束的情况。本文将介绍在Golang中如何等待线程结束，并提供了几种不同的方法。

方法一：使用wait group

Golang中的sync包提供了一个WaitGroup类型，可以很方便地等待一组线程执行完毕。WaitGroup内部有一个计数器，用来记录还有多少个线程没有结束。当每个线程执行完毕时，需要调用WaitGroup的Done()方法，计数器会减1。主线程可以通过调用WaitGroup的Wait()方法来等待所有的线程执行完毕。

下面是一个使用WaitGroup等待线程结束的示例代码：

```go package main import ( "fmt" "sync" "time" ) func worker(id int, wg *sync.WaitGroup) { defer wg.Done() fmt.Printf("Worker %d starting\n", id) time.Sleep(time.Second) fmt.Printf("Worker %d done\n", id) } func main() { var wg sync.WaitGroup for i := 1; i <= 5; i++ { wg.Add(1) go worker(i, &wg) } wg.Wait() fmt.Println("All workers done") } ```

在上面的代码中，我们定义了一个worker函数，其中参数wg是一个指向WaitGroup的指针。在worker函数中，我们使用defer关键字来确保在函数结束时调用wg.Done()方法，减少计数器的值。

在主函数中，我们使用for循环启动了5个worker线程，并将每个线程添加到WaitGroup中。最后调用wg.Wait()方法来等待所有的线程执行完毕，输出"All workers done"表示所有线程都已经结束。

方法二：使用通道

Golang中的通道（Channel）提供了一种同步线程的方式，可以通过通道等待线程结束。在每个线程执行完毕时，可以向通道发送一个信号，主线程通过接收这些信号来判断所有线程是否都已经执行完毕。

下面是一个使用通道等待线程结束的示例代码：

```go package main import ( "fmt" "time" ) func worker(id int, done chan bool) { fmt.Printf("Worker %d starting\n", id) time.Sleep(time.Second) fmt.Printf("Worker %d done\n", id) done <- true } func main() { workers := 5 done := make(chan bool) for i := 1; i <= workers; i++ { go worker(i, done) } for i := 1; i <= workers; i++ { <-done } fmt.Println("All workers done") } ```

在上面的代码中，我们定义了一个worker函数，其中参数done是一个布尔类型的通道。在worker函数中，我们首先输出线程的开始和结束信息，然后向通道发送一个true值。

在主函数中，我们定义了workers和done两个变量，分别表示线程的数量和用于接收线程完成信号的通道。在启动每个线程时，我们将通道done作为参数传递给worker函数。然后使用for循环从通道done中接收信号，表示有一个线程执行完毕。当所有线程都执行完毕时，输出"All workers done"。

方法三：使用context

Golang中的context包提供了一种优雅的方式来取消多个并发操作。我们可以使用context来等待线程结束。在每个线程执行完毕时，可以调用context的Done()方法来判断是否需要继续执行。

下面是一个使用context等待线程结束的示例代码：

```go package main import ( "context" "fmt" "time" ) func worker(ctx context.Context, id int) { for { select { case <-ctx.Done(): fmt.Printf("Worker %d done\n", id) return default: fmt.Printf("Worker %d working\n", id) time.Sleep(time.Second) } } } func main() { ctx, cancel := context.WithCancel(context.Background()) defer cancel() workers := 5 for i := 1; i <= workers; i++ { go worker(ctx, i) } time.Sleep(time.Second * 3) cancel() fmt.Println("All workers done") } ```

在上面的代码中，我们定义了一个worker函数，其中参数ctx是一个上下文对象，id表示线程的编号。在worker函数中，我们使用select语句来监听context的Done()方法，当调用cancel函数时，表示需要终止当前线程。

在主函数中，我们首先使用context的WithCancel方法创建一个上下文对象，并通过调用cancel函数来释放资源。然后启动了5个worker线程，并在3秒后调用cancel函数来终止所有线程的执行。

总结

Golang提供了多种等待线程结束的方法，可以根据具体的需求选择合适的方式。使用WaitGroup可以很方便地等待一组线程执行完毕，通道和context则提供了更加灵活的方式来同步线程的执行。

无论使用哪种方式，合理地等待线程结束可以确保程序的正确性和稳定性。通过掌握这些方法，可以更好地利用Golang的并发优势，开发高效、稳定和可靠的应用程序。