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