cleanup unused connections when load balancing mode is active (#239)

rabbitmq · Oct 31, 2024 · 4f78182 · 4f78182
1 parent ac1f614
commit 4f78182
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/consumer.rs b/src/consumer.rs
@@ -114,10 +114,12 @@ impl ConsumerBuilder {
                         let mapping = temp_client.connection_properties().await;
                         if let Some(advertised_host) = mapping.get("advertised_host") {
                             if *advertised_host == replica.host.clone() {
+                                client.close().await?;
                                 client = temp_client;
                                 break;
                             }
                         }
+                        temp_client.close().await?;
                     }
                 } else {
                     client.close().await?;

diff --git a/src/producer.rs b/src/producer.rs
@@ -145,10 +145,12 @@ impl<T> ProducerBuilder<T> {
                     let mapping = temp_client.connection_properties().await;
                     if let Some(advertised_host) = mapping.get("advertised_host") {
                         if *advertised_host == metadata.leader.host.clone() {
+                            client.close().await?;
                             client = temp_client;
                             break;
                         }
                     }
+                    temp_client.close().await?;
                 }
             } else {
                 client.close().await?;