Golang 函数如何高效处理并行任务

Go 函数中的高效并行任务处理：使用 go 关键字启动并发例程。使用 sync.WaitGroup 计数未完成例程的数量。例程完成时调用 wg.Done() 递减计数器。主程序使用 wg.Wait() 阻塞，直到所有例程完成。实战案例：并发发送 Web 请求并收集响应。

Go 函数中的高效并行任务处理

Go 语言提供了强大的并行处理能力，使开发人员能够轻松地创建并执行并发任务。本文将讨论如何使用 Go 函数高效地处理并行任务，并提供一个实战案例来演示它的用法。

Go 函数中的并发

Go 函数提供了两个关键字来支持并发执行：go 和 sync.WaitGroup。go 关键字用于启动并发的 Go 例程，而 sync.WaitGroup 用于等待所有例程完成。

等待组

sync.WaitGroup 是一个计数器，用于跟踪未完成例程的数量。当例程完成时，它会调用 wg.Done() 来递减计数器。主程序可以使用 wg.Wait() 方法来阻塞，直到所有例程完成。

实战案例：并发 Web 请求

考虑一个场景，我们需要并发发送多个 Web 请求并收集响应。以下是演示如何使用 Go 函数进行高效并发任务处理的代码：

package main

import (
    "fmt"
    "net/http"
    "sync"
)

func main() {
    // 要发送的 Web 请求 URL
    urls := []string{"https://example.com", "https://example2.com", "https://example3.com"}

    // 创建等待组
    var wg sync.WaitGroup

    for _, url := range urls {
        // 启动一个并发例程来发送 Web 请求
        wg.Add(1)
        go func(url string) {
            // 发送 GET 请求
            resp, err := http.Get(url)
            if err != nil {
                fmt.Printf("Error getting %s: %vn", url, err)
            } else {
                fmt.Printf("Status code for %s: %dn", url, resp.StatusCode)
            }

            // 例程完成，递减等待组计数
            wg.Done()
        }(url)
    }

    // 等待所有例程完成
    wg.Wait()
}

在上面的代码中：

• 我们使用 sync.WaitGroup 来跟踪未完成例程的数量。
• 对于每个 Web 请求 URL，我们启动一个并发的 Go 例程来发送请求。
• 例程使用 wg.Done() 来表示它已完成。
• 主程序使用 wg.Wait() 阻塞，直到所有例程完成。

通过使用这种方法，我们可以并发发送多个 Web 请求，并无需等待每个请求的响应即可继续执行主程序。