diff --git a/app/server/appsmith-server/src/main/java/com/appsmith/server/repositories/ce/CustomNewActionRepositoryCE.java b/app/server/appsmith-server/src/main/java/com/appsmith/server/repositories/ce/CustomNewActionRepositoryCE.java index 80351b502b01..debbd0b8b5d9 100644 --- a/app/server/appsmith-server/src/main/java/com/appsmith/server/repositories/ce/CustomNewActionRepositoryCE.java +++ b/app/server/appsmith-server/src/main/java/com/appsmith/server/repositories/ce/CustomNewActionRepositoryCE.java @@ -80,4 +80,14 @@ Flux findAllPublishedActionsByContextIdAndContextType( String contextId, CreatorContextType contextType, AclPermission permission, boolean includeJs); Flux findAllByApplicationIds(List branchedArtifactIds, List includedFields); + + // @Meta(cursorBatchSize = 10000) + // TODO Implement cursor with batch size + Flux findByApplicationId(String applicationId); + + // @Meta(cursorBatchSize = 10000) + // TODO Implement cursor with batch size + Flux findAllByIdIn(Iterable ids); + + Mono countByDeletedAtNull(); } diff --git a/app/server/appsmith-server/src/main/java/com/appsmith/server/repositories/ce/CustomNewActionRepositoryCEImpl.java b/app/server/appsmith-server/src/main/java/com/appsmith/server/repositories/ce/CustomNewActionRepositoryCEImpl.java index 0d56c4dcabb8..33d511dd0990 100644 --- a/app/server/appsmith-server/src/main/java/com/appsmith/server/repositories/ce/CustomNewActionRepositoryCEImpl.java +++ b/app/server/appsmith-server/src/main/java/com/appsmith/server/repositories/ce/CustomNewActionRepositoryCEImpl.java @@ -29,6 +29,7 @@ import java.util.List; import java.util.Optional; import java.util.Set; +import java.util.stream.StreamSupport; import static org.springframework.data.mongodb.core.aggregation.Aggregation.group; import static org.springframework.data.mongodb.core.aggregation.Aggregation.match; @@ -488,4 +489,24 @@ public Flux findAllByApplicationIds(List applicationIds, List .fields(includedFields) .all(); } + + @Override + public Flux findByApplicationId(String applicationId) { + return queryBuilder() + .criteria(Bridge.equal(NewAction.Fields.applicationId, applicationId)) + .all(); + } + + @Override + public Flux findAllByIdIn(Iterable ids) { + List idList = StreamSupport.stream(ids.spliterator(), false).toList(); + return queryBuilder().criteria(Bridge.in(NewAction.Fields.id, idList)).all(); + } + + @Override + public Mono countByDeletedAtNull() { + return queryBuilder() + .criteria(Bridge.exists(NewAction.Fields.deletedAt)) + .count(); + } } diff --git a/app/server/appsmith-server/src/main/java/com/appsmith/server/repositories/ce/NewActionRepositoryCE.java b/app/server/appsmith-server/src/main/java/com/appsmith/server/repositories/ce/NewActionRepositoryCE.java index 9b270f2fcb4f..f3bdb2c807ed 100644 --- a/app/server/appsmith-server/src/main/java/com/appsmith/server/repositories/ce/NewActionRepositoryCE.java +++ b/app/server/appsmith-server/src/main/java/com/appsmith/server/repositories/ce/NewActionRepositoryCE.java @@ -5,22 +5,12 @@ import com.appsmith.server.projections.IdPoliciesOnly; import com.appsmith.server.repositories.BaseRepository; import com.appsmith.server.repositories.CustomNewActionRepository; -import org.springframework.data.mongodb.repository.Meta; import reactor.core.publisher.Flux; -import reactor.core.publisher.Mono; import java.util.List; public interface NewActionRepositoryCE extends BaseRepository, CustomNewActionRepository { - @Meta(cursorBatchSize = 10000) - Flux findByApplicationId(String applicationId); - - @Meta(cursorBatchSize = 10000) - Flux findAllByIdIn(Iterable ids); - - Mono countByDeletedAtNull(); - Flux findIdsAndPolicyMapByApplicationIdIn(List applicationIds); Flux findIdAndDatasourceIdByApplicationIdIn(List applicationIds);