码迷,mamicode.com
首页 > 其他好文 > 详细

nsq里面WaitGroups两种实用的用法

时间:2015-03-10 15:26:09      阅读:172      评论:0      收藏:0      [点我收藏+]

标签:

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

 

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")
}

 

原文参考:

http://nsq.io/overview/internals.html

 

nsq里面WaitGroups两种实用的用法

标签:

原文地址:http://www.cnblogs.com/zhangqingping/p/4326019.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!