超過時間限制取消 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中文網其他相關文章！
