如何在 Go 的 select 语句中优先考虑上下文取消？

Go 的 Select 语句中的优先级控制

在 Go 中，select 语句允许 goroutine 之间进行异步通信。然而，case 块的评估顺序有时会导致意外的结果。当尝试优先处理特定情况时，例如在取消上下文时立即退出例程，就会出现此问题。

考虑以下代码：

func sendRegularHeartbeats(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            return
        case <-time.After(1 * time.Second):
            sendHeartbeat()
        }
    }
}

问题：
在此代码中，目的是优先处理 ctx.Done() 情况而不是 time.After() 情况。然而，由于 Go 的不确定性评估顺序，有时可能会首先评估 time.After() 情况，从而导致即使上下文被取消也会传输心跳。

解决方案：
接受的答案中提出的解决方案建议添加一个嵌套的 select 语句，并对 ctx.Done() 进行非阻塞检查。然而，这并不能完全解决问题，因为发送心跳的概率仍然很高。

更有效的方法是通过添加默认情况来优先处理 ctx.Done() 情况外部 select 语句：

func sendRegularHeartbeats(ctx context.Context) {
    ticker := time.NewTicker(time.Second)
    defer ticker.Stop()

    for {
        select {
        case <-ctx.Done():
            return
        default:
        }

        select {
        case <-ctx.Done():
            return
        case <-ticker.C:
            sendHeartbeat()
        }
    }
}

这确保了 ctx.Done() 情况在外部 select 语句中首先被评估，从而赋予它更高的优先级。如果上下文被取消，例程将立即返回并且不会发送任何心跳。如果 time.After() 的情况先到达，由于外部 select 语句中的默认情况，它将被忽略。

以上是如何在 Go 的 select 语句中优先考虑上下文取消？的详细内容。更多信息请关注PHP中文网其他相关文章！