From bba825d6ddfaca81c18890a5c3d3a1f0a346fd42 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期三, 30 七月 2025 17:12:47 +0800
Subject: [PATCH] 定时K线更新/L1目标代码更新
---
l2/l2_data_manager.py | 274 +++++++++++++++++++++++++++++++-----------------------
1 files changed, 158 insertions(+), 116 deletions(-)
diff --git a/l2/l2_data_manager.py b/l2/l2_data_manager.py
index cd31dc7..bbbe0ef 100644
--- a/l2/l2_data_manager.py
+++ b/l2/l2_data_manager.py
@@ -3,11 +3,63 @@
"""
import json
-from db import redis_manager
-import tool
-from log import logger_l2_trade_buy
+from db import redis_manager_delegate as redis_manager
+from db.redis_manager_delegate import RedisUtils
+from utils import tool
+from utils.tool import CodeDataCacheUtil
+_db = 1
_redisManager = redis_manager.RedisManager(1)
+
+
+# 涓嬪崟涓存椂淇℃伅
+class OrderBeginPosInfo(object):
+ MODE_NORMAL = 0 # 鏅�氫笅鍗�
+ MODE_FAST = 1
+ MODE_ACTIVE = 2 # 绉瀬鍔ㄤ拱
+ MODE_RADICAL = 3 # 鎵叆
+ MODE_OPEN_LIMIT_UP = 4 # 鎺�1
+
+ # mode: 0-鏅�氫氦鏄� 1-蹇�熶氦鏄�
+ def __init__(self, buy_single_index=None, buy_exec_index=-1, buy_compute_index=None, num=0, count=0,
+ max_num_set=None, buy_volume_rate=None, sell_info=None, threshold_money=None, mode=0, mode_desc=None,
+ at_limit_up=False, first_limit_up_buy=False, min_order_no = None):
+ self.buy_single_index = buy_single_index
+ self.buy_exec_index = buy_exec_index
+ self.buy_compute_index = buy_compute_index
+ self.num = num
+ self.count = count
+ self.threshold_money = threshold_money
+ if max_num_set:
+ self.max_num_set = list(max_num_set)
+ else:
+ self.max_num_set = []
+ self.buy_volume_rate = buy_volume_rate
+ self.sell_info = sell_info
+ self.mode = mode
+ self.mode_desc = mode_desc
+ # 鏄惁鏄澘涓婁拱
+ self.at_limit_up = at_limit_up
+ # 鏄惁涓洪灏佷拱
+ self.first_limit_up_buy = first_limit_up_buy
+ # 缁熻鎵规澶у崟鎴愪氦鐨勬渶灏忚鍗曞彿
+ self.min_order_no = min_order_no
+
+ def get_max_num_set(self):
+ if self.max_num_set:
+ return set(self.max_num_set)
+ return None
+
+ def to_json_str(self):
+ return json.dumps(vars(self))
+
+ def to_dict(self):
+ return vars(self)
+
+ @classmethod
+ def to_object(cls, json_str: str):
+ d = json.loads(json_str)
+ return OrderBeginPosInfo(**d)
class L2DataException(Exception):
@@ -30,168 +82,158 @@
# 浜ゆ槗鐐圭鐞嗗櫒锛岀敤浜庣鐞嗕拱鍏ョ偣锛涗拱鎾ょ偣锛涜窛绂讳拱鍏ョ偣鐨勫噣涔板叆鏁版嵁锛涜窛绂讳拱鎾ょ偣鐨勪拱鎾ゆ暟鎹�
class TradePointManager:
- @staticmethod
- def __get_redis():
- return _redisManager.getRedis()
+ __db = 1
+ __redisManager = redis_manager.RedisManager(1)
+ __buy_compute_index_info_cache = {}
+ __buy_cancel_single_pos_cache = {}
+ __instance = None
+ # 鏈�杩戠殑涓嬪崟妯″紡
+ __latest_place_order_mode_cache = {}
+
+ def __new__(cls, *args, **kwargs):
+ if not cls.__instance:
+ cls.__instance = super(TradePointManager, cls).__new__(cls, *args, **kwargs)
+ cls.load_data()
+
+ return cls.__instance
+
+ @classmethod
+ def __get_redis(cls):
+ return cls.__redisManager.getRedis()
+
+ @classmethod
+ def load_data(cls):
+ redis_ = cls.__get_redis()
+ keys = RedisUtils.keys(redis_, "buy_compute_index_info-*")
+ for k in keys:
+ code = k.split("-")[-1]
+ val = RedisUtils.get(redis_, k)
+ order = OrderBeginPosInfo.to_object(val)
+ CodeDataCacheUtil.set_cache(cls.__buy_compute_index_info_cache, code, order)
+
+ keys = RedisUtils.keys(redis_, "buy_cancel_single_pos-*")
+ for k in keys:
+ code = k.split("-")[-1]
+ val = RedisUtils.get(redis_, k)
+ CodeDataCacheUtil.set_cache(cls.__buy_cancel_single_pos_cache, code, int(val))
# 鍒犻櫎涔板叆鐐规暟鎹�
- @staticmethod
- def delete_buy_point(code):
- redis = TradePointManager.__get_redis()
- redis.delete("buy_compute_index_info-{}".format(code))
+ def delete_buy_point(self, code):
+ CodeDataCacheUtil.clear_cache(self.__buy_compute_index_info_cache, code)
+ RedisUtils.delete_async(self.__db, "buy_compute_index_info-{}".format(code))
# 鑾峰彇涔板叆鐐逛俊鎭�
# 杩斿洖鏁版嵁涓猴細涔板叆鐐� 绱绾拱棰� 宸茬粡璁＄畻鐨勬暟鎹储寮�
- @staticmethod
- def get_buy_compute_start_data(code):
- redis = TradePointManager.__get_redis()
+ def get_buy_compute_start_data(self, code):
_key = "buy_compute_index_info-{}".format(code)
- _data_json = redis.get(_key)
+ _data_json = RedisUtils.get(self.__get_redis(), _key)
if _data_json is None:
- return None, None, None, 0, 0, []
+ return None, None, None, 0, 0, [], 0
_data = json.loads(_data_json)
- return _data[0], _data[1], _data[2], _data[3], _data[4], _data[5]
+ return _data[0], _data[1], _data[2], _data[3], _data[4], _data[5], _data[6]
+
+ def get_buy_compute_start_data_cache(self, code) -> OrderBeginPosInfo:
+ cache_result = CodeDataCacheUtil.get_cache(self.__buy_compute_index_info_cache, code)
+ if cache_result[0]:
+ return cache_result[1]
+
+ return OrderBeginPosInfo()
# 璁剧疆涔板叆鐐圭殑鍊�
# buy_single_index 涔板叆淇″彿浣�
# buy_exec_index 涔板叆鎵ц浣�
# compute_index 璁＄畻浣嶇疆
# nums 绱绾拱棰�
- @staticmethod
- def set_buy_compute_start_data(code, buy_single_index, buy_exec_index, compute_index, nums, count, max_num_sets):
- redis = TradePointManager.__get_redis()
+
+ def set_buy_compute_start_data_v2(self, code, order: OrderBeginPosInfo):
+
+ if order.mode is not None:
+ self.__latest_place_order_mode_cache[code] = order.mode
+
expire = tool.get_expire()
_key = "buy_compute_index_info-{}".format(code)
- if buy_single_index is not None:
- redis.setex(_key, expire,
- json.dumps((buy_single_index, buy_exec_index, compute_index, nums, count, list(max_num_sets))))
+ data_ = None
+ if order.buy_single_index is not None:
+ data_ = order
else:
- _buy_single_index, _buy_exec_index, _compute_index, _nums, _count, _max_num_index = TradePointManager.get_buy_compute_start_data(code)
- redis.setex(_key, expire,
- json.dumps((_buy_single_index, buy_exec_index, compute_index, nums, count, list(max_num_sets))))
+ old_order = self.get_buy_compute_start_data_cache(code)
+ order.buy_single_index = old_order.buy_single_index
+ data_ = order
+ CodeDataCacheUtil.set_cache(self.__buy_compute_index_info_cache, code, data_)
+ RedisUtils.setex_async(self.__db, _key, expire, data_.to_json_str())
- # 鑾峰彇鎾や拱鍏ュ紑濮嬭绠楃殑淇℃伅
- # 杩斿洖鏁版嵁鐨勫唴瀹逛负锛氭挙閿�鐐圭储寮� 鎾や拱绾拱棰� 璁＄畻鐨勬暟鎹储寮�
- @staticmethod
- def get_buy_cancel_single_pos(code):
- redis = TradePointManager.__get_redis()
- info = redis.get("buy_cancel_single_pos-{}".format(code))
- if info is None:
- return None
- else:
- return int(info)
-
- # 璁剧疆涔版挙鐐逛俊鎭�
- # buy_num 绾拱棰� computed_index璁＄畻鍒扮殑涓嬫爣 index鎾や拱淇″彿璧风偣
+ # 鏄惁宸茬粡涓嬪崟
+ @classmethod
+ def is_placed_order(cls, order_begin_pos: OrderBeginPosInfo):
+ return order_begin_pos and order_begin_pos.buy_exec_index is not None and order_begin_pos.buy_exec_index > -1
@classmethod
- def set_buy_cancel_single_pos(cls, code, index):
- redis = TradePointManager.__get_redis()
- expire = tool.get_expire()
- redis.setex("buy_cancel_single_pos-{}".format(code), expire, index)
-
- # 鍒犻櫎涔版挙鐐规暟鎹�
- @classmethod
- def delete_buy_cancel_point(cls, code):
- redis = TradePointManager.__get_redis()
- redis.delete("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()
- redis.setex("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))
- if info is None:
- return None, 0
- else:
- info = json.loads(info)
- return info[0], info[1]
-
- @classmethod
- def delete_compute_info_for_cancel_buy(cls, code):
- redis = TradePointManager.__get_redis()
- redis.delete("compute_info_for_cancel_buy-{}".format(code))
-
- # 浠庝拱鍏ヤ俊鍙峰紑濮嬭缃定鍋滀拱涓庢定鍋滄挙鐨勫崟鏁�
- @classmethod
- def set_count_info_for_cancel_buy(cls, code, index, buy_count, cancel_count):
- redis = TradePointManager.__get_redis()
- expire = tool.get_expire()
- redis.setex("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))
- if info is None:
- return None, 0, 0
- else:
- info = json.loads(info)
- return info[0], info[1], info[2]
-
- @classmethod
- def delete_count_info_for_cancel_buy(cls, code):
- redis = TradePointManager.__get_redis()
- redis.delete("count_info_for_cancel_buy-{}".format(code))
-
+ def get_latest_place_order_mode(cls, code):
+ """
+ 鑾峰彇鏈�杩戜笅鍗曠殑妯″紡
+ @param code:
+ @return:
+ """
+ return cls.__latest_place_order_mode_cache.get(code)
# 娓呴櫎l2鏁版嵁
def clear_l2_data(code):
redis_l2 = redis_manager.RedisManager(1).getRedis()
- keys = redis_l2.keys("l2-{}-*".format(code))
- for k in keys:
- redis_l2.delete(k)
+ try:
+ keys = RedisUtils.keys(redis_l2, "l2-{}-*".format(code), auto_free=False)
+ for k in keys:
+ RedisUtils.delete(redis_l2, k, auto_free=False)
- redis_l2.delete("l2-data-latest-{}".format(code))
-
-
+ RedisUtils.delete(redis_l2, "l2-data-latest-{}".format(code), auto_free=False)
+ finally:
+ RedisUtils.realse(redis_l2)
second_930 = 9 * 3600 + 30 * 60 + 0
+
# 鍒濆鍖杔2鍥哄畾浠ｇ爜搴�
def init_l2_fixed_codes():
key = "l2-fixed-codes"
redis = _redisManager.getRedis()
- count = redis.scard(key)
- if count > 0:
- redis.delete(key)
- redis.sadd(key, "000000")
- redis.expire(key, tool.get_expire())
+ try:
+ count = RedisUtils.scard(redis, key, auto_free=False)
+ if count > 0:
+ RedisUtils.delete(redis, key, auto_free=False)
+ RedisUtils.sadd(redis, key, "000000", auto_free=False)
+ RedisUtils.expire(redis, key, tool.get_expire(), auto_free=False)
+ finally:
+ RedisUtils.realse(redis)
# 绉婚櫎l2鍥哄畾鐩戞帶浠ｇ爜
def remove_from_l2_fixed_codes(code):
key = "l2-fixed-codes"
- redis = _redisManager.getRedis()
- redis.srem(key, code)
+ RedisUtils.srem(_redisManager.getRedis(), key, code)
# 娣诲姞浠ｇ爜鍒癓2鍥哄畾鐩戞帶
def add_to_l2_fixed_codes(code):
key = "l2-fixed-codes"
- redis = _redisManager.getRedis()
- redis.sadd(key, code)
- redis.expire(key, tool.get_expire())
+ RedisUtils.sadd(_redisManager.getRedis(), key, code)
+ RedisUtils.expire(_redisManager.getRedis(), key, tool.get_expire())
# 鏄惁鍦╨2鍥哄畾鐩戞帶浠ｇ爜涓�
def is_in_l2_fixed_codes(code):
key = "l2-fixed-codes"
- redis = _redisManager.getRedis()
- return redis.sismember(key, code)
+ return RedisUtils.sismember(_redisManager.getRedis(), key, code)
if __name__ == "__main__":
- clear_l2_data("603912")
+ code = "002886"
+ TradePointManager().set_buy_compute_start_data_v2(code, OrderBeginPosInfo(buy_single_index=10,
+ buy_exec_index=30,
+ buy_compute_index=40,
+ num=20000, count=10,
+ buy_volume_rate=0.6,
+ mode=OrderBeginPosInfo.MODE_NORMAL,
+ ))
+ print(TradePointManager().get_buy_compute_start_data_cache(code).max_num_set)
+ RedisUtils.run_loop()
--
Gitblit v1.8.0