快捷導(dǎo)航

go處理線程之間的交互示例代碼

更新時(shí)間：2024年10月10日 11:01:25 作者：get200

Go語(yǔ)言以goroutine為核心實(shí)現(xiàn)并發(fā)編程,其中線程間交互主要通過(guò)Channels、WaitGroup、Mutex和Select實(shí)現(xiàn),Channels提供goroutine間的數(shù)據(jù)傳遞,本文給大家介紹go處理線程之間的交互示例代碼,感興趣的朋友一起看看吧

1. 使用 Channels

Channels 是 Go 中用于 goroutine 之間通信的主要機(jī)制。通過(guò) channels，您可以安全地在不同的 goroutine 之間傳遞數(shù)據(jù)。

示例代碼

package main
import (
    "fmt"
    "time"
)
func main() {
    // 創(chuàng)建一個(gè) channel
    ch := make(chan string)
    // 啟動(dòng)一個(gè) goroutine
    go func() {
        time.Sleep(1 * time.Second) // 模擬一些工作
        ch <- "Hello from goroutine!" // 發(fā)送數(shù)據(jù)到 channel
    }()
    // 主 goroutine 等待并接收數(shù)據(jù)
    msg := <-ch
    fmt.Println(msg)
}

2. 使用 WaitGroup

sync.WaitGroup 用于等待一組 goroutine 完成。它提供了一種簡(jiǎn)單的方式來(lái)管理并發(fā)任務(wù)的同步。

示例代碼

package main
import (
    "fmt"
    "sync"
)
func main() {
    var wg sync.WaitGroup
    for i := 0; i < 3; i++ {
        wg.Add(1) // 增加計(jì)數(shù)
        go func(i int) {
            defer wg.Done() // 減少計(jì)數(shù)
            fmt.Printf("Goroutine %d\n", i)
        }(i)
    }
    wg.Wait() // 等待所有 goroutine 完成
    fmt.Println("All goroutines finished!")
}

3. 使用 Mutex

當(dāng)多個(gè) goroutine 需要訪問(wèn)共享數(shù)據(jù)時(shí)，可以使用 sync.Mutex 來(lái)確?；コ庠L問(wèn)，防止數(shù)據(jù)競(jìng)爭(zhēng)。

示例代碼

package main
import (
    "fmt"
    "sync"
)
var (
    counter int
    mu      sync.Mutex
)
func increment(wg *sync.WaitGroup) {
    defer wg.Done()
    mu.Lock()         // 加鎖
    counter++         // 訪問(wèn)共享數(shù)據(jù)
    mu.Unlock()       // 解鎖
}
func main() {
    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go increment(&wg)
    }
    wg.Wait()
    fmt.Println("Final counter:", counter)
}

4. 使用 Select

select 語(yǔ)句可以在多個(gè) channel 上等待操作。它允許您在多個(gè) goroutine 通信時(shí)做出更復(fù)雜的決策。

示例代碼

package main
import (
    "fmt"
    "time"
)
func main() {
    ch1 := make(chan string)
    ch2 := make(chan string)
    go func() {
        time.Sleep(2 * time.Second)
        ch1 <- "Message from channel 1"
    }()
    go func() {
        time.Sleep(1 * time.Second)
        ch2 <- "Message from channel 2"
    }()
    select {
    case msg1 := <-ch1:
        fmt.Println(msg1)
    case msg2 := <-ch2:
        fmt.Println(msg2)
    }
}