今天在做代码优化的时候发现pymongo版本不同会出现问题,正常3.*能跑的代码,一旦使用4.2或者4.3版本(其他版本没有试,但应该也有问题)就会出现 ServerSelectionTimeoutError问题,最后到官网找到了问题所在:PyMongo 4 Migration Guide — PyMongo 4.3.0 documentation
原因总结
在PyMongo 4.x版本中,directConnection
URI选项和关键字参数的默认值从None
变为了False
。这一变化意味着PyMongo现在默认尝试自动发现副本集,而不再直接连接到单一的MongoDB服务器。
因此,如果你希望与单个服务器建立直接连接,你需要明确地将directConnection
设置为True
,作为URI选项或关键字参数传递给MongoClient
。
如果你在从PyMongo 3升级到4.x版本后遇到ServerSelectionTimeoutError
错误,那么很可能是因为自动发现副本集的默认行为导致的问题。在这种情况下,你需要在创建客户端时添加directConnection=True
。
解决办法
就如官网所说,加上directConnection=True就可以解决问题,
首先,我们需要确保原始URI的格式是正确的,特别是要注意是否已存在其他查询参数。如果URI已经包含查询参数(如authSource=admin
),我们应该使用&
来连接新的参数;如果URI中还没有任何查询参数,我们应该使用?
来开始参数列表。
下面是一个示例,展示了如何安全地在已存在的URI中添加directConnection=true
参数:
# 假设这是您的原始MongoDB URI,它可能包含一个或多个查询参数
mongo_uri = "mongodb://username:password@host:port/database?authSource=admin" # 检查URI是否已经包含查询参数(即是否已经有'?')
if '?' in mongo_uri: # 如果已经有查询参数,使用'&'来添加新的参数 mongo_uri_with_direct_connection = f"{mongo_uri}&directConnection=true"
else: # 如果没有查询参数,使用'?'来开始参数列表 mongo_uri_with_direct_connection = f"{mongo_uri}?directConnection=true" # 现在mongo_uri_with_direct_connection包含了directConnection=true参数
print(mongo_uri_with_direct_connection) # 使用构建好的URI创建MongoClient实例
from pymongo import MongoClient
client = MongoClient(mongo_uri_with_direct_connection) # 现在你可以使用client进行数据库操作了
db = client['your_database_name'] # 替换为你的数据库名
collection = db['your_collection_name'] # 替换为你的集合名
# ... 进行其他操作 ...
在上面的代码中,我们首先定义了一个包含用户名、密码、主机、端口、数据库名和认证源的MongoDB URI。然后,我们检查这个URI是否已包含查询参数。根据检查结果,我们使用&
或?
来添加directConnection=true
参数。最后,我们打印出更新后的URI,并使用它来创建MongoClient
实例,以便连接到MongoDB数据库。
总结:
在Python中,为MongoDB URI添加额外的参数(如directConnection=true
)是一个常见的需求。为了确保URI的格式正确,我们需要根据URI中是否已存在其他查询参数来决定使用&
还是?
来连接新参数。通过上面的示例代码,我们可以轻松地在已存在的URI中添加所需的参数,并使用更新后的URI来建立MongoDB连接。这样,我们就可以根据实际需求优化连接性能或解决特定问题了。