From 826766044162c7ad5d03e5c04a96f126f90fd19a Mon Sep 17 00:00:00 2001
From: Daniele Palaia <dpalaia@vmware.com>
Date: Thu, 31 Oct 2024 09:28:11 +0100
Subject: [PATCH] cleanup unused connections when load balancing mode is active

---
 src/consumer.rs | 2 ++
 src/producer.rs | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/consumer.rs b/src/consumer.rs
index 76a57376..23661cc8 100644
--- 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
index 538f9ca6..c781587c 100644
--- 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?;