首頁  >  文章  >  後端開發  >  Golang 函數中 goroutine 的建立與管理

Golang 函數中 goroutine 的建立與管理

王林
王林原創
2024-06-05 19:06:011061瀏覽

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

Golang 函数中 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中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn