php中文网

Golang 函数:如何并发地运行多个 goroutine?

php中文网

golang 中可使用 goroutine 实现并发,创建 goroutine 时使用 go 关键字,管理 goroutine 时使用 waitgroup,并行执行 goroutine 时使用 mutex。在实战案例中,可以通过并发获取网页内容来体验并发的优势,通过 goroutine 并行获取多个网页内容,并通过 waitgroup 等待所有 goroutine 完成,最后通过 channel 获取所有并行获取的数据。

Golang 函数:并发地运行多个 Goroutine

在 Golang 中,goroutine 是轻量级的协程,可以并发执行。这篇文章将介绍如何创建、管理、并行执行多个 goroutine。

创建 Goroutine

使用 go 关键字创建 Goroutine:

go func() {
  // Goroutine 体
}

管理 Goroutine

WaitGroup 用于等待一组 goroutine 完成执行:

立即学习“go语言免费学习笔记(深入)”;

import "sync"

var wg sync.WaitGroup

在 goroutine 中,调用 wg.Done() 来通知 WaitGroup 完成:

wg.Done()

在主函数中,调用 wg.Wait() 来等待所有 goroutine 完成:

wg.Wait()

并行执行 Goroutine

使用 sync.Mutex 来确保并行执行时数据的并发访问安全:

import "sync"

var mu sync.Mutex

在并行执行代码块时,使用 mu.Lock() 和 mu.Unlock() 来控制对共享数据的访问:

mu.Lock()
// 并行执行代码
mu.Unlock()

实战案例:并发获取网页内容

package main

import (
    "fmt"
    "io"
    "net/http"
    "time"
)

func fetchURL(url string, wg *sync.WaitGroup, results chan<- string) {
    resp, err := http.Get(url)
    if err != nil {
        fmt.Println("Error fetching URL:", err)
        return
    }
    defer resp.Body.Close()
    body, err := io.ReadAll(resp.Body)
    if err != nil {
        fmt.Println("Error reading body:", err)
        return
    }
    results <- string(body)
    wg.Done()
}

func main() {
    urls := []string{"https://example.com", "https://google.com", "https://amazon.com"}
    results := make(chan string)
    var wg sync.WaitGroup
    wg.Add(len(urls))
    for _, url := range urls {
        go fetchURL(url, &wg, results)
    }
    wg.Wait()
    close(results)

    // 获取所有并行获取的数据
    for result := range results {
        fmt.Println(result)
    }
}

以上就是Golang 函数:如何并发地运行多个 goroutine?的详细内容,更多请关注php中文网其它相关文章!