unistack-org/micro
4
0
Fork 3
Code Issues 17 Pull Requests 1 Actions Packages Projects Releases 258 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

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

@ -99,6 +99,7 @@ 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++ {
err := func() error {
// subscribe to work // subscribe to work
subWork, err := t.Broker.Subscribe(topic, workFn, broker.Queue(fmt.Sprintf("work.%d", i))) subWork, err := t.Broker.Subscribe(topic, workFn, broker.Queue(fmt.Sprintf("work.%d", i)))
if err != nil { if err != nil {
@ -107,6 +108,13 @@ func (t *Task) Run(c task.Command) error {
// unsubscribe on completion // unsubscribe on completion
defer subWork.Unsubscribe() defer subWork.Unsubscribe()
return nil
}()
if err != nil {
return err
}
} }
// subscribe to all status messages // subscribe to all status messages