From b79f40afbe849f56f7c2db94e9bcebf3f5e92ec6 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期五, 04 八月 2023 09:47:45 +0800
Subject: [PATCH] redis连接池归还
---
code_attribute/limit_up_time_manager.py | 22 +++++++++++++---------
1 files changed, 13 insertions(+), 9 deletions(-)
diff --git a/code_attribute/limit_up_time_manager.py b/code_attribute/limit_up_time_manager.py
index 7e72a56..bfc838e 100644
--- a/code_attribute/limit_up_time_manager.py
+++ b/code_attribute/limit_up_time_manager.py
@@ -3,6 +3,7 @@
娑ㄥ仠鏃堕棿绠＄悊鍣�
"""
from db import redis_manager
+from db.redis_manager import RedisUtils
from utils import global_util, tool
_redisManager = redis_manager.RedisManager(0)
@@ -11,16 +12,15 @@
def save_limit_up_time(code, time):
_time = get_limit_up_time(code)
if _time is None:
- redis = _redisManager.getRedis()
- redis.setex("limit_up_time-{}".format(code), tool.get_expire(), time)
+ RedisUtils.setex(
+ _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 = redis.get("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
@@ -29,11 +29,15 @@
def load_limit_up_time():
redis = _redisManager.getRedis()
- keys = redis.keys("limit_up_time-*")
- for key in keys:
- code = key.replace("limit_up_time-", "")
- global_util.limit_up_time[code] = redis.get(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__":
- list = [("1234578", "09:00:03",None), ("12345", "09:00:01",True), ("123456", "09:00:00",True), ("123457", "09:00:04",False)]
\ No newline at end of file
+ list = [("1234578", "09:00:03", None), ("12345", "09:00:01", True), ("123456", "09:00:00", True),
+ ("123457", "09:00:04", False)]
--
Gitblit v1.8.0