transport: pool, stop shard refill faster when context is done

Fixes #291
scylladb · Sep 23, 2022 · 76b095d · 76b095d
1 parent 31b0e68
commit 76b095d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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)