From c5e1edb0744d3aabcec3735566cd66953f8d9f60 Mon Sep 17 00:00:00 2001
From: Dugong <julien.nozais@gmail.com>
Date: Fri, 3 Jan 2025 11:50:46 +0000
Subject: [PATCH] fix stopConnector override with new retry mechanism

---
 .../kafbat/ui/client/RetryingKafkaConnectClient.java  | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/api/src/main/java/io/kafbat/ui/client/RetryingKafkaConnectClient.java b/api/src/main/java/io/kafbat/ui/client/RetryingKafkaConnectClient.java
index 35fb5d505..df2da3e55 100644
--- a/api/src/main/java/io/kafbat/ui/client/RetryingKafkaConnectClient.java
+++ b/api/src/main/java/io/kafbat/ui/client/RetryingKafkaConnectClient.java
@@ -233,14 +233,19 @@ public Mono<Void> pauseConnector(String connectorName) throws WebClientResponseE
     return withRetryOnConflictOrRebalance(super.pauseConnector(connectorName));
   }
 
+  @Override
+  public Mono<ResponseEntity<Void>> pauseConnectorWithHttpInfo(String connectorName) throws WebClientResponseException {
+    return withRetryOnConflictOrRebalance(super.pauseConnectorWithHttpInfo(connectorName));
+  }
+
   @Override
   public Mono<Void> stopConnector(String connectorName) throws WebClientResponseException {
-    return withRetryOnConflict(super.stopConnector(connectorName));
+    return withRetryOnConflictOrRebalance(super.stopConnector(connectorName));
   }
 
   @Override
-  public Mono<ResponseEntity<Void>> pauseConnectorWithHttpInfo(String connectorName) throws WebClientResponseException {
-    return withRetryOnConflictOrRebalance(super.pauseConnectorWithHttpInfo(connectorName));
+  public Mono<ResponseEntity<Void>> stopConnectorWithHttpInfo(String connectorName) throws WebClientResponseException {
+    return withRetryOnConflictOrRebalance(super.stopConnectorWithHttpInfo(connectorName));
   }
 
   @Override