fix: java

Author: shortcuts

templates/java/api.mustache

@@ -46,6 +46,7 @@ import java.util.Base64;
 import javax.annotation.Nonnull;
 import javax.crypto.Mac;
 import javax.crypto.spec.SecretKeySpec;
+import com.algolia.model.ingestion.Event;
 import com.algolia.model.ingestion.PushTaskPayload;
 import com.algolia.model.ingestion.PushTaskRecords;
 import com.algolia.model.ingestion.WatchResponse;

templates/java/api_helpers.mustache

@@ -586,7 +586,9 @@ public CompletableFuture<List<SearchForFacetValuesResponse>> searchForFacetsAsyn
 }
 
 /**
- * Helper: Chunks the given `objects` list in subset of 1000 elements max in order to make it fit in `push` requests by leveraging the Transformation pipeline setup in the Push connector (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/).
+ * Helper: Chunks the given `objects` list in subset of 1000 elements max in order to make it fit
+ * in `push` requests by leveraging the Transformation pipeline setup in the Push connector
+ * (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/).
  *
  * @summary Helper: Chunks the given `objects` list in subset of 1000 elements max in order to
  *     make it fit in `batch` requests.
@@ -598,14 +600,16 @@ public CompletableFuture<List<SearchForFacetValuesResponse>> searchForFacetsAsyn
  *     reliable.
  * @param batchSize - The size of the chunk of `objects`. The number of `batch` calls will be
  *     equal to `length(objects) / batchSize`. Defaults to 1000.
- * @param referenceIndexName - This is required when targeting an index that does not have a push connector setup (e.g. a tmp index), but you wish to attach another index's transformation to it (e.g. the source index name).
+ * @param referenceIndexName - This is required when targeting an index that does not have a push
+ *     connector setup (e.g. a tmp index), but you wish to attach another index's transformation
+ *     to it (e.g. the source index name).
  * @param requestOptions - The requestOptions to send along with the query, they will be forwarded
  *     to the `getTask` method and merged with the transporter requestOptions.
  */
 public <T> List<WatchResponse> chunkedPush(
   String indexName,
   Iterable<T> objects,
-  Action action,
+  com.algolia.model.ingestion.Action action,
   boolean waitForTasks,
   int batchSize,
   String referenceIndexName,
@@ -616,17 +620,18 @@ public <T> List<WatchResponse> chunkedPush(
   }
 
   List<WatchResponse> responses = new ArrayList<>();
-  Iterable<T> records = new ArrayList<>();
+  List<T> records = new ArrayList<>();
 
   for (T item : objects) {
-    if (requests.size() == batchSize) {
-      WatchResponse watch = this.ingestionTransporter.push(
-        indexName,
-        new PushTaskPayload().setAction(action).setRecords(this.objectsToPushTaskRecords(records)),
-        waitForTasks,
-        null,
-        requestOptions
-      );
+    if (records.size() == batchSize) {
+      WatchResponse watch =
+        this.ingestionTransporter.push(
+            indexName,
+            new PushTaskPayload().setAction(action).setRecords(this.objectsToPushTaskRecords(records)),
+            waitForTasks,
+            referenceIndexName,
+            requestOptions
+          );
       responses.add(watch);
       records.clear();
     }
@@ -635,35 +640,38 @@ public <T> List<WatchResponse> chunkedPush(
   }
 
   if (records.size() > 0) {
-      WatchResponse watch = this.ingestionTransporter.push(
-        indexName,
-        new PushTaskPayload().setAction(action).setRecords(this.objectsToPushTaskRecords(records)),
-        waitForTasks,
-        null,
-        requestOptions
-      );
-      responses.add(watch);
+    WatchResponse watch =
+      this.ingestionTransporter.push(
+          indexName,
+          new PushTaskPayload().setAction(action).setRecords(this.objectsToPushTaskRecords(records)),
+          waitForTasks,
+          referenceIndexName,
+          requestOptions
+        );
+    responses.add(watch);
   }
 
   if (waitForTasks) {
-    responses.forEach(response -> TaskUtils.retryUntil(
-        () -> {
-          try {
-            return this.ingestionTransporter.getEvent(response.runID, response.eventID);
-          } catch (AlgoliaApiException e) {
-            if (e.getStatusCode() == 404) {
-              return null;
+    responses.forEach(response -> {
+        TaskUtils.retryUntil(
+          () -> {
+            try {
+              return this.ingestionTransporter.getEvent(response.getRunID(), response.getEventID());
+            } catch (AlgoliaApiException e) {
+              if (e.getStatusCode() == 404) {
+                return null;
+              }
+
+              throw e;
             }
-    
-            throw e;
-          }
-        },
-        (Event response) -> {
-            return response != null;
-        },
-        50,
-        null
-      )
+          },
+          (Event resp) -> {
+            return resp != null;
+          },
+          50,
+          null
+        );
+      }
     );
   }
 
@@ -815,7 +823,7 @@ public <T> List<WatchResponse> saveObjectsWithTransformation(
   int batchSize,
   RequestOptions requestOptions
 ) {
-  return chunkedPush(indexName, objects, Action.ADD_OBJECT, waitForTasks, batchSize, null, requestOptions);
+  return chunkedPush(indexName, objects, com.algolia.model.ingestion.Action.ADD_OBJECT, waitForTasks, batchSize, null, requestOptions);
 }
 
 private <T> List<PushTaskRecords> objectsToPushTaskRecords(Iterable<T> objects) {
@@ -1048,9 +1056,10 @@ public <T> List<WatchResponse> partialUpdateObjectsWithTransformation(
   return chunkedPush(
     indexName,
     objects,
-    createIfNotExists ? Action.PARTIAL_UPDATE_OBJECT : Action.PARTIAL_UPDATE_OBJECT_NO_CREATE,
+    createIfNotExists ? com.algolia.model.ingestion.Action.PARTIAL_UPDATE_OBJECT : com.algolia.model.ingestion.Action.PARTIAL_UPDATE_OBJECT_NO_CREATE,
     waitForTasks,
     batchSize,
+    null,
     requestOptions
   );
 }