From 0426160a1f283c1f810e3059f6037676da8cb110 Mon Sep 17 00:00:00 2001
From: zyy3 <zyy3@zy.com>
Date: Wed, 05 Nov 2025 16:37:01 +0800
Subject: [PATCH] 1

---
 trading-order-huobi/src/main/java/com/yami/trading/huobi/jsws/WebsocketRunClient.java |   12 ++++++++++++
 1 files changed, 12 insertions(+), 0 deletions(-)

diff --git a/trading-order-huobi/src/main/java/com/yami/trading/huobi/jsws/WebsocketRunClient.java b/trading-order-huobi/src/main/java/com/yami/trading/huobi/jsws/WebsocketRunClient.java
index 93a6a2e..86fc437 100644
--- a/trading-order-huobi/src/main/java/com/yami/trading/huobi/jsws/WebsocketRunClient.java
+++ b/trading-order-huobi/src/main/java/com/yami/trading/huobi/jsws/WebsocketRunClient.java
@@ -4,11 +4,14 @@
 import com.google.gson.reflect.TypeToken;
 
 import com.yami.trading.bean.data.domain.Realtime;
+import com.yami.trading.bean.item.domain.Item;
 import com.yami.trading.huobi.data.DataCache;
 import com.yami.trading.huobi.websocket.constant.enums.EStockType;
+import com.yami.trading.service.item.ItemService;
 import lombok.extern.slf4j.Slf4j;
 import org.java_websocket.client.WebSocketClient;
 import org.java_websocket.handshake.ServerHandshake;
+import org.springframework.beans.factory.annotation.Autowired;
 
 import java.lang.reflect.Type;
 import java.math.BigDecimal;
@@ -20,6 +23,9 @@
 public class WebsocketRunClient extends WebSocketClient {
 
     private EStockType eStockType;
+
+    @Autowired
+    ItemService itemService;
 
     public WebsocketRunClient(URI serverUri, EStockType eStockType) {
         // 修改为新的WebSocket服务器地址
@@ -52,6 +58,12 @@
         if(!s.equals("pong") && !s.equals("身份验证成功")){
             try {
                 Map<String, String> stringObjectMap = jsonToMap(s);
+
+                String symbol = stringObjectMap.get("symbol");
+                Item item = itemService.findBySymbol(symbol);
+                if (!item.getType().equalsIgnoreCase(Item.US_STOCKS)) {
+                    return;
+                }
                 Realtime realtime = new Realtime();
                 realtime.setUuid(stringObjectMap.get("pid"));
                 realtime.setSymbol(stringObjectMap.get("symbol"));

--
Gitblit v1.9.3