如何处理 golang 中的 goroutines 并获得响应

在Go语言中，goroutines 是一种非常轻量级的线程，用于并发执行任务。处理 goroutines 并获取其响应可以通过多种方式实现，最常见的方法是使用通道（channels）和 sync 包中的工具，如 WaitGroup。下面我将详细介绍这两种方法，包括具体的例子。

1. 使用通道（Channels）

通道是用来在不同的 goroutines 之间安全地传递数据的。你可以使用通道来获取 goroutine 的执行结果。

例子：

假设我们需要计算多个数的平方，并获取结果。

go
package main

import (
	"fmt"
	"time"
)

func square(number int, ch chan int) {
	// 计算平方
	result := number * number
	// 模拟一些耗时
	time.Sleep(time.Second)
	// 将结果发送到通道中
	ch <- result
}

func main() {
	ch := make(chan int)
	numbers := []int{2, 4, 6, 8}

	for _, number := range numbers {
		go square(number, ch)
	}

	for range numbers {
		fmt.Println(<-ch)
	}
}

在这个例子中，我们创建了一个名为 square 的函数，它接受一个整数和一个通道，然后计算该整数的平方，并将结果发送到通道中。在 main 函数中，我们启动了多个 goroutines 来并行计算，并从通道中读取结果。

2. 使用 sync.WaitGroup

WaitGroup 是用来等待一组 goroutines 完成的。你可以在启动 goroutine 之前调用 Add 来设置计数器，然后在每个 goroutine 完成时调用 Done。

例子：

go
package main

import (
	"fmt"
	"sync"
	"time"
)

func square(number int, wg *sync.WaitGroup, results *[]int) {
	defer wg.Done()
	// 计算平方
	result := number * number
	// 模拟一些耗时
	time.Sleep(time.Second)
	// 将结果保存
	*results = append(*results, result)
}

func main() {
	var wg sync.WaitGroup
	var results []int

	numbers := []int{2, 4, 6, 8}
	for _, number := range numbers {
		wg.Add(1)
		go square(number, &wg, &results)
	}

	wg.Wait()
	fmt.Println(results)
}

在这个例子中，我们定义了一个 square 函数，它接受一个整数、一个指向 WaitGroup 的指针和一个指向结果数组的指针。每个 goroutine 完成时会调用 Done 方法。通过调用 Wait 方法，main 函数将等待所有 goroutine 完成后继续执行。

总结

使用通道和 WaitGroup 是处理 goroutines 并获取响应的两种常见方法。选择哪一种方法取决于具体的应用场景和个人偏好。通道非常适合于需要直接从 goroutines 传递数据的情况，而 WaitGroup 则适用于仅需等待一组操作完成的场景。