From 2e6cbd3ee1c2d7e1aa73f4392c79bfbfacbcb67d Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 15 Dec 2025 14:03:36 +0800
Subject: [PATCH] feat(okx): 支持多账号WebSocket连接管理

---
 src/main/java/com/xcong/excoin/modules/okxNewPrice/okxWs/PositionsWs.java |   60 +++++++++++++++++++++++++++++++++++++++---------------------
 1 files changed, 39 insertions(+), 21 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/okxNewPrice/okxWs/PositionsWs.java b/src/main/java/com/xcong/excoin/modules/okxNewPrice/okxWs/PositionsWs.java
index dad2031..3b08fde 100644
--- a/src/main/java/com/xcong/excoin/modules/okxNewPrice/okxWs/PositionsWs.java
+++ b/src/main/java/com/xcong/excoin/modules/okxNewPrice/okxWs/PositionsWs.java
@@ -19,7 +19,13 @@
 @Slf4j
 public class PositionsWs {
 
-    public static  final Map<String,BigDecimal> POSITIONSWSMAP = new ConcurrentHashMap<>();
+    // 使用双层Map,第一层key为账号名称,第二层key为数据key
+    public static final Map<String, Map<String, BigDecimal>> POSITIONSWSMAP = new ConcurrentHashMap<>();
+    
+    // 获取指定账号的Map,如果不存在则创建
+    public static Map<String, BigDecimal> getAccountMap(String accountName) {
+        return POSITIONSWSMAP.computeIfAbsent(accountName, k -> new ConcurrentHashMap<>());
+    }
 
     public static final String POSITIONSWS_CHANNEL = "positions";
 
@@ -35,22 +41,28 @@
             String connId = MallUtils.getOrderNum(POSITIONSWS_CHANNEL);
             JSONObject jsonObject = WsParamBuild.buildJsonObject(connId, option, argsArray);
             webSocketClient.send(jsonObject.toJSONString());
-            log.info("发送持仓频道频道:{}", option);
+//            log.info("发送持仓频道频道:{}", option);
         } catch (Exception e) {
             log.error("订阅持仓频道频道构建失败", e);
         }
     }
 
-    public static void handleEvent(JSONObject response) {
+    public static void initEvent(JSONObject response, String accountName) {
+//        log.info("订阅成功,数据初始化: {}", response.getJSONObject("arg"));
+        JSONObject arg = response.getJSONObject("arg");
+        initParam(arg, accountName);
+    }
+
+    public static void handleEvent(JSONObject response, String accountName) {
 
 
-        log.info("开始执行PositionsWs......");
+//        log.info("开始执行PositionsWs......");
         try {
             JSONArray dataArray = response.getJSONArray("data");
             if (dataArray == null || dataArray.isEmpty()) {
-                log.info("账户持仓频道数据为空,已当前价买入,并且初始化网格");
+//                log.info("账户持仓频道数据为空,已当前价买入,并且初始化网格");
                 JSONObject posData = new JSONObject();
-                processPositionData(posData);
+                initParam(posData, accountName);
                 return;
             }
 
@@ -58,7 +70,7 @@
                 JSONObject posData = dataArray.getJSONObject(i);
                 String instId = posData.getString("instId");
                 if (CoinEnums.HE_YUE.getCode().equals(instId)) {
-                    log.info("查询到账户{}持仓数据",CoinEnums.HE_YUE.getCode());
+//                    log.info("查询到账户{}持仓数据",CoinEnums.HE_YUE.getCode());
                     String mgnMode = posData.getString("mgnMode");
                     String posSide = posData.getString("posSide");
                     String pos = posData.getString("pos");
@@ -78,20 +90,22 @@
                     String bePx = posData.getString("bePx");
                     String realizedPnl = posData.getString("realizedPnl");
                     String settledPnl = posData.getString("settledPnl");
+                    String fee = posData.getString("fee");
+                    String fundingFee = posData.getString("fundingFee");
                     log.info(
-                            "账户持仓频道-产品类型: {}, 保证金模式: {}, 持仓方向: {}, 持仓数量: {}, 开仓平均价: {}, "
+                            "{}: 账户持仓频道-产品类型: {}, 保证金模式: {}, 持仓方向: {}, 持仓数量: {}, 开仓平均价: {}, "
                                     + "未实现收益: {}, 未实现收益率: {}, 杠杆倍数: {}, 预估强平价: {}, 初始保证金: {}, "
                                     + "维持保证金率: {}, 维持保证金: {}, 以美金价值为单位的持仓数量: {}, 占用保证金的币种: {}, "
                                     + "最新成交价: {}, 最新指数价格: {}, 盈亏平衡价: {}, 已实现收益: {}, 累计已结算收益: {}"
-                                    + "最新标记价格: {}",
-                            instId, mgnMode, posSide, pos, avgPx,
+                                    + "最新标记价格: {},累计手续费: {},累计持仓费: {},",
+                            accountName, instId, mgnMode, posSide, pos, avgPx,
                             upl, uplRatio, lever, liqPx, imr,
                             mgnRatio, mmr, notionalUsd, ccy,
                             last, idxPx, bePx, realizedPnl, settledPnl,
-                            markPx
+                            markPx,fee,fundingFee
                     );
 
-                    processPositionData(posData);
+                    initParam(posData, accountName);
                 }
             }
         } catch (Exception e) {
@@ -99,15 +113,19 @@
         }
     }
 
