update dryOps method in partial import

Shivam-z · Jul 3, 2024 · 1f5ab41 · 1f5ab41
1 parent bf4278a
commit 1f5ab41
Showing 1 changed file with 3 additions and 17 deletions.
diff --git a/...rc/main/java/com/appsmith/server/imports/internal/partial/PartialImportServiceCEImpl.java b/...rc/main/java/com/appsmith/server/imports/internal/partial/PartialImportServiceCEImpl.java
@@ -249,23 +249,9 @@ private Mono<BuildingBlockImportDTO> importResourceInPage(
                                         .thenReturn(application);
                             });
                 })
-                // execute dry run for datasource
-                .flatMap(importableArtifact -> Flux.fromIterable(mappedImportableResourcesDTO
-                                .getDatasourceDryRunQueries()
-                                .keySet())
-                        .flatMap(key -> dryOperationRepository.saveDatasourceToDb(mappedImportableResourcesDTO
-                                .getDatasourceDryRunQueries()
-                                .get(key)))
-                        .collectList()
-                        .thenReturn(importableArtifact))
-                // execute dryOps for datasourceStorage
-                .flatMap(importableArtifact -> Flux.fromIterable(mappedImportableResourcesDTO
-                                .getDatasourceStorageDryRunQueries()
-                                .keySet())
-                        .flatMap(key -> dryOperationRepository.saveDatasourceStorageToDb(mappedImportableResourcesDTO
-                                .getDatasourceStorageDryRunQueries()
-                                .get(key)))
-                        .collectList()
+                // execute dry run ops
+                .flatMap(importableArtifact -> dryOperationRepository
+                        .executeAllDbOps(mappedImportableResourcesDTO)
                         .thenReturn(importableArtifact))
                 .flatMap(application -> {
                     Map<String, Object> fieldNameValueMap = Map.of(