Go 的 Select 语句中实现优先级
Go 的 select 语句使 goroutine 能够在多个通道上复用,处理来自最先准备好的通道的数据。但是,它本身并不支持优先通道。本文探讨了一种常见的编程场景,其中人们希望在 select 语句中将某些通道优先于其他通道。
在提供的示例中,goroutine 将数据发送到两个通道:out 和 exit。虽然两个通道最初都是空的,但目的是优先考虑,以便在处理退出之前处理所有值。正如问题中所见,使用标准 select 语句并不能实现所需的行为。
为了解决这个问题,Go 提供了一个优雅的解决方案。不应依赖 select 的内置优先级,应优先考虑的通道(在本例中为 exit)应该只能由发送者 Goroutine 访问。当发送者 goroutine 准备终止时,它会关闭通道。接收器 Goroutine 使用基于范围的循环来迭代通道(对于 x := range vals),将继续处理数据,直到通道关闭。
实现此解决方案会产生以下代码:
package main import ( "fmt" "math/rand" "time" ) var ( produced = 0 processed = 0 ) func produceEndlessly(vals chan int, quit chan bool) { defer close(vals) for { select { case vals <- rand.Int(): produced++ case <-quit: return } } } func quitRandomly(quit chan bool) { d := time.Duration(rand.Int63n(5e9)) fmt.Println("SLEEP", d) time.Sleep(d) fmt.Println("SEND QUIT") quit <- true } func main() { vals, quit := make(chan int, 10), make(chan bool) go produceEndlessly(vals, quit) go quitRandomly(quit) for x := range vals { fmt.Println(x) processed++ time.Sleep(time.Duration(rand.Int63n(5e8))) } fmt.Println("Produced:", produced) fmt.Println("Processed:", processed) }
在此更新的代码中,退出通道仅从发送者 goroutine 进行通信。这确保了在发送者准备好终止之前不会选择退出。因此,接收协程将在处理退出之前耗尽 vals 中的所有数据。
以上是如何在 Go 的 Select 语句中确定通道的优先级?的详细内容。更多信息请关注PHP中文网其他相关文章!