From 4cd1733f93d0e905a8b06d284eddef064e4d1ca6 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期二, 20 二月 2024 18:15:56 +0800
Subject: [PATCH] 客户端推送消息修改/量参考日期规则修改

---
 msg/push_msg_manager.py |   32 +++++++++++++++++++++++++++++---
 1 files changed, 29 insertions(+), 3 deletions(-)

diff --git a/msg/push_msg_manager.py b/msg/push_msg_manager.py
index 0e8676d..f273b39 100644
--- a/msg/push_msg_manager.py
+++ b/msg/push_msg_manager.py
@@ -9,12 +9,37 @@
 
 TYPE_DELEGATE_QUEUE_CHANGE = "delegate_queue_change"  # 濮旀墭闃熷垪鍙樺寲
 
+TYPE_DELEGATE_ORDER_DANGER = "delegate_order_danger"  # 濮旀墭鐨勮鍗曞嵄闄�
+
 thread_pool = concurrent.futures.ThreadPoolExecutor(max_workers=5)
+
+__order_almost_deal_filter_dict = {}
+
+__delegate_order_danger_filter_dict = {}
 
 
 # 鎺ㄩ�佽鍗曞揩鎴愪氦淇℃伅
-def push_order_almost_deal(code, code_name):
-    __push_msg(TYPE_ORDER_ALMOST_DEAL, data={"code": code, "code_name": code_name})
+def push_order_almost_deal(code, code_name, buy_single_index, msg, ctype='', force=False):
+    if code not in __order_almost_deal_filter_dict:
+        __order_almost_deal_filter_dict[code] = set()
+    notify_key = f"{buy_single_index}_{ctype}"
+    if force or notify_key not in __order_almost_deal_filter_dict[code]:
+        __order_almost_deal_filter_dict[code].add(notify_key)
+        # 寮哄埗鎺ㄩ�佹垨杩欐涓嬪崟涔嬪墠鏈帹閫�
+        __push_msg(TYPE_ORDER_ALMOST_DEAL,
+                   data={"code": code, "code_name": code_name, "msg": msg})
+
+
+# 鎺ㄩ�佸鎵樼殑璁㈠崟鍗遍櫓淇℃伅
+def push_delegate_order_danger(code, code_name, buy_single_index, msg, ctype='', force=False):
+    if code not in __delegate_order_danger_filter_dict:
+        __delegate_order_danger_filter_dict[code] = set()
+    notify_key = f"{buy_single_index}_{ctype}"
+    if force or notify_key not in __delegate_order_danger_filter_dict[code]:
+        __delegate_order_danger_filter_dict[code].add(notify_key)
+        # 寮哄埗鎺ㄩ�佹垨杩欐涓嬪崟涔嬪墠鏈帹閫�
+        __push_msg(TYPE_DELEGATE_ORDER_DANGER,
+                   data={"code": code, "code_name": code_name, "msg": msg})
 
 
 # 鎺ㄩ�佸鎵橀槦鍒楀彉鍖栨秷鎭�
@@ -29,4 +54,5 @@
         if data:
             fdata["data"] = data
         middle_api_protocol.request(middle_api_protocol.load_push_msg(fdata))
-    thread_pool.submit(push)
\ No newline at end of file
+
+    thread_pool.submit(push)

--
Gitblit v1.8.0