pymongo连接超时的做法

通常我们连接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

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

发表评论 取消回复

发表评论取消回复