From c441296592cdaca7152122a2ef1d8f85b02f241c Mon Sep 17 00:00:00 2001
From: Anton Nikiforov <an.nikiforov@yadro.com>
Date: Wed, 24 Jan 2024 14:48:47 +0300
Subject: [PATCH] [#930] policer: Release task pool when context cancelled

Signed-off-by: Anton Nikiforov <an.nikiforov@yadro.com>
---
 pkg/services/policer/process.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pkg/services/policer/process.go b/pkg/services/policer/process.go
index 873a1efb..c4c82020 100644
--- a/pkg/services/policer/process.go
+++ b/pkg/services/policer/process.go
@@ -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]