How Can I Gracefully Cancel Goroutines After a Specified Time Limit in Go?

Canceling Goroutines after a Time Limit

In load testing scenarios, controlling the execution duration of Goroutines is crucial. Here's an effective approach to achieve this.

Consider the following code snippet that manages HTTP requests in Goroutines:

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

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

    go func() {
        time.Sleep(cfg.Duration * time.Second)
        ch <p>However, Goroutines from httpPost() continue running after the specified cfg.Duration has elapsed.</p><p>To address this issue, you can leverage Go's context package. By passing a context.Context object to your Goroutines, you can cancel those Goroutines when the specified timeout is reached.</p><p>Here's a revised version of your code using the context package:</p><pre class="brush:php;toolbar:false">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) <p>With this modification, when the specified cfg.Duration expires, the ctx.Done() channel is closed, signaling the cancellation of the httpPost() Goroutines, which will then return.</p>

The above is the detailed content of How Can I Gracefully Cancel Goroutines After a Specified Time Limit in Go?. For more information, please follow other related articles on the PHP Chinese website!