From 4d321ae069fce2c0f6f7884bec8fa6c4ec534d1a Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期四, 03 八月 2023 18:25:33 +0800
Subject: [PATCH] redis连接池归还
---
code_attribute/limit_up_time_manager.py | 17 +++++++++--------
1 files changed, 9 insertions(+), 8 deletions(-)
diff --git a/code_attribute/limit_up_time_manager.py b/code_attribute/limit_up_time_manager.py
index 5401abe..bfc838e 100644
--- a/code_attribute/limit_up_time_manager.py
+++ b/code_attribute/limit_up_time_manager.py
@@ -12,17 +12,15 @@
def save_limit_up_time(code, time):
_time = get_limit_up_time(code)
if _time is None:
- redis = _redisManager.getRedis()
RedisUtils.setex(
- redis, "limit_up_time-{}".format(code), tool.get_expire(), time)
+ _redisManager.getRedis(), "limit_up_time-{}".format(code), tool.get_expire(), time)
global_util.limit_up_time[code] = time
def get_limit_up_time(code):
time = global_util.limit_up_time.get(code)
if time is None:
- redis = _redisManager.getRedis()
- time = RedisUtils.get(redis, "limit_up_time-{}".format(code))
+ time = RedisUtils.get(_redisManager.getRedis(), "limit_up_time-{}".format(code))
if time is not None:
global_util.limit_up_time[code] = time
@@ -31,10 +29,13 @@
def load_limit_up_time():
redis = _redisManager.getRedis()
- keys = RedisUtils.keys(redis, "limit_up_time-*")
- for key in keys:
- code = key.replace("limit_up_time-", "")
- global_util.limit_up_time[code] = RedisUtils.get(redis, key)
+ try:
+ keys = RedisUtils.keys(redis, "limit_up_time-*", auto_free=False)
+ for key in keys:
+ code = key.replace("limit_up_time-", "")
+ global_util.limit_up_time[code] = RedisUtils.get(redis, key, auto_free=False)
+ finally:
+ redis.connection_pool.disconnect()
if __name__ == "__main__":
--
Gitblit v1.8.0