From 90886c62a62ecbd64c9b229b48e4ef2cf600ae43 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期四, 03 八月 2023 16:34:21 +0800
Subject: [PATCH] redis上级缓存添加

---
 code_attribute/gpcode_manager.py |  279 ++++++++++++++++++++++++++++++++++---------------------
 1 files changed, 174 insertions(+), 105 deletions(-)

diff --git a/code_attribute/gpcode_manager.py b/code_attribute/gpcode_manager.py
index 766915d..3c53c02 100644
--- a/code_attribute/gpcode_manager.py
+++ b/code_attribute/gpcode_manager.py
@@ -17,6 +17,7 @@
 
 class CodesNameManager:
     redisManager = redis_manager.RedisManager(0)
+    __gp_list_names_first_cache = []
 
     @classmethod
     def __get_redis(cls):
@@ -44,8 +45,8 @@
         return None
 
     @classmethod
-    def list_second_code_name_dict(cls):
-        val = RedisUtils.get(cls.__get_redis(), "gp_list_names")
+    def list_first_code_name_dict_cache(cls):
+        val = RedisUtils.get(cls.__get_redis(), "gp_list_names_first")
         if val is not None:
             val = json.loads(val)
             return val
@@ -61,16 +62,6 @@
             if val[k] == code:
                 return k
         return None
-
-    @classmethod
-    def get_second_code_name(cls, code):
-        val = RedisUtils.get(cls.__get_redis(), "gp_list_names")
-        if not val:
-            return None
-        val = json.loads(val)
-        for k in val:
-            if val[k] == code:
-                return k
 
     @classmethod
     def get_first_name_code(cls, name):
@@ -89,6 +80,34 @@
         val[name] = code
         cls.set_first_code_names(val)
 
+    # 璁剧疆棣栨澘浠g爜鍚嶇О
+    @classmethod
+    def set_first_code_names(cls, datas):
+        RedisUtils.set(cls.__get_redis(), "gp_list_names_first", json.dumps(datas))
+
+    # 鍒犻櫎棣栨澘浠g爜鍚嶇О
+    @classmethod
+    def clear_first_code_names(cls):
+        RedisUtils.delete(cls.__get_redis(), "gp_list_names_first")
+
+    @classmethod
+    def list_second_code_name_dict(cls):
+        val = RedisUtils.get(cls.__get_redis(), "gp_list_names")
+        if val is not None:
+            val = json.loads(val)
+            return val
+        return None
+
+    @classmethod
+    def get_second_code_name(cls, code):
+        val = RedisUtils.get(cls.__get_redis(), "gp_list_names")
+        if not val:
+            return None
+        val = json.loads(val)
+        for k in val:
+            if val[k] == code:
+                return k
+
     @classmethod
     def get_second_name_code(cls, name):
         val = RedisUtils.get(cls.__get_redis(), "gp_list_names")
@@ -97,20 +116,10 @@
         val = json.loads(val)
         return val.get(name)
 
-    # 璁剧疆棣栨澘浠g爜鍚嶇О
-    @classmethod
-    def set_first_code_names(cls, datas):
-        RedisUtils.set( cls.__get_redis(), "gp_list_names_first", json.dumps(datas))
-
     # 璁剧疆浜屾澘浠g爜鍚嶇О
     @classmethod
     def set_second_code_names(cls, datas):
         RedisUtils.set(cls.__get_redis(), "gp_list_names", json.dumps(datas))
-
-    # 鍒犻櫎棣栨澘浠g爜鍚嶇О
-    @classmethod
-    def clear_first_code_names(cls):
-        RedisUtils.delete(cls.__get_redis(), "gp_list_names_first")
 
     # 璁剧疆浜屾澘浠g爜鍚嶇О
     @classmethod
@@ -120,71 +129,114 @@
 
 # 棣栨澘浠g爜绠$悊
 class FirstCodeManager:
-    redisManager = redis_manager.RedisManager(0)
+    __instance = None
 
