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

解决 Go Select 语句中的优先级

在某些情况下，确定 select 语句中 case 块的执行优先级变得至关重要。具体来说，当使用表示取消事件的上下文时，必须立即处理 Done 信号以确保程序正确终止。

考虑以下代码：

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

此例程运行在一个单独的 goroutine 中并定期传输心跳。但是，当上下文被取消时，代码偶尔可能仍会在处理 Done 案例之前发送心跳。

select 语句的默认行为不保证 case 评估的顺序，因此有必要强制执行明确所需的优先级。一种不完美的方法是在执行心跳传输之前检查封闭上下文。

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

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

        // Second select
        select {
        case <-ctx.Done():
            return
        case <-ticker.C:
            // Check if context is done again in case a concurrent Done event arrived
            select {
            case <-ctx.Done():
            default:
            }
            sendHeartbeat()
        }
    }
}

虽然此方法优先考虑 Done 信号，但它引入了额外的竞争条件。如果 Done 事件和 Ticker 事件同时发生，则有可能在 Done 事件处理之前就发生心跳传输。

不幸的是，目前 Go 中还没有完美的解决方案。但是，所提供的解决方法通过引入嵌套 select 语句来对原始代码进行改进，以最大程度地减少出现此类竞争条件的可能性。

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