From c2eec6db2d3de10c4145ff450fcd19f2760f2626 Mon Sep 17 00:00:00 2001
From: Mihir Gore <133626129+mihirgt@users.noreply.github.com>
Date: Tue, 21 Jan 2025 11:22:41 +0530
Subject: [PATCH] Aggregate operation is throwing
 MongoExecutionTimeoutException in prod which is not getting caught and logged
 (#218)

---
 .../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;
     }