-    @classmethod
-    def __get_redis(cls):
-        return cls.redisManager.getRedis()
+    def __new__(cls, *args, **kwargs):
+        if not cls.__instance:
+            cls.__instance = super(FirstCodeManager, cls).__new__(cls, *args, **kwargs)
+            # 鍒濆鍖栬缃�
+            # 鑾峰彇浜ゆ槗绐楀彛鐨勯攣
+            cls.__instance.redisManager = redis_manager.RedisManager(0)
+            cls.__instance.__first_code_record_cache = RedisUtils.smembers(cls.__instance.__get_redis(),
+                                                                           "first_code_record")
+            cls.__instance.__first_code_limited_up_record_cache = RedisUtils.smembers(cls.__instance.__get_redis(),
+                                                                                      "first_code_limited_up_record")
+        return cls.__instance
+
+    def __get_redis(self):
+        return self.redisManager.getRedis()
 
     # 鍔犲叆棣栨澘鍘嗗彶璁板綍
-    @classmethod
-    def add_record(cls, codes):
+    def add_record(self, codes):
+        hasChanged = False
         for code in codes:
-            RedisUtils.sadd(cls.__get_redis(), "first_code_record", code)
-        RedisUtils.expire(cls.__get_redis(), "first_code_record", tool.get_expire())
+            if code not in self.__first_code_record_cache:
+                RedisUtils.sadd(self.__get_redis(), "first_code_record", code)
+                hasChanged = True
+            self.__first_code_record_cache.add(code)
+        if hasChanged:
+            RedisUtils.expire(self.__get_redis(), "first_code_record", tool.get_expire())
 
-    @classmethod
-    def is_in_first_record(cls, code):
-        if RedisUtils.sismember( cls.__get_redis(), "first_code_record", code):
+    def is_in_first_record(self, code):
+        if RedisUtils.sismember(self.__get_redis(), "first_code_record", code):
             return True
         else:
             return False
+
+    def is_in_first_record_cache(self, code):
+        return code in self.__first_code_record_cache
 
     # 鍔犲叆棣栨澘娑ㄥ仠杩囦唬鐮侀泦鍚�
-    @classmethod
-    def add_limited_up_record(cls, codes):
+    def add_limited_up_record(self, codes):
+        hasChanged = False
         for code in codes:
-            RedisUtils.sadd(cls.__get_redis(), "first_code_limited_up_record", code)
-        RedisUtils.expire(cls.__get_redis(), "first_code_limited_up_record", tool.get_expire())
+            if code not in self.__first_code_limited_up_record_cache:
+                RedisUtils.sadd(self.__get_redis(), "first_code_limited_up_record", code)
+                hasChanged = True
+            self.__first_code_limited_up_record_cache.add(code)
+        if hasChanged:
+            RedisUtils.expire(self.__get_redis(), "first_code_limited_up_record", tool.get_expire())
 
     # 鏄惁娑ㄥ仠杩�
-    @classmethod
-    def is_limited_up(cls, code):
-        if RedisUtils.sismember( cls.__get_redis(), "first_code_limited_up_record", code):
+    def is_limited_up(self, code):
+        if RedisUtils.sismember(self.__get_redis(), "first_code_limited_up_record", code):
             return True
         else:
             return False
+
+    def is_limited_up_cache(self, code):
+        return code in self.__first_code_limited_up_record_cache
 
 
 # 鎯宠涔扮殑浠g爜
 class WantBuyCodesManager:
-    redisManager = redis_manager.RedisManager(0)
-    __redis_key = "want_buy_codes"
+    __instance = None
 
-    @classmethod
-    def __get_redis(cls):
-        return cls.redisManager.getRedis()
+    def __new__(cls, *args, **kwargs):
+        if not cls.__instance:
+            cls.__instance = super(WantBuyCodesManager, cls).__new__(cls, *args, **kwargs)
+            # 鍒濆鍖栬缃�
+            # 鑾峰彇浜ゆ槗绐楀彛鐨勯攣
+            cls.__instance.redisManager = redis_manager.RedisManager(0)
+            cls.__instance.__redis_key = "want_buy_codes"
+            cls.__instance.__want_buy_codes_cache = RedisUtils.smembers(cls.__instance.__get_redis(),
+                                                                        cls.__instance.__redis_key)
+        return cls.__instance
 
