feat(java): add bridge to transformation on search

Author: shortcuts

clients/algoliasearch-client-java/algoliasearch/src/main/java/com/algolia/ApiClient.java

@@ -28,7 +28,7 @@ public abstract class ApiClient implements Closeable {
 
   private final Requester requester;
   private final ExecutorService executor;
-  private AuthInterceptor authInterceptor;
+  public AuthInterceptor authInterceptor;
 
   /** Constructs a new instance of the {@link ApiClient}. */
   protected ApiClient(

clients/algoliasearch-client-java/algoliasearch/src/main/java/com/algolia/internal/interceptors/AuthInterceptor.java

@@ -23,6 +23,14 @@ public void setApiKey(String apiKey) {
     this.apiKey = apiKey;
   }
 
+  public String getApiKey() {
+    return this.apiKey;
+  }
+
+  public String getApplicationId() {
+    return this.applicationId;
+  }
+
   @Nonnull
   @Override
   public Response intercept(Chain chain) throws IOException {

specs/search/helpers/saveObjectsWithTransformation.yml

@@ -5,6 +5,7 @@ method:
       - javascript
       - go
       - python
+      - java
     tags:
       - Records
     operationId: saveObjectsWithTransformation

templates/java/api.mustache

@@ -45,10 +45,21 @@ import java.util.Base64;
 import javax.annotation.Nonnull;
 import javax.crypto.Mac;
 import javax.crypto.spec.SecretKeySpec;
+import com.algolia.model.ingestion.PushTaskPayload;
+import com.algolia.model.ingestion.PushTaskRecords;
+import com.algolia.model.ingestion.WatchResponse;
 {{/isSearchClient}}
 
 {{#operations}}
 public class {{classname}} extends ApiClient {
+    {{#isSearchClient}}
+    private IngestionClient ingestionTransporter;
+
+    public void setTransformationRegion(String region) {
+      this.ingestionTransporter = new IngestionClient(this.authInterceptor.getApplicationId(), this.authInterceptor.getApiKey(), region);
+    }
+    {{/isSearchClient}}
+
     {{#hasRegionalHost}}
     private static final String[] allowedRegions = { {{#allowedRegions}}"{{.}}"{{^-last}},{{/-last}}{{/allowedRegions}} };
 
@@ -243,4 +254,4 @@ public class {{classname}} extends ApiClient {
 
   {{> api_helpers}}
 }
-{{/operations}}
+{{/operations}}

templates/java/api_helpers.mustache

@@ -653,6 +653,109 @@ public <T> List<BatchResponse> chunkedBatch(
   return chunkedBatch(indexName, objects, action, waitForTasks, 1000, requestOptions);
 }
 
+/**
+ * Helper: Similar to the `saveObjects` method but requires a Push connector
+ * (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/)
+ * to be created first, in order to transform records before indexing them to Algolia. The
+ * `region` must have been passed to the client instantiation method.
+ *
+ * @param indexName The `indexName` to replace `objects` in.
+ * @param objects The array of `objects` to store in the given Algolia `indexName`.
+ * @throws AlgoliaRetryException When the retry has failed on all hosts
+ * @throws AlgoliaApiException When the API sends an http error code
+ * @throws AlgoliaRuntimeException When an error occurred during the serialization
+ */
+public <T> WatchResponse saveObjectsWithTransformation(String indexName, Iterable<T> objects) {
+  return saveObjectsWithTransformation(indexName, objects, null);
+}
+
+/**
+ * Helper: Similar to the `saveObjects` method but requires a Push connector
+ * (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/)
+ * to be created first, in order to transform records before indexing them to Algolia. The
+ * `region` must have been passed to the client instantiation method.
+ *
+ * @param indexName The `indexName` to replace `objects` in.
+ * @param objects The array of `objects` to store in the given Algolia `indexName`.
+ * @param requestOptions The requestOptions to send along with the query, they will be merged with
+ *     the transporter requestOptions. (optional)
+ */
+public <T> WatchResponse saveObjectsWithTransformation(String indexName, Iterable<T> objects, RequestOptions requestOptions) {
+  return saveObjectsWithTransformation(indexName, objects, false, requestOptions);
+}
+
+/**
+ * Helper: Similar to the `saveObjects` method but requires a Push connector
+ * (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/)
+ * to be created first, in order to transform records before indexing them to Algolia. The
+ * `region` must have been passed to the client instantiation method.
+ *
+ * @param indexName The `indexName` to replace `objects` in.
+ * @param objects The array of `objects` to store in the given Algolia `indexName`.
+ * @param waitForTasks - Whether or not we should wait until every `batch` tasks has been
+ *     processed, this operation may slow the total execution time of this method but is more
+ *     reliable.
+ * @param requestOptions The requestOptions to send along with the query, they will be merged with
+ *     the transporter requestOptions. (optional)
+ */
+public <T> WatchResponse saveObjectsWithTransformation(
+  String indexName,
+  Iterable<T> objects,
+  boolean waitForTasks,
+  RequestOptions requestOptions
+) {
+  return saveObjectsWithTransformation(indexName, objects, waitForTasks, 1000, requestOptions);
+}
+
+/**
+ * Helper: Similar to the `saveObjects` method but requires a Push connector
+ * (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/)
+ * to be created first, in order to transform records before indexing them to Algolia. The
+ * `region` must have been passed to the client instantiation method.
+ *
+ * @param indexName The `indexName` to replace `objects` in.
+ * @param objects The array of `objects` to store in the given Algolia `indexName`.
+ * @param waitForTasks - Whether or not we should wait until every `batch` tasks has been
+ *     processed, this operation may slow the total execution time of this method but is more
+ *     reliable.
+ * @param batchSize The size of the chunk of `objects`. The number of `batch` calls will be equal
+ *     to `length(objects) / batchSize`.
+ * @param requestOptions The requestOptions to send along with the query, they will be merged with
+ *     the transporter requestOptions. (optional)
+ */
+public <T> WatchResponse saveObjectsWithTransformation(
+  String indexName,
+  Iterable<T> objects,
+  boolean waitForTasks,
+  int batchSize,
+  RequestOptions requestOptions
+) {
+  if (this.ingestionTransporter == null) {
+    throw new AlgoliaRuntimeException("`setTransformationRegion` must have been called before calling this method.");
+  }
+
+  return this.ingestionTransporter.push(
+      indexName,
+      new PushTaskPayload().setAction(com.algolia.model.ingestion.Action.ADD_OBJECT).setRecords(this.objectsToPushTaskRecords(objects)),
+      waitForTasks,
+      requestOptions
+    );
+}
+
+private <T> List<PushTaskRecords> objectsToPushTaskRecords(Iterable<T> iterable) {
+  List<PushTaskRecords> payload = new ArrayList<>();
+  for (T item : iterable) {
+    if (item instanceof PushTaskRecords) {
+      payload.add((PushTaskRecords) item);
+    } else {
+      throw new AlgoliaRuntimeException(
+        "each object must have an `objectID` key in order to be used with the" + " WithTransformation methods"
+      );
+    }
+  }
+  return payload;
+}
+
 /**
  * Helper: Saves the given array of objects in the given index. The `chunkedBatch` helper is used
  * under the hood, which creates a `batch` requests with at most 1000 objects in it.
@@ -777,6 +880,114 @@ public List<BatchResponse> deleteObjects(String indexName, List<String> objectID
   return chunkedBatch(indexName, objects, Action.DELETE_OBJECT, waitForTasks, batchSize, requestOptions);
 }
 
+/**
+ * Helper: Similar to the `partialUpdateObjects` method but requires a Push connector
+ * (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/)
+ * to be created first, in order to transform records before indexing them to Algolia. The
+ * `region` must have been passed to the client instantiation method.
+ *
+ * @param indexName The `indexName` to update `objects` in.
+ * @param objects The array of `objects` to update in the given Algolia `indexName`.
+ * @param createIfNotExists To be provided if non-existing objects are passed, otherwise, the call
+ *     will fail.
+ */
+public <T> WatchResponse partialUpdateObjectsWithTransformation(String indexName, Iterable<T> objects, boolean createIfNotExists) {
+  return partialUpdateObjectsWithTransformation(indexName, objects, createIfNotExists, false, null);
+}
+
+/**
+ * Helper: Similar to the `partialUpdateObjects` method but requires a Push connector
+ * (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/)
+ * to be created first, in order to transform records before indexing them to Algolia. The
+ * `region` must have been passed to the client instantiation method.
+ *
+ * @param indexName The `indexName` to update `objects` in.
+ * @param objects The array of `objects` to update in the given Algolia `indexName`.
+ * @param createIfNotExists To be provided if non-existing objects are passed, otherwise, the call
+ *     will fail.
+ * @param waitForTasks - Whether or not we should wait until every `batch` tasks has been
+ *     processed, this operation may slow the total execution time of this method but is more
+ *     reliable.
+ */
+public <T> WatchResponse partialUpdateObjectsWithTransformation(
+  String indexName,
+  Iterable<T> objects,
+  boolean createIfNotExists,
+  boolean waitForTasks
+) {
+  return partialUpdateObjectsWithTransformation(indexName, objects, createIfNotExists, waitForTasks, null);
+}
+
+/**
+ * Helper: Similar to the `partialUpdateObjects` method but requires a Push connector
+ * (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/)
+ * to be created first, in order to transform records before indexing them to Algolia. The
+ * `region` must have been passed to the client instantiation method.
+ *
+ * @param indexName The `indexName` to update `objects` in.
+ * @param objects The array of `objects` to update in the given Algolia `indexName`.
+ * @param createIfNotExists To be provided if non-existing objects are passed, otherwise, the call
+ *     will fail.
+ * @param waitForTasks - Whether or not we should wait until every `batch` tasks has been
+ *     processed, this operation may slow the total execution time of this method but is more
+ *     reliable.
+ * @param requestOptions The requestOptions to send along with the query, they will be merged with
+ *     the transporter requestOptions. (optional)
+ */
+public <T> WatchResponse partialUpdateObjectsWithTransformation(
+  String indexName,
+  Iterable<T> objects,
+  boolean createIfNotExists,
+  boolean waitForTasks,
+  RequestOptions requestOptions
+) {
+  return partialUpdateObjectsWithTransformation(indexName, objects, createIfNotExists, waitForTasks, 1000, null);
+}
+
+/**
+ * Helper: Similar to the `partialUpdateObjects` method but requires a Push connector
+ * (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/)
+ * to be created first, in order to transform records before indexing them to Algolia. The
+ * `region` must have been passed to the client instantiation method.
+ *
+ * @param indexName The `indexName` to update `objects` in.
+ * @param objects The array of `objects` to update in the given Algolia `indexName`.
+ * @param createIfNotExists To be provided if non-existing objects are passed, otherwise, the call
+ *     will fail.
+ * @param waitForTasks - Whether or not we should wait until every `batch` tasks has been
+ *     processed, this operation may slow the total execution time of this method but is more
+ *     reliable.
+ * @param batchSize The size of the chunk of `objects`. The number of `batch` calls will be equal
+ *     to `length(objects) / batchSize`.
+ * @param requestOptions The requestOptions to send along with the query, they will be merged with
+ *     the transporter requestOptions. (optional)
+ */
+public <T> WatchResponse partialUpdateObjectsWithTransformation(
+  String indexName,
+  Iterable<T> objects,
+  boolean createIfNotExists,
+  boolean waitForTasks,
+  int batchSize,
+  RequestOptions requestOptions
+) {
+  if (this.ingestionTransporter == null) {
+    throw new AlgoliaRuntimeException("`setTransformationRegion` must have been called before calling this method.");
+  }
+
+  return this.ingestionTransporter.push(
+      indexName,
+      new PushTaskPayload()
+        .setAction(
+          createIfNotExists
+            ? com.algolia.model.ingestion.Action.PARTIAL_UPDATE_OBJECT
+            : com.algolia.model.ingestion.Action.PARTIAL_UPDATE_OBJECT_NO_CREATE
+        )
+        .setRecords(this.objectsToPushTaskRecords(objects)),
+      waitForTasks,
+      requestOptions
+    );
+}
+
 /**
  * Helper: Replaces object content of all the given objects according to their respective
  * `objectID` field. The `chunkedBatch` helper is used under the hood, which creates a `batch`