[#930] policer: Release task pool when context cancelled

Signed-off-by: Anton Nikiforov <an.nikiforov@yadro.com>
pull/944/head
Anton Nikiforov 2024-01-24 14:48:47 +03:00 committed by Evgenii Stratonikov
parent 675eec91f3
commit c441296592
1 changed files with 2 additions and 0 deletions

View File

@ -19,6 +19,7 @@ func (p *Policer) shardPolicyWorker(ctx context.Context) {
for {
select {
case <-ctx.Done():
p.taskPool.Release()
return
default:
}
@ -36,6 +37,7 @@ func (p *Policer) shardPolicyWorker(ctx context.Context) {
for i := range addrs {
select {
case <-ctx.Done():
p.taskPool.Release()
return
default:
addr := addrs[i]