-    @classmethod
-    def clear(cls):
-        RedisUtils.delete(cls.__get_redis(), cls.__redis_key)
+    __want_buy_codes_cache = set()
 
-    @classmethod
-    def add_code(cls, code):
-        RedisUtils.sadd(cls.__get_redis(), cls.__redis_key, code)
-        RedisUtils.expire(cls.__get_redis(), cls.__redis_key, tool.get_expire())
+    def __get_redis(self):
+        return self.redisManager.getRedis()
 
-    @classmethod
-    def remove_code(cls, code):
-        RedisUtils.srem( cls.__get_redis(),cls.__redis_key, code)
+    def clear(self):
+        RedisUtils.delete(self.__get_redis(), self.__redis_key)
 
-    @classmethod
-    def is_in(cls, code):
-        return RedisUtils.sismember( cls.__get_redis(), cls.__redis_key, code)
+    def add_code(self, code):
+        self.__want_buy_codes_cache.add(code)
+        RedisUtils.sadd(self.__get_redis(), self.__redis_key, code)
+        RedisUtils.expire(self.__get_redis(), self.__redis_key, tool.get_expire())
 
-    @classmethod
-    def list_code(cls):
-        return RedisUtils.smembers(cls.__get_redis(), cls.__redis_key)
+    def remove_code(self, code):
+        self.__want_buy_codes_cache.discard(code)
+        RedisUtils.srem(self.__get_redis(), self.__redis_key, code)
+
+    def sync(self):
+        self.__want_buy_codes_cache.clear()
+        codes = RedisUtils.smembers(self.__get_redis(), self.__redis_key)
+        if codes:
+            self.__want_buy_codes_cache += set(codes)
+
+    def is_in(self, code):
+        return RedisUtils.sismember(self.__get_redis(), self.__redis_key, code)
+
+    def is_in_cache(self, code):
+        return code in self.__want_buy_codes_cache
+
+    def list_code(self):
+        return RedisUtils.smembers(self.__get_redis(), self.__redis_key)
+
+    def list_code_cache(self):
+        return self.__want_buy_codes_cache
 
 
 # 鏆傚仠涓嬪崟浠g爜绠$悊
@@ -212,7 +264,7 @@
 
     @classmethod
     def is_in(cls, code):
-        return RedisUtils.sismember( cls.__get_redis(), cls.__redis_key, code)
+        return RedisUtils.sismember(cls.__get_redis(), cls.__redis_key, code)
 
     @classmethod
     def list_code(cls):
@@ -264,6 +316,9 @@
     CodesNameManager.set_second_code_names(old_name_codes)
 
 
+__gp_list_first_codes_cache = set()
+
+
 # -------------------------------棣栨澘浠g爜绠$悊-------------------------------
 # 娣诲姞棣栨澘浠g爜
 # code_datas 鎺橀噾杩斿洖鐨勬暟鎹�
@@ -273,7 +328,7 @@
     codes_set = set()
     for code in codes:
         codes_set.add(code)
-    old_codes_set = RedisUtils.smembers(redis_instance, "gp_list_first")
+    old_codes_set = __gp_list_first_codes_cache
     if old_codes_set is None:
         old_codes_set = set()
     del_set = old_codes_set - codes_set
@@ -281,8 +336,13 @@
     for code in add_codes:
         RedisUtils.sadd(redis_instance, "gp_list_first", code)
     for code in del_set:
-        redis_instance.srem("gp_list_first", code)
-    RedisUtils.expire(redis_instance, "gp_list_first", tool.get_expire())
+        RedisUtils.srem(redis_instance, "gp_list_first", code)
+    if add_codes or del_set:
+        RedisUtils.expire(redis_instance, "gp_list_first", tool.get_expire())
+    # 鏇存柊缂撳瓨
+    __gp_list_first_codes_cache.clear()
+    for code in codes_set:
+        __gp_list_first_codes_cache.add(code)
 
     old_name_codes = CodesNameManager.list_first_code_name_dict()
     if old_name_codes is None:
