add flag for includeSoftDeleted in scroll entities API (#10831)

datahub-project · Jul 2, 2024 · 286a3cc · 286a3cc
1 parent 3175571
commit 286a3cc
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/...servlet/src/main/java/io/datahubproject/openapi/controller/GenericEntitiesController.java b/...servlet/src/main/java/io/datahubproject/openapi/controller/GenericEntitiesController.java
@@ -147,7 +147,9 @@ public ResponseEntity<S> getEntities(
       @RequestParam(value = "systemMetadata", required = false, defaultValue = "false")
           Boolean withSystemMetadata,
       @RequestParam(value = "skipCache", required = false, defaultValue = "false")
-          Boolean skipCache)
+          Boolean skipCache,
+      @RequestParam(value = "includeSoftDeleted", required = false, defaultValue = "false")
+          Boolean includeSoftDeleted)
       throws URISyntaxException {
 
     EntitySpec entitySpec = entityRegistry.getEntitySpec(entityName);
@@ -175,7 +177,8 @@ public ResponseEntity<S> getEntities(
         searchService.scrollAcrossEntities(
             opContext
                 .withSearchFlags(flags -> DEFAULT_SEARCH_FLAGS)
-                .withSearchFlags(flags -> flags.setSkipCache(skipCache)),
+                .withSearchFlags(flags -> flags.setSkipCache(skipCache))
+                .withSearchFlags(flags -> flags.setIncludeSoftDeleted(includeSoftDeleted)),
             List.of(entitySpec.getName()),
             query,
             null,