Python连接MongoDB耗时25秒，VS Code插件秒连，求解决建议

问题场景 #

使用以下Python代码连接MongoDB并插入数据时，创建MongoClient实例的步骤耗时长达25秒，但相同数据库、电脑、IP及URI条件下，MongoDB VS Code扩展点击「连接」仅需1秒即可成功建立连接：

uri = f"mongodb+srv://username:password@cluster0.xxxxxxx.mongodb.net/?appName=ClusterX"
client = MongoClient(uri, server_api=ServerApi('1'))
# 执行到此处耗时25秒
db = client.collections
db.collection.insert_one({"x": 11}).inserted_id
# 执行到此处耗时25.3秒
db = client.collections
db.collection.insert_one({"x": 12}).inserted_id
# 执行到此处耗时25.315秒
db = client.collections
db.collection.insert_one({"x": 12}).inserted_id
# 执行到此处耗时25.330秒

可能的问题原因 #

1. SRV DNS解析效率差异：mongodb+srv格式的URI需要解析SRV记录，Python pymongo默认的DNS解析逻辑与VS Code扩展使用的解析方式（如系统原生解析、缓存机制）不同，导致解析耗时过长。
2. pymongo版本陈旧：旧版本的pymongo在SRV记录解析、连接握手等环节存在性能瓶颈，而VS Code扩展使用了更新的驱动版本。
3. 默认连接参数未优化：MongoClient默认的连接超时、服务器心跳检查等参数设置保守，初始连接时触发了不必要的等待或检查流程；VS Code扩展则默认启用了更激进的优化参数。
4. SSL/TLS握手延迟：Python的SSL配置可能未使用系统证书池，导致证书验证环节耗时增加，VS Code扩展则直接复用系统证书配置。

解决建议 #

1. 优化DNS解析 #

安装dnspython库替代默认解析逻辑，提升SRV记录解析速度：

pip install dnspython

修改代码指定DNS解析器：

import dnspython.resolver
from pymongo import MongoClient
from pymongo.server_api import ServerApi

uri = f"mongodb+srv://username:password@cluster0.xxxxxxx.mongodb.net/?appName=ClusterX"
client = MongoClient(
    uri,
    server_api=ServerApi('1'),
    dns_resolver=dnspython.resolver.Resolver()
)

2. 升级pymongo版本 #

更新到最新稳定版pymongo，修复旧版本的性能问题：

pip install --upgrade pymongo

3. 调整连接参数 #

设置更合理的超时与心跳参数，减少初始连接的等待时间：

client = MongoClient(
    uri,
    server_api=ServerApi('1'),
    connectTimeoutMS=5000,    # 连接超时设为5秒
    socketTimeoutMS=5000,     # 套接字超时设为5秒
    serverSelectionTimeoutMS=5000,  # 服务器选择超时设为5秒
    heartbeatFrequencyMS=30000      # 心跳检查频率改为30秒（减少初始连接开销）
)

4. 改用标准连接字符串（跳过SRV解析） #

如果已知集群的具体节点地址，直接使用标准连接字符串，避免SRV解析步骤：

# 替换为实际集群的节点地址
uri = "mongodb://username:password@cluster0-shard-00-00.xxxxxxx.mongodb.net:27017,cluster0-shard-00-01.xxxxxxx.mongodb.net:27017,cluster0-shard-00-02.xxxxxxx.mongodb.net:27017/?ssl=true&replicaSet=atlas-xxxxxx-shard-0&authSource=admin&retryWrites=true&w=majority"
client = MongoClient(uri, server_api=ServerApi('1'))

5. 验证DNS解析耗时 #

单独测试SRV记录的解析耗时，确认是否为DNS问题：

import dnspython.resolver
import time

start = time.time()
answers = dnspython.resolver.resolve('_mongodb._tcp.cluster0.xxxxxxx.mongodb.net', 'SRV')
print(f"DNS解析耗时: {time.time() - start:.2f}秒")
for rdata in answers:
    print(f"目标: {rdata.target}, 端口: {rdata.port}")

内容的提问来源于stack exchange，提问作者iamnomadgg