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 | 72 ++++++++++++++++++++++++++++++++++-
1 files changed, 69 insertions(+), 3 deletions(-)
diff --git a/code_attribute/gpcode_manager.py b/code_attribute/gpcode_manager.py
index 98f25de..48ab178 100644
--- a/code_attribute/gpcode_manager.py
+++ b/code_attribute/gpcode_manager.py
@@ -1,6 +1,7 @@
"""
鑲＄エ浠ｇ爜绠＄悊鍣�
"""
+import copy
import json
import time
@@ -19,9 +20,61 @@
__db = 0
+class BuyOpenLimitUpCodeManager:
+ """
+ 鎺�1浠ｇ爜绠＄悊
+ """
+ __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:
__mysqldb = Mysqldb()
__code_name_dict = {}
+ __instance = None
+
+ def __new__(cls, *args, **kwargs):
+ if not cls.__instance:
+ cls.__instance = super(CodesNameManager, cls).__new__(cls, *args, **kwargs)
+ cls.__load_data()
+ return cls.__instance
+
+ @classmethod
+ def __load_data(cls):
+ cls.__code_name_dict = cls.list_code_name_dict()
@classmethod
def list_code_name_dict(cls):
@@ -50,6 +103,18 @@
cls.__mysqldb.execute(f"update code_name set code_name ='{name}', update_time = now() where code= '{code}'")
else:
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):
+ """
+ 鎵归噺娣诲姞浠ｇ爜鍚嶇О
+ @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):
@@ -98,6 +163,7 @@
# 鎯宠涔扮殑浠ｇ爜
class WantBuyCodesManager:
__instance = None
+ __db = 0
redisManager = redis_manager.RedisManager(0)
__redis_key = "want_buy_codes"
@@ -126,7 +192,7 @@
def remove_code(self, code):
self.__want_buy_codes_cache.discard(code)
- RedisUtils.srem(self.__get_redis(), self.__redis_key, code)
+ RedisUtils.srem_async(self.__db, self.__redis_key, code)
def sync(self):
codes = self.list_code()
@@ -397,7 +463,7 @@
def remove_code(self, code):
self.__codes_set.discard(code)
- RedisUtils.srem(self.__get_redis(), "green-trade-codes", code)
+ RedisUtils.srem_async(self.__db, "green-trade-codes", code)
def is_in(self, code):
return RedisUtils.sismember(self.__get_redis(), "green-trade-codes", code)
@@ -532,7 +598,7 @@
# 鑾峰彇浠ｇ爜鐨勫悕绉�
def get_code_name(code):
- return CodesNameManager.get_code_name(code)
+ return CodesNameManager().get_code_name(code)
def get_name_codes():
--
Gitblit v1.8.0