From 8595dc22aa9dde6aba6d0f8cdcf1656a8a59513b Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期三, 02 八月 2023 18:05:10 +0800
Subject: [PATCH] redis封装

---
 code_attribute/gpcode_first_screen_manager.py |   17 +++++++++--------
 1 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/code_attribute/gpcode_first_screen_manager.py b/code_attribute/gpcode_first_screen_manager.py
index cc078c7..5bc3366 100644
--- a/code_attribute/gpcode_first_screen_manager.py
+++ b/code_attribute/gpcode_first_screen_manager.py
@@ -5,9 +5,10 @@
 # 璁剧疆棣栨澘鏈瓫閫夌殑鐩爣绁�
 import json
 
+from db.redis_manager import RedisUtils
 from utils import tool
 from db import redis_manager
-from third_data import  block_info
+from third_data import block_info
 
 __redisManager = redis_manager.RedisManager(0)
 
@@ -23,11 +24,11 @@
 # 4.鏈�杩戠偢寮�鏃堕棿
 # 5.鏄惁宸茬粡娑ㄥ仠
 def __save_first_code_data(code, data):
-    __redisManager.getRedis().setex(f"first_code_data-{code}", tool.get_expire(), json.dumps(data))
+    RedisUtils.setex(__redisManager.getRedis(), f"first_code_data-{code}", tool.get_expire(), json.dumps(data))
 
 
 def __get_first_code_data(code):
-    val = __get_redis().get(f"first_code_data-{code}")
+    val = RedisUtils.get(__get_redis(), f"first_code_data-{code}")
     if val is None:
         return None
     return json.loads(val)
@@ -38,24 +39,24 @@
     redis = __redisManager.getRedis()
     if codes:
         for code in codes:
-            redis.sadd("first_no_screen_codes", code)
-        redis.expire("first_no_screen_codes", tool.get_expire())
+            RedisUtils.sadd(redis, "first_no_screen_codes", code)
+        RedisUtils.expire(redis, "first_no_screen_codes", tool.get_expire())
 
 
 def clear_first_no_screen_codes():
     redis = __redisManager.getRedis()
-    redis.delete("first_no_screen_codes")
+    RedisUtils.delete(redis, "first_no_screen_codes")
 
 
 def __remove_first_no_screen_codes(codes):
     redis = __redisManager.getRedis()
     if codes:
         for code in codes:
-            redis.srem("first_no_screen_codes", code)
+            RedisUtils.srem(redis, "first_no_screen_codes", code)
 
 
 def __get_first_no_screen_codes():
-    codes = __get_redis().smembers("first_no_screen_codes")
+    codes = RedisUtils.smembers(__get_redis(), "first_no_screen_codes")
     if not codes:
         return set()
     return codes

--
Gitblit v1.8.0