revert precheck error in send method

shotover · Sep 17, 2024 · 5fa3428 · 5fa3428
1 parent 47fe3ea
commit 5fa3428
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/shotover/src/connection.rs b/shotover/src/connection.rs
@@ -96,7 +96,7 @@ impl SinkConnection {
         self.error.clone().unwrap()
     }
 
-    pub fn get_error(&mut self) -> Option<ConnectionError> {
+    pub fn try_set_get_error(&mut self) -> Option<ConnectionError> {
         if self.error.is_none() {
             self.error = self.connection_closed_rx.try_recv().ok();
         }
@@ -110,9 +110,6 @@ impl SinkConnection {
 
         if let Some(error) = &self.error {
             Err(error.clone())
-        } else if let Ok(error) = self.connection_closed_rx.try_recv() {
-            self.error = Some(error.clone());
-            Err(error)
         } else {
             self.out_tx.send(messages).map_err(|_| self.set_get_error())
         }

diff --git a/shotover/src/transforms/kafka/sink_cluster/connections.rs b/shotover/src/transforms/kafka/sink_cluster/connections.rs
@@ -335,7 +335,7 @@ impl KafkaConnection {
 
     pub fn get_error(&mut self) -> Option<ConnectionError> {
         match self {
-            KafkaConnection::Regular(c) => c.get_error(),
+            KafkaConnection::Regular(c) => c.try_set_get_error(),
             KafkaConnection::ScramOverMtls(c) => c.get_error(),
         }
     }

diff --git a/shotover/src/transforms/kafka/sink_cluster/scram_over_mtls/connection.rs b/shotover/src/transforms/kafka/sink_cluster/scram_over_mtls/connection.rs
@@ -87,7 +87,7 @@ impl ScramOverMtlsConnection {
     }
 
     pub fn get_error(&mut self) -> Option<ConnectionError> {
-        self.connection.get_error()
+        self.connection.try_set_get_error()
     }
 
     pub fn pending_requests_count(&self) -> usize {