From 6b406debb265972806d4ea0aec50091ea98a7ba7 Mon Sep 17 00:00:00 2001
From: Mihir Gore <mihir.gore@traceable.ai>
Date: Tue, 21 Jan 2025 09:58:03 +0530
Subject: [PATCH] Aggregate operation is throwing
 MongoExecutionTimeoutException in prod which is not getting caught and logged

---
 .../core/documentstore/mongo/query/MongoQueryExecutor.java     | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/document-store/src/main/java/org/hypertrace/core/documentstore/mongo/query/MongoQueryExecutor.java b/document-store/src/main/java/org/hypertrace/core/documentstore/mongo/query/MongoQueryExecutor.java
index f9721903..40890779 100644
--- a/document-store/src/main/java/org/hypertrace/core/documentstore/mongo/query/MongoQueryExecutor.java
+++ b/document-store/src/main/java/org/hypertrace/core/documentstore/mongo/query/MongoQueryExecutor.java
@@ -23,7 +23,6 @@
 import static org.hypertrace.core.documentstore.mongo.query.parser.MongoSortTypeExpressionParser.getSortClause;
 
 import com.mongodb.BasicDBObject;
-import com.mongodb.MongoCommandException;
 import com.mongodb.ServerAddress;
 import com.mongodb.ServerCursor;
 import com.mongodb.client.AggregateIterable;
@@ -181,7 +180,7 @@ public MongoCursor<BasicDBObject> aggregate(
               .allowDiskUse(true);
 
       return iterable.cursor();
-    } catch (final MongoCommandException e) {
+    } catch (final Exception e) {
       log.error("Execution failed for query: {}. Aggregation Pipeline: {}", query, pipeline);
       throw e;
     }