Golang 函數中 goroutine 的建立與管理

Go語言中，建立goroutine使用go關鍵字加函數呼叫。管理goroutine時，使用sync.WaitGroup進行同步；使用context套件可取消goroutine。實戰中可用於並行處理網路請求、圖片處理等任務。

Golang 函數中goroutine 的建立和管理

Goroutine（協程）是Go 語言中的輕量級並行執行單元，可以在單一線程中同時並發運行多個任務。

建立Goroutine

建立一個goroutine 非常簡單，可以使用go 關鍵字後跟一個函數呼叫即可：

func hello() {
    fmt.Println("Hello from goroutine")
}

func main() {
    go hello() // 创建一个执行 hello() 函数的 goroutine
}

Goroutine 管理

同步

在處理共享資源時，需要對goroutine 進行同步。使用sync.WaitGroup 可以等待一組goroutine 完成：

var wg sync.WaitGroup

func hello(name string) {
    wg.Add(1)
    defer wg.Done()

    fmt.Println("Hello", name)
}

func main() {
    wg.Add(3)
    go hello("John")
    go hello("Mary")
    go hello("Bob")
    wg.Wait() // 等待所有 goroutine 完成
}

取消

可以使用context 套件取消goroutine：

import (
    "context"
    "fmt"
    "time"
)

func heavyComputation(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            fmt.Println("Computation cancelled")
            return
        default:
            // 执行计算
        }
    }
}

func main() {
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel()

    go heavyComputation(ctx)
    time.Sleep(10 * time.Second) // 10 秒后取消计算
}

實戰案例

並行處理網路請求：

func main() {
    urls := []string{"https://example.com", "https://example.net", "https://example.org"}

    var wg sync.WaitGroup

    for _, url := range urls {
        wg.Add(1)
        go func(url string) {
            resp, err := http.Get(url)
            if err != nil {
                log.Fatal(err)
            }

            resp.Body.Close()
            wg.Done()
        }(url)
    }

    wg.Wait()
}

#並行處理圖片處理：

func main() {
    images := []string{"image1.jpg", "image2.jpg", "image3.jpg"}

    var wg sync.WaitGroup

    for _, image := range images {
        wg.Add(1)
        go func(image string) {
            img, err := image.Decode(image)
            if err != nil {
                log.Fatal(err)
            }

            // 处理图像

            wg.Done()
        }(image)
    }

    wg.Wait()
}

以上是Golang 函數中 goroutine 的建立與管理的詳細內容。更多資訊請關注PHP中文網其他相關文章！