mongo aggregate查询加limit行数限制，防止返回结果过多导致archery挂掉 (#1948)

* mongo aggregate查询加limit行数限制，防止返回结果过多导致archery挂掉
hhyo · Nov 4, 2022 · 027904f · 027904f
1 parent 3ff5c95
commit 027904f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sql/engines/mongo.py b/sql/engines/mongo.py
@@ -934,6 +934,8 @@ def query(self, db_name=None, sql="", limit_num=0, close_conn=True, **kwargs):
         if "condition" in query_dict:
             if method == "aggregate":
                 condition = query_dict["condition"]
+                # 给aggregate查询加limit行数限制，防止返回结果过多导致archery挂掉
+                condition.append({"$limit": limit_num})
             if method == "find":
                 condition = de.decode(query_dict["condition"])
             find_cmd += "(condition)"