@@ -297,7 +357,7 @@
 def remove_first_gp_code(codes):
     redis_instance = __redisManager.getRedis()
     for code in codes:
-        redis_instance.srem("gp_list_first", code)
+        RedisUtils.srem(redis_instance, "gp_list_first", code)
 
 
 # 鑾峰彇棣栨澘浠g爜
@@ -309,7 +369,7 @@
 # 鏄惁鍦ㄩ鏉块噷闈�
 def is_in_first_gp_codes(code):
     redis_instance = __redisManager.getRedis()
-    return redis_instance.sismember("gp_list_first", code)
+    return RedisUtils.sismember(redis_instance, "gp_list_first", code)
 
 
 # 鑾峰彇鍚嶇О瀵瑰簲鐨勪唬鐮�
@@ -366,13 +426,13 @@
 
 def rm_gp(code):
     redis_instance = __redisManager.getRedis()
-    redis_instance.srem("gp_list", code)
+    RedisUtils.srem(redis_instance, "gp_list", code)
     remove_first_gp_code([code])
 
 
 def is_in_gp_pool(code):
     redis_instance = __redisManager.getRedis()
-    return redis_instance.sismember("gp_list", code) or is_in_first_gp_codes(code)
+    return RedisUtils.sismember(redis_instance, "gp_list", code) or is_in_first_gp_codes(code)
 
 
 def get_gp_list():
@@ -401,35 +461,38 @@
     return list
 
 
-# 鑾峰彇鏀剁洏浠�
-def get_price_pre(code):
-    redis_instance = __redisManager.getRedis()
-    result = RedisUtils.get(redis_instance, "price-pre-{}".format(code))
-    if result is not None:
-        return float(result)
-    return None
+class CodePrePriceManager:
+    __price_pre_cache = {}
+    __redisManager = redis_manager.RedisManager(0)
 
+    # 鑾峰彇鏀剁洏浠�
+    @classmethod
+    def get_price_pre(cls, code):
+        redis_instance = cls.__redisManager.getRedis()
+        result = RedisUtils.get(redis_instance, "price-pre-{}".format(code))
+        if result is not None:
+            return float(result)
+        return None
 
-__price_pre_cache = {}
+    # 鑾峰彇缂撳瓨
+    @classmethod
+    def get_price_pre_cache(cls, code):
+        if code in cls.__price_pre_cache:
+            return float(cls.__price_pre_cache[code])
+        val = cls.get_price_pre(code)
+        if val:
+            cls.__price_pre_cache[code] = val
+        return val
 
-
-# 鑾峰彇缂撳瓨
-def get_price_pre_cache(code):
-    if code in __price_pre_cache:
-        return __price_pre_cache[code]
-    val = get_price_pre(code)
-    if val:
-        __price_pre_cache[code] = val
-    return val
-
-
-# 璁剧疆鏀剁洏浠�
-def set_price_pre(code, price, force=False):
-    codes = get_gp_list()
-    if code not in codes and not FirstCodeManager.is_in_first_record(code) and not force:
-        return
-    redis_instance = __redisManager.getRedis()
-    RedisUtils.setex(redis_instance, "price-pre-{}".format(code), tool.get_expire(), str(price))
+    # 璁剧疆鏀剁洏浠�
+    @classmethod
+    def set_price_pre(cls, code, price, force=False):
+        codes = get_gp_list()
+        if code not in codes and not FirstCodeManager().is_in_first_record_cache(code) and not force:
+            return
+        redis_instance = cls.__redisManager.getRedis()
+        RedisUtils.setex(redis_instance, "price-pre-{}".format(code), tool.get_expire(), str(price))
+        cls.__price_pre_cache[code] = float(price)
 
 
 __limit_up_price_dict = {}
@@ -440,7 +503,7 @@
     # 璇诲彇鍐呭瓨涓殑鍊�
     if code in __limit_up_price_dict:
         return __limit_up_price_dict[code]
-    price = get_price_pre(code)
+    price = CodePrePriceManager.get_price_pre_cache(code)
     if price is None:
         return None
     limit_up_price = tool.to_price(decimal.Decimal(str(price)) * decimal.Decimal("1.1"))
