From 2f2516749615da866e96d8d24e499b7ecbb63a3e Mon Sep 17 00:00:00 2001 From: Administrator <admin@example.com> Date: 星期一, 23 六月 2025 12:28:52 +0800 Subject: [PATCH] 默认交易模式变更/真实下单位置计算位置修改 --- l2/l2_data_util.py | 388 +++++++++++++++++++++++++++++++++++++++++++------------ 1 files changed, 303 insertions(+), 85 deletions(-) diff --git a/l2/l2_data_util.py b/l2/l2_data_util.py index bac68e1..e12f024 100644 --- a/l2/l2_data_util.py +++ b/l2/l2_data_util.py @@ -9,13 +9,17 @@ import logging import time -import constant -import gpcode_manager -import l2_data_log -import log -from db import redis_manager -import tool +import numpy +import constant +from code_attribute import gpcode_manager +from db.redis_manager_delegate import RedisUtils +from l2 import l2_data_log, l2_data_source_util +from log_module import log, log_export, async_log_util +from db import redis_manager_delegate as redis_manager +from utils import tool + +__db = 1 _redisManager = redis_manager.RedisManager(1) # l2鏁版嵁绠$悊 # 鏈湴鏈�鏂颁竴娆′笂浼犵殑鏁版嵁 @@ -26,26 +30,38 @@ # 鐢ㄤ簬鍔犲揩鏁版嵁澶勭悊锛岀敤绌烘崲鏃堕棿 local_today_num_operate_map = {} +# 涔板叆璁㈠崟鍙锋槧灏�,鍙湁鍘熺敓鐨凩2鏁版嵁鎵嶆湁 +local_today_buyno_map = {} -def load_l2_data(code, force=False): - redis = _redisManager.getRedis() +# 鍗栧嚭璁㈠崟鍙锋槧灏勶紝鍙湁鍘熺敓鐨凩2鏁版嵁鎵嶆湁 +local_today_sellno_map = {} + +# 宸茬粡鎾ゅ崟鐨勮鍗曞彿 +local_today_canceled_buyno_map = {} + + +def load_l2_data(code, load_latest=True, force=False): # 鍔犺浇鏈�杩戠殑l2鏁版嵁 - if local_latest_datas.get(code) is None or force: - # 鑾峰彇鏈�杩戠殑鏁版嵁 - _data = redis.get("l2-data-latest-{}".format(code)) - if _data is not None: - if code in local_latest_datas: - local_latest_datas[code] = json.loads(_data) - else: - local_latest_datas.setdefault(code, json.loads(_data)) + if load_latest: + if local_latest_datas.get(code) is None or force: + # 鑾峰彇鏈�杩戠殑鏁版嵁 + _data = RedisUtils.get(_redisManager.getRedis(), "l2-data-latest-{}".format(code)) + if _data is not None: + if code in local_latest_datas: + local_latest_datas[code] = json.loads(_data) + else: + local_latest_datas.setdefault(code, json.loads(_data)) # 鑾峰彇浠婃棩鐨勬暟鎹� if local_today_datas.get(code) is None or force: - datas = log.load_l2_from_log() + datas = log_export.load_l2_from_log() datas = datas.get(code) if datas is None: datas = [] local_today_datas[code] = datas + data_normal = True + if datas and len(datas) < datas[-1]["index"] + 1: + data_normal = False # 浠庢暟鎹簱鍔犺浇 # datas = [] @@ -60,6 +76,44 @@ # local_today_datas[code] = new_datas # 鏍规嵁浠婃棩鏁版嵁鍔犺浇 load_num_operate_map(local_today_num_operate_map, code, local_today_datas.get(code), force) + load_buy_no_map(local_today_buyno_map, code, local_today_datas.get(code), force) + load_sell_no_map(local_today_sellno_map, code, local_today_datas.get(code), force) + load_canceled_buy_no_map(local_today_canceled_buyno_map, code, local_today_datas.get(code), force) + return data_normal + return True + + +# L2鏁版嵁鏄惁姝e父 +def l2_data_is_normal(code): + datas = local_today_datas.get(code) + if not datas: + # 鍒濆鍖� + local_today_datas[code] = [] + load_num_operate_map(local_today_num_operate_map, code, local_today_datas.get(code)) + load_buy_no_map(local_today_buyno_map, code, local_today_datas.get(code)) + load_sell_no_map(local_today_sellno_map, code, local_today_datas.get(code)) + load_canceled_buy_no_map(local_today_canceled_buyno_map, code, local_today_datas.get(code)) + + if datas and len(datas) != datas[-1]["index"] + 1: + return False + return True + + +# 鍔犺浇鎵�鏈夌殑l2鏁版嵁 +def load_l2_data_all(force=False): + datas = log_export.load_l2_from_log() + for code in datas: + if force: + local_today_datas[code] = datas[code] + else: + if code not in local_today_datas: + local_today_datas[code] = datas[code] + load_num_operate_map(local_today_num_operate_map, code, local_today_datas.get(code), force) + load_buy_no_map(local_today_buyno_map, code, local_today_datas.get(code), force) + load_sell_no_map(local_today_sellno_map, code, local_today_datas.get(code), force) + load_canceled_buy_no_map(local_today_canceled_buyno_map, code, local_today_datas.get(code), force) + + constant.L2_DATA_IS_LOADED = True # 灏嗘暟鎹牴鎹畁um-operate鍒嗙被 @@ -76,6 +130,64 @@ local_today_num_operate_map[code].get(key).append(data) +# 灏嗘暟鎹牴鎹畂rderNo鍒嗙被,鍘熺敓鏁版嵁鎵嶆湁 +def load_buy_no_map(local_today_buyno_map, code, source_datas, clear=False): + # 鍙湁鍘熺敓L2鏁版嵁鎵嶄細鏈夋鎿嶄綔 + if constant.L2_SOURCE_TYPE != constant.L2_SOURCE_TYPE_HUAXIN: + return + if local_today_buyno_map.get(code) is None: + local_today_buyno_map[code] = {} + if clear: + local_today_buyno_map[code] = {} + + for data in source_datas: + if data["val"]["operateType"] != 0: + continue + # 鍙~鍏呬拱鍏ユ暟鎹� + key = "{}".format(data["val"]["orderNo"]) + if local_today_buyno_map[code].get(key) is None: + local_today_buyno_map[code].setdefault(key, data) + + +# 灏嗘暟鎹牴鎹畂rderNo鍒嗙被,鍘熺敓鏁版嵁鎵嶆湁 +def load_sell_no_map(local_today_sellno_map, code, source_datas, clear=False): + # 鍙湁鍘熺敓L2鏁版嵁鎵嶄細鏈夋鎿嶄綔 + if constant.L2_SOURCE_TYPE != constant.L2_SOURCE_TYPE_HUAXIN: + return + if local_today_sellno_map.get(code) is None: + local_today_sellno_map[code] = {} + if clear: + local_today_sellno_map[code] = {} + + for data in source_datas: + if data["val"]["operateType"] != 2: + continue + # 鍙~鍏呬拱鍏ユ暟鎹� + key = "{}".format(data["val"]["orderNo"]) + if local_today_sellno_map[code].get(key) is None: + local_today_sellno_map[code].setdefault(key, data) + + +# 灏嗘暟鎹牴鎹畂rderNo鍒嗙被宸叉挙璁㈠崟,鍘熺敓鏁版嵁鎵嶆湁 +def load_canceled_buy_no_map(local_today_canceled_buyno_map, code, source_datas, clear=False): + # 鍙湁鍘熺敓L2鏁版嵁鎵嶄細鏈夋鎿嶄綔 + if constant.L2_SOURCE_TYPE != constant.L2_SOURCE_TYPE_HUAXIN: + return + if local_today_canceled_buyno_map.get(code) is None: + local_today_canceled_buyno_map[code] = {} + if clear: + local_today_canceled_buyno_map[code] = {} + + for data in source_datas: + # 鍙暀涓嬩拱鎾� + if data["val"]["operateType"] != 1: + continue + # 鍙~鍏呬拱鍏ユ暟鎹� + key = "{}".format(data["val"]["orderNo"]) + if local_today_canceled_buyno_map[code].get(key) is None: + local_today_canceled_buyno_map[code].setdefault(key, data) + + @tool.async_call def saveL2Data(code, datas, msg=""): start_time = round(time.time() * 1000) @@ -86,58 +198,57 @@ redis_instance = _redisManager.getRedis() try: - if redis_instance.setnx("l2-save-{}".format(code), "1") > 0: - + if RedisUtils.setnx(redis_instance, "l2-save-{}".format(code), "1", auto_free=False) > 0: # 璁$畻淇濈暀鐨勬椂闂� expire = tool.get_expire() i = 0 for _data in datas: i += 1 key = "l2-" + _data["key"] - value = redis_instance.get(key) + value = RedisUtils.get(redis_instance, key, auto_free=False) if value is None: # 鏂板 try: value = {"index": _data["index"], "re": _data["re"]} - redis_instance.setex(key, expire, json.dumps(value)) + RedisUtils.setex(redis_instance, key, expire, json.dumps(value), auto_free=False) except: logging.error("鏇存L2鏁版嵁鍑洪敊锛歿} key:{}".format(code, key)) else: json_value = json.loads(value) if json_value["re"] != _data["re"]: json_value["re"] = _data["re"] - redis_instance.setex(key, expire, json.dumps(json_value)) + RedisUtils.setex(redis_instance, key, expire, json.dumps(json_value), auto_free=False) finally: - redis_instance.delete("l2-save-{}".format(code)) + RedisUtils.delete(redis_instance, "l2-save-{}".format(code), auto_free=False) + RedisUtils.realse(redis_instance) print("淇濆瓨鏂版暟鎹敤鏃讹細", msg, "鑰楁椂锛歿}".format(round(time.time() * 1000) - start_time)) return datas # 淇濆瓨l2鏁版嵁 -def save_l2_data(code, datas, add_datas, randomKey=None): - redis = _redisManager.getRedis() +def save_l2_data(code, datas, add_datas): # 鍙湁鏈夋柊鏇炬暟鎹墠闇�瑕佷繚瀛� - if len(add_datas) > 0: + if add_datas: # 淇濆瓨鏈�杩戠殑鏁版嵁 __start_time = round(time.time() * 1000) - redis.setex("l2-data-latest-{}".format(code), tool.get_expire(), json.dumps(datas)) - l2_data_log.l2_time(code, randomKey, round(time.time() * 1000) - __start_time, "淇濆瓨鏈�杩憀2鏁版嵁鐢ㄦ椂") - # 璁剧疆杩涘唴瀛� - local_latest_datas[code] = datas - __set_l2_data_latest_count(code, len(datas)) + if datas: + RedisUtils.setex_async(__db, "l2-data-latest-{}".format(code), tool.get_expire(), + json.dumps(datas)) + # l2_data_log.l2_time(code, round(time.time() * 1000) - __start_time, "淇濆瓨鏈�杩憀2鏁版嵁鐢ㄦ椂") + # 璁剧疆杩涘唴瀛� + local_latest_datas[code] = datas + set_l2_data_latest_count(code, len(datas)) try: - log.logger_l2_data.info("{}-{}", code, add_datas) + async_log_util.l2_data_log.info(log.logger_l2_data, f"{code}-{add_datas}") except Exception as e: logging.exception(e) - saveL2Data(code, add_datas) # 璁剧疆鏈�鏂扮殑l2鏁版嵁閲囬泦鐨勬暟閲� -def __set_l2_data_latest_count(code, count): - redis = _redisManager.getRedis() +def set_l2_data_latest_count(code, count): key = "latest-l2-count-{}".format(code) - redis.setex(key, 2, count) + RedisUtils.setex(_redisManager.getRedis(), key, 2, count) pass @@ -145,10 +256,9 @@ def get_l2_data_latest_count(code): if code is None or len(code) < 1: return 0 - redis = _redisManager.getRedis() key = "latest-l2-count-{}".format(code) - result = redis.get(key) + result = RedisUtils.get(_redisManager.getRedis(), key) if result is None: return 0 else: @@ -164,12 +274,25 @@ channel = data["channel"] capture_time = data["captureTime"] process_time = data["processTime"] + count = data["count"] data = data["data"] - limit_up_price = gpcode_manager.get_limit_up_price(code) - - datas = L2DataUtil.format_l2_data(data, code, limit_up_price) # 鑾峰彇娑ㄥ仠浠� - return day, client, channel, code, capture_time, process_time, datas, data + return day, client, channel, code, capture_time, process_time, data, count + + +# 鍏冩暟鎹槸鍚︽湁宸紓 +def is_origin_data_diffrent(data1, data2): + if data1 is None or data2 is None: + return True + if len(data1) != len(data2): + return True + # 姣旇緝 + data_length = len(data1) + step = len(data1) // 10 + for i in range(0, data_length, step): + if json.dumps(data1[i]) != json.dumps(data2[i]): + return True + return False class L2DataUtil: @@ -177,11 +300,7 @@ def is_same_time(cls, time1, time2): if constant.TEST: return True - time1_s = time1.split(":") - time1_second = int(time1_s[0]) * 3600 + int(time1_s[1]) * 60 + int(time1_s[2]) - time2_s = time2.split(":") - time2_second = int(time2_s[0]) * 3600 + int(time2_s[1]) * 60 + int(time2_s[2]) - if abs(time2_second - time1_second) < 3: + if abs(tool.trade_time_sub(time1, time2)) < 3: return True else: return False @@ -242,7 +361,8 @@ # 淇濆瓨鍒版暟鎹簱锛屾洿鏂皉e鐨勬暟鎹� save_list.append(_ldata) if len(save_list) > 0: - saveL2Data(code, save_list, "淇濆瓨绾犳鏁版嵁") + # 鏆傛椂涓嶅皢鏁版嵁淇濆瓨鍒皉edis + # saveL2Data(code, save_list, "淇濆瓨绾犳鏁版嵁") local_latest_datas[code] = latest_data return _datas @@ -286,7 +406,7 @@ # 鏁版嵁閲嶅娆℃暟榛樿涓�1 datas.append({"key": key, "val": item, "re": 1}) dataIndexs.setdefault(key, len(datas) - 1) - # TODO 娴嬭瘯鐨勬椂鍊欏紑鍚紝鏂逛究璁板綍澶у崟鏁版嵁 + # 娴嬭瘯鐨勬椂鍊欏紑鍚紝鏂逛究璁板綍澶у崟鏁版嵁 # l2_data_util.save_big_data(code, same_time_num, data) return datas @@ -323,11 +443,16 @@ if int(val["operateType"]) != 2: return False + return True - price = float(val["price"]) - num = int(val["num"]) - # if price * num * 100 < 50 * 10000: - # return False + # 娑ㄥ仠鍗栨挙 + @classmethod + def is_limit_up_price_sell_cancel(cls, val): + if int(val["limitPrice"]) != 1: + return False + + if int(val["operateType"]) != 3: + return False return True # 鏄惁娑ㄥ仠涔版挙 @@ -339,10 +464,12 @@ if int(val["operateType"]) != 1: return False - price = float(val["price"]) - num = int(val["num"]) - # if price * num * 100 < 50 * 10000: - # return False + return True + + @classmethod + def is_buy_cancel(cls, val): + if int(val["operateType"]) != 1: + return False return True # 鏄惁鍗栨挙 @@ -359,47 +486,138 @@ return True return False + # 鏄惁涓轰拱 + @classmethod + def is_buy(cls, val): + if int(val["operateType"]) == 0: + return True + return False + + # l2鏃堕棿宸�� + @classmethod + def time_sub_as_ms(cls, val1, val2): + # 璁$畻鏃堕棿宸�� + sub_s = tool.trade_time_sub(val1["time"], val2["time"]) + sub_ms = int(val1["tms"]) - int(val2["tms"]) + fs = sub_s * 1000 + sub_ms + return fs + + @classmethod + def get_time_with_ms(cls, val): + return val["time"] + "." + "{:0>3}".format(int(val["tms"])) + class L2TradeQueueUtils(object): + # 涔板叆鏁版嵁鏄惁宸叉挙 + @classmethod + def __is_cancel(cls, code, data, total_datas, local_today_num_operate_map): + val = data["val"] + cancel_datas = local_today_num_operate_map.get( + "{}-{}-{}".format(val["num"], "1", val["price"])) + # 鏄惁鏈変拱鎾ゆ暟鎹� + if cancel_datas: + for cancel_data in cancel_datas: + buy_index = l2_data_source_util.L2DataSourceUtils.get_buy_index_with_cancel_data_v2(cancel_data, + local_today_buyno_map.get( + code)) + if buy_index == data["index"]: + return True + return False + # 鑾峰彇鎴愪氦杩涘害绱㈠紩 @classmethod - def find_traded_progress_index(cls, buy_1_price, total_datas, local_today_num_operate_map, queueList, + def find_traded_progress_index(cls, code, buy_1_price, total_datas, local_today_num_operate_map, queueList, + last_index, latest_not_limit_up_time=None): + + def find_traded_progress_index_simple(queues): + index_set = set() + for num in queues: + buy_datas = local_today_num_operate_map.get( + "{}-{}-{}".format(num, "0", buy_1_price_format)) + if buy_datas is not None and len(buy_datas) > 0: + for data in buy_datas: + # 鍦ㄦ渶杩戜竴娆¢潪娑ㄥ仠涔�1鏇存柊鐨勬椂闂翠箣鍚庢墠鏈夋晥 + if latest_not_limit_up_time is None or tool.trade_time_sub(data["val"]["time"], + latest_not_limit_up_time) >= 0: + if data["index"] >= last_index: + index_set.add(data["index"]) + index_list = list(index_set) + index_list.sort() + num_list = [] + new_index_list = [] + for index in index_list: + for i in range(0, total_datas[index]["re"]): + num_list.append(total_datas[index]["val"]["num"]) + new_index_list.append(index) + index_list_str = ",".join(list(map(str, num_list))) + queue_list_str = ",".join(list(map(str, queues))) + find_index = index_list_str.find(queue_list_str) + if find_index >= 0: + temp_str = index_list_str[0:find_index] + if temp_str.endswith(","): + temp_str = temp_str[:-1] + if temp_str == "": + return new_index_list[0], new_index_list[0:len(queues)] + start_index = len(temp_str.split(",")) + return new_index_list[start_index], new_index_list[start_index:start_index + len(queues)] + return None, None + + # 3涓暟鎹互涓婄殑涓嶉渶瑕佸垽鏂渶杩戠殑涓�娆℃湭娑ㄥ仠鏃堕棿 + if len(queueList) >= 3: + latest_not_limit_up_time = None + + # 鍒ゆ柇鍖归厤鐨勪綅缃槸鍚﹀彲淇� + def is_trust(indexes): + cha = [] + for i in range(1, len(indexes)): + cha.append(indexes[i] - indexes[i - 1] - 1) + if len(cha) <= 1: + return True + # 鏍囧噯宸皬浜�1 + std_result = numpy.std(cha) + if std_result < 10: + # 缁濆鍙俊 + return True + + for i in range(0, len(cha)): + if abs(cha[i]) > 10: + # 鏈夎秴杩�10 鐨勯渶瑕佸垽鏂袱涓浉涓存暟鎹棿鐨勬湭鎾ょ殑涔板叆鏁伴噺 + buy_count = 0 + for index in range(indexes[i] + 1, indexes[i + 1] - 1): + if L2DataUtil.is_limit_up_price_buy(total_datas[index]["val"]): + if not cls.__is_cancel(code, total_datas[index], total_datas, local_today_num_operate_map): + buy_count += total_datas[index]["re"] + # 鏆傚畾3涓宸寖鍥� + if buy_count >= 3: + return False + return True + if len(queueList) == 0: return None + # last_index涓嶈兘鎾わ紝濡傛灉宸叉挙灏辨竻闆� + if cls.__is_cancel(code, total_datas[last_index], total_datas, local_today_num_operate_map): + last_index = 0 # 琛ラ綈鏁存暟浣�5浣� buy_1_price_format = f"{buy_1_price}" while buy_1_price_format.find(".") < 4: buy_1_price_format = "0" + buy_1_price_format - index_set = set() - for num in queueList: - buy_datas = local_today_num_operate_map.get( - "{}-{}-{}".format(num, "0", buy_1_price_format)) - if buy_datas is not None and len(buy_datas) > 0: - for data in buy_datas: - # 鍦ㄦ渶杩戜竴娆¢潪娑ㄥ仠涔�1鏇存柊鐨勬椂闂翠箣鍚庢墠鏈夋晥 - if latest_not_limit_up_time is None or tool.trade_time_sub(data["val"]["time"], - latest_not_limit_up_time) >= 0: - index_set.add(data["index"]) - index_list = list(index_set) - index_list.sort() - num_list = [] - new_index_list = [] - for index in index_list: - for i in range(0, total_datas[index]["re"]): - num_list.append(total_datas[index]["val"]["num"]) - new_index_list.append(index) - index_list_str = ",".join(list(map(str, num_list))) - queue_list_str = ",".join(list(map(str, queueList))) - find_index = index_list_str.find(queue_list_str) - if find_index >= 0: - temp_str = index_list_str[0:find_index] - if temp_str.endswith(","): - temp_str = temp_str[:-1] - return new_index_list[len(temp_str.split(","))] + # --------鍥犲瓙鏌ユ壘娉曪紙鍥犲瓙鐨勭獥鍙f渶澶т负锛歭en(queueList) ,鏈�灏忎负锛歭en(queueList)/2锛�--------- + max_win_len = len(queueList) + min_win_len = len(queueList) // 2 + if max_win_len == min_win_len: + min_win_len = max_win_len - 1 + for win_len in range(max_win_len, min_win_len, -1): + # 绐楀彛绉诲姩 + for i in range(0, max_win_len - win_len + 1): + queues = queueList[i:i + win_len] + f_start_index, f_indexs = find_traded_progress_index_simple(queues) + if f_start_index and is_trust(f_indexs): + return f_start_index + raise Exception("灏氭湭鎵惧埌鎴愪氦杩涘害") if __name__ == "__main__": - pass + print(L2DataUtil.time_sub_as_ms({"time": "09:46:05", "tms": 480}, {"time": "09:46:04", "tms": 90})) -- Gitblit v1.8.0