nsq里面WaitGroups两种实用的用法

转载于 2015-03-10 14:54:00 发布 · 98 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/zhangqingping/p/4326019.html

本文介绍如何使用Go语言中的WaitGroup实现多个goroutine的同步执行及完成等待，并通过通道来通知goroutine进行安全退出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

看过好几遍了，觉得挺实用的，记录备忘一下。

1.开启很多个 goroutine 之后，等待执行完毕

type WaitGroupWrapper struct {
    sync.WaitGroup
}

func (w *WaitGroupWrapper) Wrap(cb func()) {
    w.Add(1)
    go func() {
        cb()
        w.Done()
    }()
}

// can be used as follows:
wg := WaitGroupWrapper{}
wg.Wrap(func() { n.idPump() })
...
wg.Wait()

2.程序要关闭了，通知各个goroutine安全退出

func work() {
    exitChan := make(chan int)
    go task1(exitChan)
    go task2(exitChan)
    time.Sleep(5 * time.Second)
    close(exitChan)
}
func task1(exitChan chan int) {
    <-exitChan
    log.Printf("task1 exiting")
}

func task2(exitChan chan int) {
    <-exitChan
    log.Printf("task2 exiting")
}