unistack-org/micro
4
0
Fork 3
Code Issues 19 Pull Requests 2 Actions 2 Packages Projects Releases 262 Wiki Activity

prevent resource leak (#1080)

Browse Source
This commit is contained in:
Shu xian 2020-01-03 21:31:47 +08:00 committed by Asim Aslam
parent 4e2339749c
commit 31362bc331
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
sync/task/broker/broker.go
View File

@ -99,14 +99,22 @@ func (t *Task) Run(c task.Command) error {
// subscribe for the pool size // subscribe for the pool size
for i := 0; i < t.Options.Pool; i++ { for i := 0; i < t.Options.Pool; i++ {
// subscribe to work err := func() error {
subWork, err := t.Broker.Subscribe(topic, workFn, broker.Queue(fmt.Sprintf("work.%d", i))) // subscribe to work
subWork, err := t.Broker.Subscribe(topic, workFn, broker.Queue(fmt.Sprintf("work.%d", i)))
if err != nil {
return err
}
// unsubscribe on completion
defer subWork.Unsubscribe()
return nil
}()
if err != nil { if err != nil {
return err return err
} }
// unsubscribe on completion
defer subWork.Unsubscribe()
} }
// subscribe to all status messages // subscribe to all status messages