redis连接池归还

Administrator

2023-08-03 4d321ae069fce2c0f6f7884bec8fa6c4ec534d1a

 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__":