From e5048ebdfa6b484c181693e77102845e05139735 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期五, 04 八月 2023 18:11:12 +0800
Subject: [PATCH] redis连接池归还

---
 code_attribute/gpcode_manager.py |   74 ++++++++++++++++++++++++-------------
 1 files changed, 48 insertions(+), 26 deletions(-)

diff --git a/code_attribute/gpcode_manager.py b/code_attribute/gpcode_manager.py
index b7f1cfa..3e0471d 100644
--- a/code_attribute/gpcode_manager.py
+++ b/code_attribute/gpcode_manager.py
@@ -221,8 +221,8 @@
         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)
+        self.__want_buy_codes_cache.clear()
         if codes:
             self.__want_buy_codes_cache += set(codes)
 
@@ -242,33 +242,53 @@
 # 鏆傚仠涓嬪崟浠g爜绠$悊
 # 涓庨粦鍚嶅崟鐨勫尯鍒槸鏆傚仠浜ゆ槗浠g爜鍙槸涓嶄氦鏄擄紝涓嶈兘绉婚櫎L2鐩戞帶浣�
 class PauseBuyCodesManager:
-    redisManager = redis_manager.RedisManager(0)
-    __redis_key = "pause_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(PauseBuyCodesManager, cls).__new__(cls, *args, **kwargs)
+            # 鍒濆鍖栬缃�
+            # 鑾峰彇浜ゆ槗绐楀彛鐨勯攣
+            cls.__instance.redisManager = redis_manager.RedisManager(0)
+            cls.__instance.__redis_key = "pause_buy_codes"
+            cls.__instance.__pause_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)
+    def __get_redis(self):
+        return self.redisManager.getRedis()
 
-    @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 clear(self):
+        self.__pause_buy_codes_cache.clear()
+        RedisUtils.delete(self.__get_redis(), self.__redis_key)
 
-    @classmethod
-    def remove_code(cls, code):
-        RedisUtils.srem(cls.__get_redis(), cls.__redis_key, code)
+    def sync(self):
+        data = RedisUtils.smembers(self.__get_redis(),
+                                   self.__redis_key)
+        self.__pause_buy_codes_cache.clear()
+        if data:
+            self.__pause_buy_codes_cache |= data
 
-    @classmethod
-    def is_in(cls, code):
-        return RedisUtils.sismember(cls.__get_redis(), cls.__redis_key, code)
+    def add_code(self, code):
+        self.__pause_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.__pause_buy_codes_cache.discard(code)
+        RedisUtils.srem(self.__get_redis(), self.__redis_key, code)
+
+    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.__pause_buy_codes_cache
+
+    def list_code(self):
+        return RedisUtils.smembers(self.__get_redis(), self.__redis_key)
+
+    def list_code_cache(self):
+        return self.__pause_buy_codes_cache
 
 
 def __parse_codes_data(code_datas):
@@ -425,9 +445,11 @@
         for d in gpset:
             RedisUtils.sadd(redis_instance, "gp_limit_up_list", json.dumps(d), auto_free=False)
         RedisUtils.expire(redis_instance, "gp_limit_up_list", tool.get_expire(), auto_free=False)
-        RedisUtils.setex(redis_instance, "gp_limit_up_list_update_time", tool.get_expire(), round(time.time() * 1000), auto_free=False)
+        RedisUtils.setex(redis_instance, "gp_limit_up_list_update_time", tool.get_expire(), round(time.time() * 1000),
+                         auto_free=False)
     finally:
         RedisUtils.realse(redis_instance)
+
 
 # 鑾峰彇娑ㄥ仠鍒楄〃
 def get_limit_up_list():
@@ -450,7 +472,7 @@
 
 
 def get_gp_list():
-    codes = RedisUtils.smembers( __redisManager.getRedis(), "gp_list")
+    codes = RedisUtils.smembers(__redisManager.getRedis(), "gp_list")
     first_codes = get_first_gp_codes()
     return set.union(codes, first_codes)
 
@@ -501,7 +523,7 @@
         codes = get_gp_list()
         if code not in codes and not FirstCodeManager().is_in_first_record_cache(code) and not force:
             return
-        RedisUtils.setex( cls.__redisManager.getRedis(), "price-pre-{}".format(code), tool.get_expire(), str(price))
+        RedisUtils.setex(cls.__redisManager.getRedis(), "price-pre-{}".format(code), tool.get_expire(), str(price))
         cls.__price_pre_cache[code] = float(price)
 
 
@@ -760,7 +782,7 @@
 
 # 璁剧疆姝e湪鎿嶄綔鐨勪唬鐮�
 def set_operate(code):
-    RedisUtils.setex( __redisManager.getRedis(), "gp_operate-{}".format(code), 30, "1")
+    RedisUtils.setex(__redisManager.getRedis(), "gp_operate-{}".format(code), 30, "1")
 
 
 # 鎵归噺璁剧疆姝e湪鎿嶄綔鐨勪唬鐮�

--
Gitblit v1.8.0