@@ -456,7 +519,7 @@
 
 # 鑾峰彇璺屽仠浠�
 def get_limit_down_price(code):
-    price = get_price_pre(code)
+    price = CodePrePriceManager.get_price_pre_cache(code)
     if price is None:
         return None
     return tool.to_price(decimal.Decimal(str(price)) * decimal.Decimal("0.9"))
@@ -471,8 +534,14 @@
     return None
 
 
+__current_price_cache = {}
+
+
 # 璁剧疆鐜颁环
 def set_price(code, price):
+    if code in __current_price_cache and __current_price_cache[code] == price:
+        return
+    __current_price_cache[code] = price
     redis_instance = __redisManager.getRedis()
     RedisUtils.setex(redis_instance, "price-{}".format(code), tool.get_expire(), price)
 
@@ -518,13 +587,13 @@
     keys = RedisUtils.keys(redis_instance, "listen_code-*-*")
 
     for key in keys:
-        result = redis_instance.get(key)
+        result = RedisUtils.get(redis_instance, key)
         if result:
             # 绉婚櫎闇�瑕佹坊鍔犵殑浠g爜
             codes_set.discard(result)
             client_id_, pos_, code_ = int(key.split("-")[1]), int(key.split("-")[2]), result
             key_ = "code_listen_pos-{}".format(code_)
-            val = redis_instance.get(key_)
+            val = RedisUtils.get(redis_instance, key_)
             if val is None:
                 RedisUtils.setex(redis_instance, key_, tool.get_expire(), json.dumps((client_id_, pos_)))
             else:
@@ -541,7 +610,7 @@
 def init_listen_code_by_pos(client_id, pos):
     redis_instance = __redisManager.getRedis()
     key = "listen_code-{}-{}".format(client_id, pos)
-    redis_instance.setnx(key, "")
+    RedisUtils.setnx(redis_instance, key, "")
     RedisUtils.expire(redis_instance, key, tool.get_expire())
 
 
@@ -580,7 +649,7 @@
             index = key.split("-")[-1]
             if int(index) + 1 > constant.L2_CODE_COUNT_PER_DEVICE:
                 continue
-            result = redis_instance.get(key)
+            result = RedisUtils.get(redis_instance, key)
             if result is None or len(result) == 0:
                 available_positions.append((client_id, int(key.replace("listen_code-{}-".format(client_id), ""))))
             else:
@@ -617,7 +686,7 @@
         k = "listen_code-{}-*".format(client_id)
         keys = RedisUtils.keys(redis_instance, k)
         for key in keys:
-            code = redis_instance.get(key)
+            code = RedisUtils.get(redis_instance, key)
             if not code:
                 free_count += 1
     return free_count
@@ -626,7 +695,7 @@
 # 鑾峰彇姝e湪鐩戝惉鐨勪唬鐮佺殑浣嶇疆
 def get_listen_code_pos(code):
     redis_instance = __redisManager.getRedis()
-    val = redis_instance.get("code_listen_pos-{}".format(code))
+    val = RedisUtils.get(redis_instance, "code_listen_pos-{}".format(code))
     if val is None:
         return None, None
     val = json.loads(val)
@@ -642,7 +711,7 @@
 # 鏄惁姝e湪鐩戝惉
 def is_listen(code):
     redis_instance = __redisManager.getRedis()
-    val = redis_instance.get("code_listen_pos-{}".format(code))
+    val = RedisUtils.get(redis_instance, "code_listen_pos-{}".format(code))
     if val is None:
         return False
     else:
@@ -666,7 +735,7 @@
 # 鏄惁姝e湪鎿嶄綔
 def is_operate(code):
     redis_instance = __redisManager.getRedis()
-    return redis_instance.get("gp_operate-{}".format(code)) is not None
+    return RedisUtils.get(redis_instance, "gp_operate-{}".format(code)) is not None
 
 
 # 璁剧疆姝e湪鎿嶄綔鐨勪唬鐮�

--
Gitblit v1.8.0