修复创建mongodb原生连接时的错误 (hhyo#1291)

weideguo · Dec 22, 2021 · 70bf9b6 · 70bf9b6
1 parent e5e777b
commit 70bf9b6
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/sql/engines/mongo.py b/sql/engines/mongo.py
@@ -503,11 +503,12 @@ def execute_check(self, db_name=None, sql=''):
         return check_result
 
     def get_connection(self, db_name=None):
-        self.db_name = db_name or self.instance.db_name
-        self.conn = pymongo.MongoClient(self.host, self.port, authSource=self.db_name, connect=True,
+        self.db_name = db_name or self.instance.db_name or 'admin'
+        auth_db = self.instance.db_name or 'admin'
+        self.conn = pymongo.MongoClient(self.host, self.port, authSource=auth_db, connect=True,
                                         connectTimeoutMS=10000)
         if self.user and self.password:
-            self.conn[self.db_name].authenticate(self.user, self.password, self.db_name)
+            self.conn[self.db_name].authenticate(self.user, self.password, auth_db)
         return self.conn
 
     def close(self):