Go中如何在指定時間後優雅地取消goroutine？

超過時間限制取消 Goroutine

在負載測試場景中，控制 Goroutines 的執行時長至關重要。這是實現此目的的有效方法。

考慮以下在 Goroutines 中管理 HTTP 請求的程式碼片段：

func attack(cfg AttackConfig) {
    // some code ...

    var ar attackResponse
    ch := make(chan uint8, 8)

    go func() {
        time.Sleep(cfg.Duration * time.Second)
        ch <- CALL_TIME_RAN_OUT
    }()

    for {
        if atomic.LoadInt32(&currConnections) < atomic.LoadInt32(&maxConnections) - 1 {
            go httpPost(cfg, &ar, ch)
        }

        switch <-ch {
        // some other cases ...
        case CALL_TIME_RAN_OUT:
            fmt.Printf("%d seconds have elapsed. Shutting down!", cfg.Duration)
            return
        }
    }
}

但是，來自 httpPost() 的 Goroutines 在指定的 cfg.Duration 後繼續運行已經過去了。

要解決這個問題，您可以利用 Go 的 context 套件。透過將 context.Context 物件傳遞給您的 Goroutines，您可以在達到指定逾時時取消這些 Goroutines。

這是使用context 套件的程式碼的修訂版本：

import (
    "context"
    "fmt"
    "golang.org/x/net/context"
    "time"
)

func attack(cfg AttackConfig) {
    // some code ...

    var ar attackResponse

    // Define a timeout context
    ctx, cancel := context.WithTimeout(context.Background(), cfg.Duration*time.Second)
    defer cancel()

    go func() {
        time.Sleep(cfg.Duration * time.Second)
        cancel()
    }()

    for {
        if atomic.LoadInt32(&currConnections) < atomic.LoadInt32(&maxConnections) - 1 {
            go httpPost(ctx, cfg, &ar)
        }

        select {
        // some other cases ...
        case <-ctx.Done():
            fmt.Printf("%d seconds have elapsed. Shutting down!", cfg.Duration)
            return
        }
    }
}

func httpPost(ctx context.Context, cfg AttackConfig, a *attackResponse) {
    // some code here to create HTTP client ...

    for {
        // some code to make HTTP call ...

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

透過此修改，當指定的cfg.Duration 到期時，ctx.Done() 通道將關閉，表示取消httpPost() Goroutine，這將然後返回。

以上是Go中如何在指定時間後優雅地取消goroutine？的詳細內容。更多資訊請關注PHP中文網其他相關文章！