From 173a8ace6fd8142a8e2ff62c3b496940c5f603cd Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期三, 30 十月 2024 18:23:26 +0800
Subject: [PATCH] 量的代码结构修改/扫入策略修改

---
 code_attribute/gpcode_manager.py |   30 +++++++++++++++++++++++++++---
 1 files changed, 27 insertions(+), 3 deletions(-)

diff --git a/code_attribute/gpcode_manager.py b/code_attribute/gpcode_manager.py
index 98f25de..38fe868 100644
--- a/code_attribute/gpcode_manager.py
+++ b/code_attribute/gpcode_manager.py
@@ -22,6 +22,17 @@
 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 +61,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):
+        """
+        鎵归噺娣诲姞浠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):
@@ -98,6 +121,7 @@
 # 鎯宠涔扮殑浠g爜
 class WantBuyCodesManager:
     __instance = None
+    __db = 0
     redisManager = redis_manager.RedisManager(0)
     __redis_key = "want_buy_codes"
 
@@ -126,7 +150,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 +421,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 +556,7 @@
 
 # 鑾峰彇浠g爜鐨勫悕绉�
 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