From a3c05bddc4c61d63b804ee541cf6dd01717ba94d Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期三, 02 八月 2023 16:39:08 +0800
Subject: [PATCH] redis封装

---
 trade/trade_queue_manager.py |   41 +++++++++++++++++++++--------------------
 1 files changed, 21 insertions(+), 20 deletions(-)

diff --git a/trade/trade_queue_manager.py b/trade/trade_queue_manager.py
index e5eb59f..65f1527 100644
--- a/trade/trade_queue_manager.py
+++ b/trade/trade_queue_manager.py
@@ -4,6 +4,7 @@
 
 from code_attribute import gpcode_manager
 from db import redis_manager
+from db.redis_manager import RedisUtils
 from utils import tool
 from trade import trade_manager
 
@@ -19,37 +20,37 @@
     # 淇濆瓨鏈�澶ч噺
     def __save_max_buy1_volume(self, code, volume):
         key = "max_buy1_volumn-{}".format(code)
-        self.__get_redis().setex(key, tool.get_expire(), volume)
+        RedisUtils.setex(self.__get_redis(), key, tool.get_expire(), volume)
 
     def __get_max_buy1_volume(self, code):
         key = "max_buy1_volumn-{}".format(code)
-        val = self.__get_redis().get(key)
+        val = RedisUtils.get(self.__get_redis(), key)
         if val is not None:
             return int(val)
         return None
 
     def __del_max_buy1_volume(self, code):
         key = "max_buy1_volumn-{}".format(code)
-        val = self.__get_redis().delete(key)
+        val = RedisUtils.delete(self.__get_redis(), key)
 
     def __save_recod(self, code, time_str, volumn):
 
         # 淇濆瓨姣忎竴娆$殑
         key = "buy1_volumn-{}-{}".format(code, time_str)
-        self.__get_redis().setex(key, tool.get_expire(), volumn)
+        RedisUtils.setex(self.__get_redis(), key, tool.get_expire(), volumn)
         # 淇濆瓨鏈�杩戠殑
         key = "buy1_volumn_latest_info-{}".format(code)
-        self.__get_redis().setex(key, tool.get_expire(), json.dumps((time_str, volumn)))
+        RedisUtils.setex(self.__get_redis(), key, tool.get_expire(), json.dumps((time_str, volumn)))
 
     # 淇濆瓨涓婁竴娆℃暟鎹�
     def __save_last_recod(self, code, time_str, volumn):
         # 淇濆瓨鏈�杩戠殑
         key = "buy1_volumn_last_info-{}".format(code)
-        self.__get_redis().setex(key, tool.get_expire(), json.dumps((time_str, volumn)))
+        RedisUtils.setex(self.__get_redis(), key, tool.get_expire(), json.dumps((time_str, volumn)))
 
     def __get_last_record(self, code):
         key = "buy1_volumn_last_info-{}".format(code)
-        val = self.__get_redis().get(key)
+        val = RedisUtils.get(self.__get_redis(), key)
         if val is None:
             return None, None
         val = json.loads(val)
@@ -57,7 +58,7 @@
 
     def __get_latest_record(self, code):
         key = "buy1_volumn_latest_info-{}".format(code)
-        val = self.__get_redis().get(key)
+        val = RedisUtils.get(self.__get_redis(), key)
         if val is None:
             return None, None
         val = json.loads(val)
@@ -66,17 +67,17 @@
     # 娣诲姞璁板綍
     def __add_recod(self, code):
         key = "buy1_volumn_codes"
-        self.__get_redis().sadd(key, code)
-        self.__get_redis().expire(key, 10)
+        RedisUtils.sadd( self.__get_redis(), key, code)
+        RedisUtils.expire(self.__get_redis(), key, 10)
 
     # 鑾峰彇褰撳墠姝e湪鐩戝惉鐨勪唬鐮�
     def get_current_codes(self):
         key = "buy1_volumn_codes"
-        return self.__get_redis().smembers(key)
+        return RedisUtils.smembers(self.__get_redis(), key)
 
     def get_buy_1_volumn(self, code, time_str):
         key = "buy1_volumn-{}-{}".format(code, time_str)
-        return self.__get_redis().get(key)
+        return RedisUtils.get(self.__get_redis(), key)
 
     # 杩斿洖鏄惁闇�瑕佹洿鏂版暟鎹�,鏄惁闇�瑕佹挙鍗曪紝鎾ゅ崟鍘熷洜
     def save(self, code, time_str, volumn, price):
@@ -174,13 +175,13 @@
     def __save_recod(self, code, time_str, volumn):
         # 淇濆瓨姣忎竴娆$殑
         key = "buy1_volumn_juejin-{}-{}".format(code, time_str)
-        self.__get_redis().setex(key, tool.get_expire(), volumn)
+        RedisUtils.setex(self.__get_redis(), key, tool.get_expire(), volumn)
         key = "buy1_volumn_juejin_latest_info-{}".format(code)
-        self.__get_redis().setex(key, tool.get_expire(), volumn)
+        RedisUtils.setex(self.__get_redis(), key, tool.get_expire(), volumn)
 
     def __get_latest_record(self, code):
         key = "buy1_volumn_juejin_latest_info-{}".format(code)
-        val = self.__get_redis().get(key)
+        val = RedisUtils.get(self.__get_redis(), key)
         if val is None:
             return None, None
         val = json.loads(val)
@@ -220,11 +221,11 @@
     def __save_latest_recod(self, code, info):
         # 淇濆瓨姣忎竴娆$殑
         key = "ths_l2_latest_trade_info-{}".format(code)
-        self.__get_redis().setex(key, tool.get_expire(), json.dumps(info))
+        RedisUtils.setex(self.__get_redis(), key, tool.get_expire(), json.dumps(info))
 
     def __get_latest_record(self, code):
         key = "ths_l2_latest_trade_info-{}".format(code)
-        val = self.__get_redis().get(key)
+        val = RedisUtils.get(self.__get_redis(), key)
         if val is None:
             return None
         return json.loads(val)
@@ -233,14 +234,14 @@
 
     def __add_buy1_code(self, code):
         key = "buy1_volumn_codes"
-        self.__get_redis().sadd(key, code)
-        self.__get_redis().expire(key, 10)
+        RedisUtils.sadd( self.__get_redis(), key, code)
+        RedisUtils.expire(self.__get_redis(), key, 10)
 
         # 鑾峰彇褰撳墠姝e湪鐩戝惉鐨勪唬鐮�
 
     def get_current_codes(self):
         key = "buy1_volumn_codes"
-        return self.__get_redis().smembers(key)
+        return RedisUtils.smembers(self.__get_redis(), key)
 
     def save_recod(self, code, data):
         _str = json.dumps(data)

--
Gitblit v1.8.0