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