adapt to connect retry mechanism

kafbat · Jan 3, 2025 · 77b5a70 · 77b5a70
1 parent cb06d48
commit 77b5a70
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 11 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
@@ -277,6 +277,12 @@ public Mono<ResponseEntity<Void>> resumeConnectorWithHttpInfo(String connectorNa
     return withRetryOnConflictOrRebalance(super.resumeConnectorWithHttpInfo(connectorName));
   }
 
+  @Override
+  public Mono<ResponseEntity<Void>> resetConnectorOffsets(String connectorName)
+      throws WebClientResponseException {
+    return withRetryOnConflictOrRebalance(super.resetConnectorOffsets(connectorName));
+  }
+
   @Override
   public Mono<ResponseEntity<Connector>> setConnectorConfigWithHttpInfo(String connectorName,
                                                                         Map<String, Object> requestBody)

diff --git a/api/src/test/java/io/kafbat/ui/KafkaConnectServiceTests.java b/api/src/test/java/io/kafbat/ui/KafkaConnectServiceTests.java
@@ -59,20 +59,10 @@ public void setUp() {
                 "file", "/tmp/test",
                 "test.password", "test-credentials")))
         .exchange()
+        .expectStatus().isOk()
         .expectBody()
         .returnResult();
 
-    webTestClient.get()
-        .uri("/api/clusters/{clusterName}/connects/{connectName}/connectors/{connectorName}",
-            LOCAL, connectName, connectorName)
-        .exchange()
-        .expectStatus().isOk();
-
-    // Kafka Connect may return transient HTTP 500 errors during rebalances
-    if (creationResult.getStatus() != HttpStatus.OK) {
-      log.warn(
-          "Ignoring a transient error while setting up the tested connector, because it has been created anyway.");
-    }
   }
 
   @AfterEach