Asynchronously close brokers during a RefreshBrokers (#2693)

Signed-off-by: Baptiste Massemin <baptiste.massemin@shopify.com>
IBM · Oct 30, 2023 · 82f0e48 · 82f0e48
1 parent ee1944b
commit 82f0e48
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/client.go b/client.go
@@ -519,16 +519,16 @@ func (client *client) RefreshBrokers(addrs []string) error {
 	defer client.lock.Unlock()
 
 	for _, broker := range client.brokers {
-		_ = broker.Close()
-		delete(client.brokers, broker.ID())
+		safeAsyncClose(broker)
 	}
+	client.brokers = make(map[int32]*Broker)
 
 	for _, broker := range client.seedBrokers {
-		_ = broker.Close()
+		safeAsyncClose(broker)
 	}
 
 	for _, broker := range client.deadSeeds {
-		_ = broker.Close()
+		safeAsyncClose(broker)
 	}
 
 	client.seedBrokers = nil