From 76de6761d6c7b63ea3c45480876e9457f4e0600a Mon Sep 17 00:00:00 2001 From: Shrikant Sharat Kandula Date: Sat, 2 Mar 2024 18:46:08 +0530 Subject: [PATCH] chore: Use queryBuilder for count op (#31328) --- .../repositories/ce/CustomNewActionRepositoryCEImpl.java | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) 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 30129f6a59..4f259e58d7 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 @@ -21,7 +21,6 @@ import org.springframework.data.mongodb.core.aggregation.MatchOperation; import org.springframework.data.mongodb.core.aggregation.ProjectionOperation; import org.springframework.data.mongodb.core.convert.MongoConverter; import org.springframework.data.mongodb.core.query.Criteria; -import org.springframework.data.mongodb.core.query.Query; import org.springframework.data.mongodb.core.query.Update; import reactor.core.publisher.Flux; import reactor.core.publisher.Mono; @@ -345,12 +344,9 @@ public class CustomNewActionRepositoryCEImpl extends BaseAppsmithRepositoryImpl< where(NewAction.Fields.publishedAction + ".datasource._id").is(new ObjectId(datasourceId)); Criteria datasourceCriteria = - notDeleted().orOperator(unpublishedDatasourceCriteria, publishedDatasourceCriteria); + new Criteria().orOperator(unpublishedDatasourceCriteria, publishedDatasourceCriteria); - Query query = new Query(); - query.addCriteria(datasourceCriteria); - - return mongoOperations.count(query, NewAction.class); + return queryBuilder().criteria(datasourceCriteria).count(); } @Override