远程MongoDB连接正常但Python聚合触发ServerSelectionTimeoutError

1. 调整MongoClient超时参数 #

Python的pymongo驱动默认serverSelectionTimeoutMS为3000ms，若聚合操作因网络微小延迟或数据处理耗时触发超时，可显式延长超时阈值：

from pymongo import MongoClient

client = MongoClient(
    connection_string,
    serverSelectionTimeoutMS=10000,  # 延长至10秒
    socketTimeoutMS=60000,  # 套接字超时设为60秒
    connectTimeoutMS=10000
)
# 聚合时额外指定操作超时
result = collection.aggregate(pipeline, maxTimeMS=60000)

2. 统一读取偏好设置 #

Compass默认使用primaryPreferred读取偏好（优先从主节点读取，主节点不可用时切换到从节点），而pymongo默认仅从主节点读取。若主节点负载过高，会导致Python端聚合超时：

from pymongo import ReadPreference

client = MongoClient(connection_string, read_preference=ReadPreference.PRIMARY_PREFERRED)

3. 优化聚合批处理大小 #

Compass会自动优化结果分批获取逻辑，而pymongo默认批处理尺寸较小，多次网络请求易累积超时。可手动指定更大的批处理大小：

result = collection.aggregate(pipeline, batchSize=1000)
# 避免一次性加载所有结果，逐个处理文档
for doc in result:
    # 处理逻辑
    pass

4. 验证驱动与服务器版本兼容性 #

pymongo版本与MongoDB服务器不兼容时，可能导致聚合协议处理异常引发超时。先通过db.version()查看服务器版本，再安装对应兼容的驱动版本：

# 示例：服务器为5.0.x时安装pymongo 4.x
pip install pymongo~=4.0

5. 启用TCP KeepAlive #

尽管ufw已开放端口，中间网络设备可能因长时间无数据传输断开连接。Compass默认启用TCP KeepAlive，而pymongo需手动配置：

client = MongoClient(
    connection_string,
    socketKeepAlive=True,
    tcpKeepAliveTime=30,  # 每30秒发送保活包
    tcpKeepAliveInterval=10
)

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