#python怎样实现redis断开后自动重连的机制
近来在做的一个项目,利用redis
实现消息队列,在发布端用lpush
,将数据写入到队列中,在订阅端用rpop
方法依次读出每条数据并处理,需要在windows服务中循环读取redis
里的数据并做进一步处理。
在这过程中遇到一个比较麻烦的问题,就是redis
的连接每过一段时间会被redis服务器主动断开,查看redis
的日志,记录的原因是psubscribe scheduled to be closed ASAP for overcoming of output buffer limits
,刚开始的时候发现确实在处理消息时耗时太长,引起队列堵塞,导致缓冲区的内容过多,超出了redis连接的内存限制而被强制断开了,经过优化后处理每条消息的时长已经缩短到3毫秒左右,但偶尔仍然会出现redis连接被强制断开的情况。
考虑到这个服务要常驻在系统中的,就算redis
服务器不主动断开连接,也有可能会出现redis
服务器宕机或需要重启的情况,所以要建立redis
连接断开后自动重连的机制比较可靠,于是写了一个getRedis
方法,当在redis
操作中抛出异常时,就自动重连直至连接成功后再返回。
刚开始写的代码,运行后发现redis的连接异常一直捕获不到,因为redis实例化时虽然传入了ip,端口等参数,但是没有真正连接的,所以并没有触发异常。知道了原因,解决也很简单,就是在实例化redis连接后调用一下ping
方法或get
方法(key随意,就算是一个不存在的key也不影响结果),这样当连接有问题时就会抛出异常,这时候再去尝试重连,直至成功再返回实例就可以了,最终代码如下:
from redis import ConnectionPool, Redis
#获取redis服务器连接
def getRedis():
while True:
try:
pool = ConnectionPool(host='192.168.8.211', port=6379, db=0, password=None,
decode_responses=True, health_check_interval=30)
redis = Redis(connection_pool=pool)
redis.ping()
except Exception as e:
print('redis连接失败,正在尝试重连')
continue
else:
return redis