通常我们连接mongo数据库会这样做:
client = pymongo.MongoClient(host, port)
但是会发现一个问题,如果mongo连接不上,会在这里阻塞很久。解决办法可以看官方文档http://api.mongodb.com/python/current/api/pymongo/mongo_client.html#pymongo.mongo_client.MongoClient
MongoClient提供了一个serverSelectionTimeoutMS选项,从这里我们可以看到默认是30s连接等待,难怪连接不上卡这么久
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
def connect(self, host, port): count = 0 while True: self.client = MongoClient(host, port, serverSelectionTimeoutMS=3) try: self.client.admin.command("ping") except ConnectionFailure, err_msg: log.error(err_msg) count = count + 1 else: break if count == 3: return False return True |
优化后,我们可以快速知道是否连接成功,并且尝试3次,不用阻塞30s这么久
未经允许不得转载:MR LYU » pymongo连接超时的做法