chore: generated code for commit 1ba6c86. [skip ci]

Co-authored-by: Pierre Millot <pierre.millot@algolia.com>
algolia · May 20, 2022 · 8c45208 · 8c45208
1 parent 1ba6c86
commit 8c45208
Showing 1 changed file with 53 additions and 0 deletions.
diff --git a/...iasearch-client-java-2/algoliasearch-core/src/main/java/com/algolia/api/SearchClient.java b/...iasearch-client-java-2/algoliasearch-core/src/main/java/com/algolia/api/SearchClient.java
@@ -17,6 +17,7 @@
 import java.util.Map;
 import java.util.Random;
 import java.util.concurrent.CompletableFuture;
+import java.util.function.IntUnaryOperator;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 import okhttp3.Call;
@@ -6119,4 +6120,56 @@ public CompletableFuture<UpdateApiKeyResponse> updateApiKeyAsync(
   ) throws AlgoliaRuntimeException {
     return this.updateApiKeyAsync(key, apiKey, null);
   }
+
+  public void waitForTask(
+    String indexName,
+    Long taskID,
+    RequestOptions requestOptions,
+    int maxTrial,
+    IntUnaryOperator timeout
+  ) {
+    TaskUtils.retryUntil(
+      () -> {
+        return this.getTaskAsync(indexName, taskID, requestOptions);
+      },
+      (GetTaskResponse task) -> {
+        return task.getStatus() == TaskStatus.PUBLISHED;
+      },
+      maxTrial,
+      timeout
+    );
+  }
+
+  public void waitForTask(
+    String indexName,
+    Long taskID,
+    RequestOptions requestOptions
+  ) {
+    this.waitForTask(
+        indexName,
+        taskID,
+        requestOptions,
+        TaskUtils.DEFAULT_MAX_TRIAL,
+        TaskUtils.DEFAULT_TIMEOUT
+      );
+  }
+
+  public void waitForTask(
+    String indexName,
+    Long taskID,
+    int maxTrial,
+    IntUnaryOperator timeout
+  ) {
+    this.waitForTask(indexName, taskID, null, maxTrial, timeout);
+  }
+
+  public void waitForTask(String indexName, Long taskID) {
+    this.waitForTask(
+        indexName,
+        taskID,
+        null,
+        TaskUtils.DEFAULT_MAX_TRIAL,
+        TaskUtils.DEFAULT_TIMEOUT
+      );
+  }
 }