Как я могу спать с отзывчивой отменой контекста?

Question

В Go я хочу time.Sleep в течение некоторого времени (например, ожидание между попытками), но хочу быстро вернуться, если контекст отменяется (не только из крайнего срока, но и вручную).

Какой правильный или лучший способ сделать это?Спасибо!

Answer 1

Вы можете использовать select, чтобы получить это:

package main

import (
    "fmt"
    "time"
    "context"
)

func main() {
    fmt.Println("Hello, playground")
    ctx, cancel := context.WithCancel(context.Background())
    defer cancel()
    go func(){
        t := time.Now()
        select{
        case <-ctx.Done(): //context cancelled
        case <-time.After(2 * time.Second): //timeout
        }
        fmt.Printf("here after: %v\n", time.Since(t))
    }()

    cancel() //cancel manually, comment out to see timeout kick in
    time.Sleep(3 * time.Second)
    fmt.Println("done")

}

Вот ссылка Go-Playground

Answer 2

Мне удалось сделать нечто подобное, скомбинировав CancelContext с TimeoutContext ...

Вот пример кода:

cancelCtx, cancel := context.WithCancel(context.Background())
defer cancel()
// The program "sleeps" for 5 seconds.
timeoutCtx, _ := context.WithTimeout(cancelCtx, 5*time.Second)
select {
case <-timeoutCtx.Done():
    if cancelCtx.Err() != nil {
        log.Printf("Context cancelled")
    }
}

В в этом репо выможно найти полное использование вышеуказанного кода.Извините за короткий ответ, я еще не включил компьютер, и мне не так просто ответить с телефона ...