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