diff --git a/algoliasearch-common/src/main/java/com/algolia/search/objects/tasks/sync/Task.java b/algoliasearch-common/src/main/java/com/algolia/search/objects/tasks/sync/Task.java index ddd47a5e9..b2a10b3a9 100644 --- a/algoliasearch-common/src/main/java/com/algolia/search/objects/tasks/sync/Task.java +++ b/algoliasearch-common/src/main/java/com/algolia/search/objects/tasks/sync/Task.java @@ -2,8 +2,10 @@ import com.algolia.search.APIClient; import com.algolia.search.exceptions.AlgoliaException; +import com.algolia.search.objects.RequestOptions; import com.fasterxml.jackson.annotation.JsonIgnoreProperties; import com.fasterxml.jackson.annotation.JsonInclude; +import javax.annotation.Nonnull; @JsonInclude(JsonInclude.Include.NON_NULL) @JsonIgnoreProperties(ignoreUnknown = true) @@ -20,7 +22,7 @@ public Task setAPIClient(APIClient apiClient) { * @throws AlgoliaException */ public void waitForCompletion() throws AlgoliaException { - apiClient.waitTask(this, 100); + waitForCompletion(RequestOptions.empty); } /** @@ -30,7 +32,29 @@ public void waitForCompletion() throws AlgoliaException { * @throws AlgoliaException */ public void waitForCompletion(long timeToWait) throws AlgoliaException { - apiClient.waitTask(this, timeToWait); + waitForCompletion(timeToWait, RequestOptions.empty); + } + + /** + * Wait for the completion of this task + * + * @param requestOptions Options to pass to this request + * @throws AlgoliaException + */ + public void waitForCompletion(@Nonnull RequestOptions requestOptions) throws AlgoliaException { + waitForCompletion(100, requestOptions); + } + + /** + * Wait for the completion of this task + * + * @param timeToWait the time to wait in milliseconds + * @param requestOptions Options to pass to this request + * @throws AlgoliaException + */ + public void waitForCompletion(long timeToWait, @Nonnull RequestOptions requestOptions) + throws AlgoliaException { + apiClient.waitTask(this, timeToWait, requestOptions); } @Override