如何在 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中文網其他相關文章！