GolangNote

Golang笔记

Golang sync.WaitGroup 的 Wait 超时处理

Permalink

sync.WaitGroup 使用 Add(1)Done()Wait()组合来实现多协程等待,如果某一协程未能合理处理错误,导致无法退出,此时需要引入超时机制。下面是一种超时处理方法。

Go: WaitGroup 超时处理
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
package main

import (
    "fmt"
    "sync"
    "time"
)

func main() {
    wg := sync.WaitGroup{}
    wg.Add(1)

    done := make(chan struct{})

    go func() {
        wg.Wait()
        done <- struct{}{}
    }()

    go func() {
        time.Sleep(5 * time.Second)
        wg.Done()
    }()

    timeout := time.Duration(10) * time.Second
    fmt.Printf("Wait for waitgroup (up to %s)\n", timeout)

    select {
    case <-done:
        fmt.Printf("Wait group finished\n")
    case <-time.After(timeout):
        fmt.Printf("Timed out waiting for wait group\n")
    }

    fmt.Printf("Free at last\n")
}

未超时,输出:

plaintext: 未超时输出
1
2
3
Wait for waitgroup (up to 10s)
Wait group finished
Free at last

修改 Sleep 时间

Go: 延迟响应
1
2
3
4
go func() {
    time.Sleep(15 * time.Second)
    wg.Done()
}()

超时输出:

plaintext: 超时输出
1
2
3
Wait for waitgroup (up to 10s)
Timed out waiting for wait group
Free at last

本文网址: https://golangnote.com/topic/259.html 转摘请注明来源

Related articles

Golang quicktemplate 模版快速入门

Golang 有很多的模版引擎,自带的 `html/template` 也很好,大多数情况都能满足需求,只是有些逻辑、条件判断不好在模版里实现, `quicktemplate` 是个很好的选择。...

Golang Range 的性能提升Tip

Go 语言里使用 range 可以方便遍历数组(array)、切片(slice)、字典(map)和信道(chan)。这里主要关注他们的性能。...

golang snappy 的使用场合

google 自家的 snappy 压缩优点是非常高的速度和合理的压缩率。压缩率比 gzip 小,CPU 占用小。...

Write a Comment to "Golang sync.WaitGroup 的 Wait 超时处理"

Submit Comment Login
Based on Golang + fastHTTP + sdb | go1.22.3 Processed in 0ms