Minor fix.

Signed-off-by: Yury-Fridlyand <[email protected]>
Bit-Quill · Dec 19, 2023 · e5a4e10 · e5a4e10
1 parent 647d802
commit e5a4e10
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/java/client/src/main/java/babushka/connectors/handlers/CallbackDispatcher.java b/java/client/src/main/java/babushka/connectors/handlers/CallbackDispatcher.java
@@ -40,7 +40,10 @@ public class CallbackDispatcher {
    *     response.
    */
   public Pair<Integer, CompletableFuture<Response>> registerRequest() {
-    int callbackId = requestId.getAndIncrement();
+    int callbackId = 0;
+    do {
+      callbackId = requestId.getAndIncrement();
+    } while (responses.containsKey(callbackId));
     var future = new CompletableFuture<Response>();
     responses.put(callbackId, future);
     return Pair.of(callbackId, future);