pymongo连接超时的做法

来源:互联网 发布:spark编程模型 编辑:程序博客网 时间:2024/04/28 15:05

通常我们连接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连接等待,难怪连接不上卡这么久。

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这么久