feat(repository-elasticsearch): add a raw delete query

kestra-io · Jul 4, 2022 · f305477 · f305477
1 parent 2e32fce
commit f305477
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/...rch/src/main/java/io/kestra/repository/elasticsearch/AbstractElasticSearchRepository.java b/...rch/src/main/java/io/kestra/repository/elasticsearch/AbstractElasticSearchRepository.java
@@ -19,6 +19,8 @@
 import org.opensearch.action.ShardOperationFailedException;
 import org.opensearch.action.admin.indices.settings.get.GetSettingsRequest;
 import org.opensearch.action.admin.indices.settings.get.GetSettingsResponse;
+import org.opensearch.action.delete.DeleteRequest;
+import org.opensearch.action.delete.DeleteResponse;
 import org.opensearch.action.get.GetRequest;
 import org.opensearch.action.get.GetResponse;
 import org.opensearch.action.index.IndexRequest;
@@ -237,6 +239,20 @@ protected UpdateResponse deleteRequest(String index, String id) {
         return this.updateRequest(index, id, delete);
     }
 
+    protected DeleteResponse rawDeleteRequest(String index, String id) {
+        DeleteRequest request = new DeleteRequest(this.indicesConfigs.get(index).getIndex(), id);
+        request.setRefreshPolicy(WriteRequest.RefreshPolicy.IMMEDIATE);
+
+        try {
+            DeleteResponse response = client.delete(request, RequestOptions.DEFAULT);
+            handleWriteErrors(response);
+
+            return response;
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
+
     protected SearchRequest searchRequest(String index, SearchSourceBuilder sourceBuilder, boolean scroll) {
         SearchRequest searchRequest = new SearchRequest()
             .indices(this.indicesConfigs.get(index).getIndex())