From 65afea1ba534b51f947cbe7989d7f4d650bbc9e6 Mon Sep 17 00:00:00 2001 From: Administrator <admin@example.com> Date: 星期一, 04 十一月 2024 18:29:00 +0800 Subject: [PATCH] 代码异常保护 --- code_attribute/gpcode_manager.py | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 54 insertions(+), 0 deletions(-) diff --git a/code_attribute/gpcode_manager.py b/code_attribute/gpcode_manager.py index 64bd83d..48ab178 100644 --- a/code_attribute/gpcode_manager.py +++ b/code_attribute/gpcode_manager.py @@ -1,6 +1,7 @@ """ 鑲$エ浠g爜绠$悊鍣� """ +import copy import json import time @@ -17,6 +18,47 @@ __redisManager = redis_manager.RedisManager(0) __db = 0 + + +class BuyOpenLimitUpCodeManager: + """ + 鎺�1浠g爜绠$悊 + """ + __db = 2 + __redisManager = redis_manager.RedisManager(2) + __instance = None + __codes_cache = set() + + def __new__(cls, *args, **kwargs): + if not cls.__instance: + cls.__instance = super(BuyOpenLimitUpCodeManager, cls).__new__(cls, *args, **kwargs) + cls.__load_data() + return cls.__instance + + @classmethod + def __get_redis(cls): + return cls.__redisManager.getRedis() + + @classmethod + def __load_data(cls): + val = RedisUtils.get(cls.__get_redis(), "buy_open_limit_up_codes") + if val: + val = json.loads(val) + cls.__codes_cache = set(val) + + def set_codes(self, codes): + self.__codes_cache = copy.deepcopy(codes) + RedisUtils.set_async(self.__db, "buy_open_limit_up_codes", json.dumps(list(codes))) + + def get_codes(self): + return self.__codes_cache + + def is_in_cache(self, code): + if not self.__codes_cache: + return False + if code in self.__codes_cache: + return True + return False class CodesNameManager: @@ -63,6 +105,18 @@ cls.__mysqldb.execute(f"insert into code_name(code,code_name,update_time ) values('{code}','{name}',now())") @classmethod + def add_code_names(cls, code_name_dict): + """ + 鎵归噺娣诲姞浠g爜鍚嶇О + @param code_name_dict: + @return: + """ + for code in code_name_dict: + if code in cls.__code_name_dict and cls.__code_name_dict[code] == code_name_dict[code]: + continue + cls.add_code_name(code, code_name_dict[code]) + + @classmethod def add_first_code_name(cls, code, name): cls.add_code_name(code, name) -- Gitblit v1.8.0