bugfix: RefreshBrokers leaks original seed brokers

IBM · Apr 1, 2022 · f41a8d4 · f41a8d4
1 parent 9473433
commit f41a8d4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/client.go b/client.go
@@ -459,6 +459,10 @@ func (client *client) RefreshBrokers(addrs []string) error {
 		delete(client.brokers, broker.ID())
 	}
 
+	for _, broker := range client.seedBrokers {
+		_ = broker.Close()
+	}
+
 	client.seedBrokers = nil
 	client.deadSeeds = nil
 

diff --git a/client_test.go b/client_test.go
@@ -528,11 +528,11 @@ func TestClientRefreshBrokers(t *testing.T) {
 	initialSeed.Returns(metadataResponse1)
 
 	c, err := NewClient([]string{initialSeed.Addr()}, NewTestConfig())
-	client := c.(*client)
-
 	if err != nil {
 		t.Fatal(err)
 	}
+	defer c.Close()
+	client := c.(*client)
 
 	if len(client.Brokers()) != 2 {
 		t.Error("Meta broker is not 2")