golang 获取协程执行结果

使用Go语言开发的最大特点之一就是轻量级的协程（Goroutine）。协程是一种轻量级的线程，能够在不同的函数之间进行切换执行，并且能够并发处理任务。但是在实际应用中，我们有时候需要等待协程执行完毕并获取其执行结果。本文将介绍在Go语言中如何获取协程的执行结果。 ## 使用channel获取协程执行结果 在Go语言中，可以通过channel来实现协程之间的通信。当协程执行完毕后，可以通过向channel发送数据来传递执行结果。调用channel的接收操作会阻塞等待，直到有数据被发送到channel中。下面是一个示例代码： ```go package main import ( "fmt" ) func calculateSum(a, b int, result chan int) { sum := a + b result <- sum } func main() { result := make(chan int) go calculateSum(1, 2, result) fmt.Println("Waiting for goroutine to finish...") sum := <-result fmt.Println("Sum:", sum) } ``` 在上面的代码中，我们定义了一个`calculateSum`函数，该函数接受两个整数参数，并将计算结果发送到`result`通道中。在`main`函数中，我们创建了一个名为`result`的通道，并通过`go`关键字启动了一个协程。然后，我们通过`sum := <-result`语句从通道中接收计算结果，并打印出来。 运行上面的代码，输出结果为： ``` Waiting for goroutine to finish... Sum: 3 ``` 从输出结果可以看出，主协程会等待子协程执行完毕，并通过通道获取结果。这种方式可以确保我们在需要时可以取到子协程的执行结果。 ## 使用sync包实现等待协程执行完毕 除了使用channel外，Go语言还提供了sync包中的`WaitGroup`类型来实现等待协程执行完毕。`WaitGroup`类型提供了三个方法：`Add`、`Done`和`Wait`，可以用于协程之间的同步。下面是一个示例代码： ```go package main import ( "fmt" "sync" ) func calculateSum(a, b int, wg *sync.WaitGroup) int { defer wg.Done() sum := a + b return sum } func main() { var wg sync.WaitGroup wg.Add(1) go func() { defer wg.Done() fmt.Println("Calculating sum...") sum := calculateSum(1, 2, &wg) fmt.Println("Sum:", sum) }() wg.Wait() } ``` 在上面的代码中，我们首先创建了一个`WaitGroup`对象`wg`，然后通过调用`wg.Add(1)`方法增加等待的协程数量。在协程内部，我们使用`defer wg.Done()`语句在协程执行完毕时调用`Done`方法，表示协程已经执行完毕。最后，我们调用`wg.Wait()`方法等待所有的协程执行完毕。 运行上面的代码，输出结果为： ``` Calculating sum... Sum: 3 ``` 从输出结果中可以看出，主协程会等待所有的子协程执行完毕，并在所有协程执行完毕后继续执行。 ## 使用context包来控制协程执行 除了使用channel和sync包外，Go语言还提供了context包来控制协程的执行。通过context包，我们可以在需要时取消协程的执行，或者设置超时。下面是一个示例代码： ```go package main import ( "context" "fmt" "time" ) func calculateSum(ctx context.Context, a, b int) int { select { case <-ctx.Done(): return 0 default: sum := a + b return sum } } func main() { ctx, cancel := context.WithTimeout(context.Background(), time.Second*3) defer cancel() go func() { sum := calculateSum(ctx, 1, 2) fmt.Println("Sum:", sum) }() select { case <-ctx.Done(): fmt.Println("Context canceled") } } ``` 在上面的代码中，我们首先通过`context.WithTimeout`函数创建了一个具有3秒超时时间的上下文`ctx`，并通过`defer`语句调用`cancel`方法来取消上下文。然后，我们启动了一个协程，在该协程中调用`calculateSum`函数计算两个整数的和。在`main`函数中，我们通过`select`语句等待上下文的终止，当上下文超时或者被取消时，`ctx.Done()`通道会被关闭，`select`语句就会执行相应的分支。 运行上面的代码，输出结果为： ``` Sum: 3 Context canceled ``` 从输出结果可以看出，子协程在主协程的上下文被取消后立即退出。 总结： 使用channel、sync和context包是Go语言中获取协程执行结果的常用方式。通过这些方式，我们可以在需要时等待协程执行完毕，并获取其执行结果。这些方法能够很好地提高程序的并发性能，同时也保证了协程之间的同步。因此，在开发过程中，根据不同的需求选择合适的方式来获取协程的执行结果是非常重要的。