在 Go 中，我想要time.Sleep一段時(shí)間（例如在重試之間等待），但如果上下文被取消（不僅是從截止日期，而且是手動(dòng)），我想快速返回。這樣做的正確或最佳方法是什么？謝謝！
3 回答
翻閱古今
TA貢獻(xiàn)1780條經(jīng)驗(yàn) 獲得超5個(gè)贊
您可以使用select它來(lái)實(shí)現(xiàn)：
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")
}
茅侃侃
TA貢獻(xiàn)1842條經(jīng)驗(yàn) 獲得超22個(gè)贊
select您可以像其他人提到的那樣使用；但是，其他答案有一個(gè)錯(cuò)誤，因?yàn)閠imer.After()如果不清理就會(huì)泄漏內(nèi)存。
func SleepWithContext(ctx context.Context, d time.Duration) {
timer := time.NewTimer(d)
select {
case <-ctx.Done():
if !timer.Stop() {
<-timer.C
}
case <-timer.C:
}
}
拉丁的傳說(shuō)
TA貢獻(xiàn)1789條經(jīng)驗(yàn) 獲得超8個(gè)贊
這是一個(gè)sleepContext您可以用來(lái)代替的函數(shù)time.Sleep：
func sleepContext(ctx context.Context, delay time.Duration) {
select {
case <-ctx.Done():
case <-time.After(delay):
}
}
以及一些示例用法（Go Playground 上的完整可運(yùn)行代碼）：
func main() {
ctx := context.Background()
fmt.Println(time.Now())
sleepContext(ctx, 1*time.Second)
fmt.Println(time.Now())
ctxTimeout, cancel := context.WithTimeout(ctx, 500*time.Millisecond)
sleepContext(ctxTimeout, 1*time.Second)
cancel()
fmt.Println(time.Now())
}
- 3 回答
- 0 關(guān)注
- 229 瀏覽
添加回答
舉報(bào)
0/150
提交
取消