-    private static void processPositionData(JSONObject posData) {
-        WsMapBuild.saveBigDecimalToMap(POSITIONSWSMAP, "avgPx", WsMapBuild.parseBigDecimalSafe(posData.getString("avgPx")));
-        WsMapBuild.saveBigDecimalToMap(POSITIONSWSMAP, "pos", WsMapBuild.parseBigDecimalSafe(posData.getString("pos")));
-        WsMapBuild.saveBigDecimalToMap(POSITIONSWSMAP, "upl", WsMapBuild.parseBigDecimalSafe(posData.getString("upl")));
-        WsMapBuild.saveBigDecimalToMap(POSITIONSWSMAP, "imr", WsMapBuild.parseBigDecimalSafe(posData.getString("imr")));
-        WsMapBuild.saveBigDecimalToMap(POSITIONSWSMAP, "mgnRatio", WsMapBuild.parseBigDecimalSafe(posData.getString("mgnRatio")));
-        WsMapBuild.saveBigDecimalToMap(POSITIONSWSMAP, "markPx", WsMapBuild.parseBigDecimalSafe(posData.getString("markPx")));
-        WsMapBuild.saveBigDecimalToMap(POSITIONSWSMAP, "bePx", WsMapBuild.parseBigDecimalSafe(posData.getString("bePx")));
-        WsMapBuild.saveBigDecimalToMap(POSITIONSWSMAP, "realizedPnl", WsMapBuild.parseBigDecimalSafe(posData.getString("realizedPnl")));
+    private static void initParam(JSONObject posData, String accountName) {
+        Map<String, BigDecimal> accountMap = getAccountMap(accountName);
+        WsMapBuild.saveBigDecimalToMap(accountMap, "avgPx", WsMapBuild.parseBigDecimalSafe(posData.getString("avgPx")));
+        WsMapBuild.saveBigDecimalToMap(accountMap, "pos", WsMapBuild.parseBigDecimalSafe(posData.getString("pos")));
+        WsMapBuild.saveBigDecimalToMap(accountMap, "upl", WsMapBuild.parseBigDecimalSafe(posData.getString("upl")));
+        WsMapBuild.saveBigDecimalToMap(accountMap, "imr", WsMapBuild.parseBigDecimalSafe(posData.getString("imr")));
+        WsMapBuild.saveBigDecimalToMap(accountMap, "mgnRatio", WsMapBuild.parseBigDecimalSafe(posData.getString("mgnRatio")));
+        WsMapBuild.saveBigDecimalToMap(accountMap, "markPx", WsMapBuild.parseBigDecimalSafe(posData.getString("markPx")));
+        WsMapBuild.saveBigDecimalToMap(accountMap, "bePx", WsMapBuild.parseBigDecimalSafe(posData.getString("bePx")));
+        WsMapBuild.saveBigDecimalToMap(accountMap, "realizedPnl", WsMapBuild.parseBigDecimalSafe(posData.getString("realizedPnl")));
+        WsMapBuild.saveBigDecimalToMap(accountMap, "fee", WsMapBuild.parseBigDecimalSafe(posData.getString("fee")));
+        WsMapBuild.saveBigDecimalToMap(accountMap, "fundingFee", WsMapBuild.parseBigDecimalSafe(posData.getString("fundingFee")));
 
+        WsMapBuild.saveBigDecimalToMap(accountMap, CoinEnums.READY_STATE.name(), WsMapBuild.parseBigDecimalSafe(CoinEnums.READY_STATE_YES.getCode()));
     }
 }

--
Gitblit v1.9.1