Aggregate operation is throwing MongoExecutionTimeoutException in pro…

…d which is not getting caught and logged
hypertrace · Jan 21, 2025 · 6b406de · 6b406de
1 parent 049e1bf
commit 6b406de
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/...store/src/main/java/org/hypertrace/core/documentstore/mongo/query/MongoQueryExecutor.java b/...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;
     }