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

Fixes #291
scylladb · Sep 23, 2022 · cb19b3c · cb19b3c
1 parent 8c54361
commit cb19b3c
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)