From 1f54990ccb7ea5c797eeb597d14ba3c6c94c628a Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 15 Dec 2025 17:26:21 +0800
Subject: [PATCH] feat(okxNewPrice): 实现多空双向交易策略

---
 src/main/java/com/xcong/excoin/modules/okxNewPrice/okxWs/TradeOrderWs.java |  109 +++++++++++++++++++++++++++++++++++++-----------------
 1 files changed, 75 insertions(+), 34 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/okxNewPrice/okxWs/TradeOrderWs.java b/src/main/java/com/xcong/excoin/modules/okxNewPrice/okxWs/TradeOrderWs.java
index 6b1be8c..1e86124 100644
--- a/src/main/java/com/xcong/excoin/modules/okxNewPrice/okxWs/TradeOrderWs.java
+++ b/src/main/java/com/xcong/excoin/modules/okxNewPrice/okxWs/TradeOrderWs.java
@@ -22,20 +22,26 @@
 @Slf4j
 public class TradeOrderWs {
 
-    public static  final Map<String,String> TRADEORDERWSMAP = new ConcurrentHashMap<>();
+    // 使用双层Map,第一层key为账号名称,第二层key为数据key
+    public static  final Map<String, Map<String,String>> TRADEORDERWSMAP = new ConcurrentHashMap<>();
+    
+    // 获取指定账号的Map,如果不存在则创建
+    public static Map<String, String> getAccountMap(String accountName) {
+        return TRADEORDERWSMAP.computeIfAbsent(accountName, k -> new ConcurrentHashMap<>());
+    }
 
     public static final String ORDERWS_CHANNEL = "order";
 
-    public static void orderEvent(WebSocketClient webSocketClient, String side) {
+    public static void orderEvent(WebSocketClient webSocketClient, String side, String accountName) {
 
-        log.info("开始执行TradeOrderWs......");
-        String accountReadyState = AccountWs.ACCOUNTWSMAP.get(CoinEnums.READY_STATE.name());
+//        log.info("开始执行TradeOrderWs......");
+        String accountReadyState = AccountWs.getAccountMap(accountName).get(CoinEnums.READY_STATE.name());
         if (!CoinEnums.READY_STATE_YES.getCode().equals(accountReadyState)) {
             log.info("账户通道未就绪,取消发送");
             return;
         }
-        BigDecimal positionsReadyState = PositionsWs.POSITIONSWSMAP.get(CoinEnums.READY_STATE.name()) == null
-                ? BigDecimal.ZERO : PositionsWs.POSITIONSWSMAP.get(CoinEnums.READY_STATE.name());
+        BigDecimal positionsReadyState = PositionsWs.getAccountMap(accountName).get(CoinEnums.READY_STATE.name()) == null
+                ? BigDecimal.ZERO : PositionsWs.getAccountMap(accountName).get(CoinEnums.READY_STATE.name());
         if (WsMapBuild.parseBigDecimalSafe(CoinEnums.READY_STATE_YES.getCode()).compareTo(positionsReadyState) != 0) {
             log.info("仓位通道未就绪,取消发送");
             return;
@@ -45,30 +51,64 @@
             log.warn("下单参数 side 为空,取消发送");
             return;
         }
-        String buyCnt = "";
-        if (OrderParamEnums.HOLDING.getValue().equals(side)){
-            log.info("当前状态为持仓中,取消发送");
-            return;
-        }else if (OrderParamEnums.OUT.getValue().equals(side)){
-            log.info("当前状态为止损");
-            side = OrderParamEnums.SELL.getValue();
-            buyCnt = String.valueOf(PositionsWs.POSITIONSWSMAP.get("pos"));
-        }else if (OrderParamEnums.INIT.getValue().equals(side)){
-            log.info("当前状态为初始化");
-            side = OrderParamEnums.BUY.getValue();
-            buyCnt = InstrumentsWs.INSTRUMENTSWSMAP.get(CoinEnums.BUY_CNT.name());
-        }else if (OrderParamEnums.BUY.getValue().equals(side)){
-            log.info("当前状态为加仓");
-            String buyCntTime = TRADEORDERWSMAP.get("buyCntTime");
-            String buyCntStr = InstrumentsWs.INSTRUMENTSWSMAP.get(CoinEnums.BUY_CNT.name());
-            buyCnt = String.valueOf(new BigDecimal(buyCntTime).multiply(new BigDecimal(buyCntStr)));
-        }else if (OrderParamEnums.SELL.getValue().equals(side)){
-            log.info("当前状态为减仓");
-            buyCnt = String.valueOf(PositionsWs.POSITIONSWSMAP.get("pos"));
-        }else{
-            log.warn("交易状态异常,取消发送");
+        String posSide = InstrumentsWs.getAccountMap(accountName).get(CoinEnums.POSSIDE.name());
+        // 校验必要参数
+        if (StrUtil.isBlank(posSide)) {
+            log.warn("下单参数 posSide 为空,取消发送");
             return;
         }
+
+        String buyCnt = "";
+        if (CoinEnums.POSSIDE_LONG.getCode().equals(posSide)){
+            if (OrderParamEnums.HOLDING.getValue().equals(side)){
+                log.info("当前状态为持仓中,取消发送");
+                return;
+            }else if (OrderParamEnums.OUT.getValue().equals(side)){
+                log.info("当前状态为止损");
+                side = OrderParamEnums.SELL.getValue();
+                buyCnt = String.valueOf(PositionsWs.getAccountMap(accountName).get("pos"));
+            }else if (OrderParamEnums.INIT.getValue().equals(side)){
+                log.info("当前状态为初始化");
+                side = OrderParamEnums.BUY.getValue();
+                buyCnt = InstrumentsWs.getAccountMap(accountName).get(CoinEnums.BUY_CNT.name());
+            }else if (OrderParamEnums.BUY.getValue().equals(side)){
+                log.info("当前状态为加仓");
+                String buyCntTime = getAccountMap(accountName).get("buyCntTime");
+                String buyCntStr = InstrumentsWs.getAccountMap(accountName).get(CoinEnums.BUY_CNT.name());
+                buyCnt = String.valueOf(new BigDecimal(buyCntTime).multiply(new BigDecimal(buyCntStr)));
+            }else if (OrderParamEnums.SELL.getValue().equals(side)){
+                log.info("当前状态为减仓");
+                buyCnt = String.valueOf(PositionsWs.getAccountMap(accountName).get("pos"));
+            }else{
+                log.warn("交易状态异常,取消发送");
+                return;
+            }
+        }else if (CoinEnums.POSSIDE_SHORT.getCode().equals(posSide)){
+            if (OrderParamEnums.HOLDING.getValue().equals(side)){
+                log.info("当前状态为持仓中,取消发送");
+                return;
+            }else if (OrderParamEnums.OUT.getValue().equals(side)){
+                log.info("当前状态为止损");
+                side = OrderParamEnums.BUY.getValue();
+                buyCnt = String.valueOf(PositionsWs.getAccountMap(accountName).get("pos"));
+            }else if (OrderParamEnums.INIT.getValue().equals(side)){
+                log.info("当前状态为初始化");
+                side = OrderParamEnums.SELL.getValue();
+                buyCnt = InstrumentsWs.getAccountMap(accountName).get(CoinEnums.BUY_CNT.name());
+            }else if (OrderParamEnums.BUY.getValue().equals(side)){
+                log.info("当前状态为减仓");
+                buyCnt = String.valueOf(PositionsWs.getAccountMap(accountName).get("pos"));
+            }else if (OrderParamEnums.SELL.getValue().equals(side)){
+                log.info("当前状态为加仓");
+                String buyCntTime = getAccountMap(accountName).get("buyCntTime");
+                String buyCntStr = InstrumentsWs.getAccountMap(accountName).get(CoinEnums.BUY_CNT.name());
+                buyCnt = String.valueOf(new BigDecimal(buyCntTime).multiply(new BigDecimal(buyCntStr)));
+            }else{
+                log.warn("交易状态异常,取消发送");
+                return;
+            }
+        }
+
 
         if (StrUtil.isBlank(buyCnt)) {
             log.warn("下单数量 buyCnt 为空,取消发送");
@@ -83,7 +123,8 @@
             args.put("tdMode", CoinEnums.CROSS.getCode());
             args.put("clOrdId", clOrdId);
             args.put("side", side);
-            args.put("posSide", CoinEnums.POSSIDE_LONG.getCode());
+
+            args.put("posSide", posSide);
             args.put("ordType", CoinEnums.ORDTYPE_MARKET.getCode());
             args.put("sz", buyCnt);
             argsArray.add(args);
@@ -93,12 +134,12 @@
             webSocketClient.send(jsonObject.toJSONString());
             log.info("发送下单频道:{},数量:{}", side, buyCnt);
 
-            WsMapBuild.saveStringToMap(TRADEORDERWSMAP, "buyCntTime",String.valueOf(BigDecimal.ONE));
-            WsMapBuild.saveStringToMap(TRADEORDERWSMAP, "clOrdId", clOrdId);
-            WsMapBuild.saveStringToMap(TRADEORDERWSMAP, "state", CoinEnums.ORDER_FILLED.getCode());
+            WsMapBuild.saveStringToMap(getAccountMap(accountName), "buyCntTime",String.valueOf(BigDecimal.ONE));
+        WsMapBuild.saveStringToMap(getAccountMap(accountName), "clOrdId", clOrdId);
+        WsMapBuild.saveStringToMap(getAccountMap(accountName), "state", CoinEnums.ORDER_FILLED.getCode());
 
-            WsMapBuild.saveBigDecimalToMap(PositionsWs.POSITIONSWSMAP, CoinEnums.READY_STATE.name(), WsMapBuild.parseBigDecimalSafe(CoinEnums.READY_STATE_NO.getCode()));
-            WsMapBuild.saveStringToMap(AccountWs.ACCOUNTWSMAP, CoinEnums.READY_STATE.name(), CoinEnums.READY_STATE_NO.getCode());
+            WsMapBuild.saveBigDecimalToMap(PositionsWs.getAccountMap(accountName), CoinEnums.READY_STATE.name(), WsMapBuild.parseBigDecimalSafe(CoinEnums.READY_STATE_NO.getCode()));
+        WsMapBuild.saveStringToMap(AccountWs.getAccountMap(accountName), CoinEnums.READY_STATE.name(), CoinEnums.READY_STATE_NO.getCode());
 
         } catch (Exception e) {
             log.error("下单构建失败", e);

--
Gitblit v1.9.1