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

---
 l2/l2_data_manager.py |   31 ++++++++++++++++---------------
 1 files changed, 16 insertions(+), 15 deletions(-)

diff --git a/l2/l2_data_manager.py b/l2/l2_data_manager.py
index bb5473c..d9242a1 100644
--- a/l2/l2_data_manager.py
+++ b/l2/l2_data_manager.py
@@ -51,7 +51,7 @@
     def get_buy_compute_start_data(code):
         redis = TradePointManager.__get_redis()
         _key = "buy_compute_index_info-{}".format(code)
-        _data_json = redis.get(_key)
+        _data_json = RedisUtils.get(redis, _key)
         if _data_json is None:
             return None, None, None, 0, 0, [], 0
         _data = json.loads(_data_json)
@@ -89,15 +89,15 @@
                      volume_rate)
         CodeDataCacheUtil.set_cache(TradePointManager.__buy_compute_index_info_cache, code, data_)
         RedisUtils.setex(
-            redis,_key, expire,
-                    json.dumps(data_))
+            redis, _key, expire,
+            json.dumps(data_))
 
     # 鑾峰彇鎾や拱鍏ュ紑濮嬭绠楃殑淇℃伅
     # 杩斿洖鏁版嵁鐨勫唴瀹逛负锛氭挙閿�鐐圭储寮� 鎾や拱绾拱棰� 璁$畻鐨勬暟鎹储寮�
     @staticmethod
     def get_buy_cancel_single_pos(code):
         redis = TradePointManager.__get_redis()
-        info = redis.get("buy_cancel_single_pos-{}".format(code))
+        info = RedisUtils.get(redis, "buy_cancel_single_pos-{}".format(code))
         if info is None:
             return None
         else:
@@ -110,27 +110,27 @@
     def set_buy_cancel_single_pos(cls, code, index):
         redis = TradePointManager.__get_redis()
         expire = tool.get_expire()
-        RedisUtils.setex(redis,"buy_cancel_single_pos-{}".format(code), expire, index)
+        RedisUtils.setex(redis, "buy_cancel_single_pos-{}".format(code), expire, index)
 
     # 鍒犻櫎涔版挙鐐规暟鎹�
     @classmethod
     def delete_buy_cancel_point(cls, code):
         redis = TradePointManager.__get_redis()
-        RedisUtils.delete( redis, "buy_cancel_single_pos-{}".format(code))
+        RedisUtils.delete(redis, "buy_cancel_single_pos-{}".format(code))
 
     # 璁剧疆涔版挙绾拱棰�
     @classmethod
     def set_compute_info_for_cancel_buy(cls, code, index, nums):
         redis = TradePointManager.__get_redis()
         expire = tool.get_expire()
-        RedisUtils.setex( redis,"compute_info_for_cancel_buy-{}".format(code), expire, json.dumps((index, nums)))
+        RedisUtils.setex(redis, "compute_info_for_cancel_buy-{}".format(code), expire, json.dumps((index, nums)))
         logger_l2_trade_buy.info("{}淇濆瓨鎾ゅ崟绾拱棰濅俊鎭細{}锛寋}", code, index, nums)
 
     # 鑾峰彇涔版挙绾拱棰濊绠椾俊鎭�
     @classmethod
     def get_compute_info_for_cancel_buy(cls, code):
         redis = TradePointManager.__get_redis()
-        info = redis.get("compute_info_for_cancel_buy-{}".format(code))
+        info = RedisUtils.get(redis, "compute_info_for_cancel_buy-{}".format(code))
         if info is None:
             return None, 0
         else:
@@ -147,14 +147,15 @@
     def set_count_info_for_cancel_buy(cls, code, index, buy_count, cancel_count):
         redis = TradePointManager.__get_redis()
         expire = tool.get_expire()
-        RedisUtils.setex(redis,"count_info_for_cancel_buy-{}".format(code), expire, json.dumps((index, buy_count, cancel_count)))
+        RedisUtils.setex(redis, "count_info_for_cancel_buy-{}".format(code), expire,
+                         json.dumps((index, buy_count, cancel_count)))
         logger_l2_trade_buy.info("{}淇濆瓨鎾ゅ崟绾拱棰濅俊鎭細{}锛寋}", code, index, buy_count, cancel_count)
 
     # 鑾峰彇涔版挙绾拱棰濊绠椾俊鎭�
     @classmethod
     def get_count_info_for_cancel_buy(cls, code):
         redis = TradePointManager.__get_redis()
-        info = redis.get("count_info_for_cancel_buy-{}".format(code))
+        info = RedisUtils.get(redis, "count_info_for_cancel_buy-{}".format(code))
         if info is None:
             return None, 0, 0
         else:
@@ -170,9 +171,9 @@
 # 娓呴櫎l2鏁版嵁
 def clear_l2_data(code):
     redis_l2 = redis_manager.RedisManager(1).getRedis()
-    keys = RedisUtils.keys( redis_l2, "l2-{}-*".format(code))
+    keys = RedisUtils.keys(redis_l2, "l2-{}-*".format(code))
     for k in keys:
-        RedisUtils.delete(redis_l2,k)
+        RedisUtils.delete(redis_l2, k)
 
     RedisUtils.delete(redis_l2, "l2-data-latest-{}".format(code))
 
@@ -184,7 +185,7 @@
 def init_l2_fixed_codes():
     key = "l2-fixed-codes"
     redis = _redisManager.getRedis()
-    count = redis.scard(key)
+    count = RedisUtils.scard(redis, key)
     if count > 0:
         RedisUtils.delete(redis, key)
     RedisUtils.sadd(redis, key, "000000")
@@ -195,7 +196,7 @@
 def remove_from_l2_fixed_codes(code):
     key = "l2-fixed-codes"
     redis = _redisManager.getRedis()
-    redis.srem(key, code)
+    RedisUtils.srem(redis, key, code)
 
 
 # 娣诲姞浠g爜鍒癓2鍥哄畾鐩戞帶
@@ -210,7 +211,7 @@
 def is_in_l2_fixed_codes(code):
     key = "l2-fixed-codes"
     redis = _redisManager.getRedis()
-    return redis.sismember(key, code)
+    return RedisUtils.sismember(redis, key, code)
 
 
 if __name__ == "__main__":

--
Gitblit v1.8.0