From 76b095d1eaa888797f597409902bd2e77c3d6408 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20Putra?= <pawel.putra@scylladb.com>
Date: Fri, 23 Sep 2022 14:56:44 +0200
Subject: [PATCH] transport: pool, stop shard refill faster when context is
 done

Fixes #291
---
 transport/pool.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/transport/pool.go b/transport/pool.go
index adadb984..39480551 100644
--- a/transport/pool.go
+++ b/transport/pool.go
@@ -233,7 +233,9 @@ func (r *PoolRefiller) fill(ctx context.Context) {
 		if r.pool.loadConn(i) != nil {
 			continue
 		}
-
+		if ctx.Err() != nil {
+			return
+		}
 		si.Shard = uint16(i)
 		span := startSpan()
 		conn, err := OpenShardConn(ctx, r.addr, si, r.cfg)