在 Go 语言中,处理线程(称为"goroutine")之间的交互主要使用以下几种方法:
1. 使用 Channels
Channels 是 Go 中用于 goroutine 之间通信的主要机制。通过 channels,您可以安全地在不同的 goroutine 之间传递数据。
示例代码
go
package main
import (
"fmt"
"time"
)
func main() {
// 创建一个 channel
ch := make(chan string)
// 启动一个 goroutine
go func() {
time.Sleep(1 * time.Second) // 模拟一些工作
ch <- "Hello from goroutine!" // 发送数据到 channel
}()
// 主 goroutine 等待并接收数据
msg := <-ch
fmt.Println(msg)
}
2. 使用 WaitGroup
sync.WaitGroup
用于等待一组 goroutine 完成。它提供了一种简单的方式来管理并发任务的同步。
示例代码
go
package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
for i := 0; i < 3; i++ {
wg.Add(1) // 增加计数
go func(i int) {
defer wg.Done() // 减少计数
fmt.Printf("Goroutine %d\n", i)
}(i)
}
wg.Wait() // 等待所有 goroutine 完成
fmt.Println("All goroutines finished!")
}
3. 使用 Mutex
当多个 goroutine 需要访问共享数据时,可以使用 sync.Mutex
来确保互斥访问,防止数据竞争。
示例代码
go
package main
import (
"fmt"
"sync"
)
var (
counter int
mu sync.Mutex
)
func increment(wg *sync.WaitGroup) {
defer wg.Done()
mu.Lock() // 加锁
counter++ // 访问共享数据
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
语句可以在多个 channel 上等待操作。它允许您在多个 goroutine 通信时做出更复杂的决策。
示例代码
go
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)
}
}
总结
- Channels:用于 goroutine 之间的安全通信。
- WaitGroup:用于等待多个 goroutine 完成。
- Mutex:用于保护共享数据的访问,防止数据竞争。
- Select:用于处理多个 channel 的接收和发送。
通过以上方法,您可以有效地处理 goroutine 之间的交互和同步。