From 05c48c8c5fd2570ed078d99c2231e2cf7cf28a4f Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 09 Jun 2026 10:08:59 +0800
Subject: [PATCH] ``` feat(gateApi): 添加网格交易最大持仓限制功能

---
 src/main/java/com/xcong/excoin/modules/gateApi/GridElement.java |  181 ++++++++++++++++++++++++++++++++++++++------
 1 files changed, 154 insertions(+), 27 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/gateApi/GridElement.java b/src/main/java/com/xcong/excoin/modules/gateApi/GridElement.java
index 75359b5..8601f69 100644
--- a/src/main/java/com/xcong/excoin/modules/gateApi/GridElement.java
+++ b/src/main/java/com/xcong/excoin/modules/gateApi/GridElement.java
@@ -7,34 +7,56 @@
 import java.util.concurrent.ConcurrentHashMap;
 
 /**
- * 网格元素,表示网格交易策略中的一个价格层级。
+ * 网格价格层级,策略的最小操作单元。
  *
- * <p>每个网格层级包含该价格位置上的多仓和空仓挂单信息,
- * 通过编号和网格价格唯一标识一个网格层级。
+ * <h3>定位</h3>
+ * 每个 GridElement 对应网格中的一个价格点,同时持有该点的多仓和空仓挂单状态。
+ * 与传统的"价格队列+Map"模式不同,GridElement 将价格、方向、状态、订单ID 聚合到一个对象中,
+ * 并维护全局静态 HashMap 索引实现 O(1) 双向查询。
  *
- * <h3>关键字段</h3>
- * <ul>
- *   <li><b>编号</b>:网格层级的唯一标识</li>
- *   <li><b>网格价格</b>:该层级对应的触发价格</li>
- *   <li><b>多仓挂单</b>:是否已有多头挂单、多头挂单参数</li>
- *   <li><b>空仓挂单</b>:是否有空头挂单、空头挂单参数</li>
- * </ul>
+ * <h3>ID 体系与链表</h3>
+ * <pre>
+ *   ID ≦ -1:  空仓队列区域(降序),ID 自减,gridPrice 递减
+ *   ID  =  0:  基座位置,gridPrice = shortBaseEntryPrice
+ *   ID ≧  1:  多仓队列区域(升序),ID 自增,gridPrice 递增
+ *
+ *   链表: ... ← -3 ← -2 ← -1 ← 0 → 1 → 2 → 3 → ...
+ *         (通过 upId/downId + INDEX 实现 O(1) 遍历)
+ * </pre>
+ *
+ * <h3>字段分组</h3>
+ * <table>
+ *   <tr><th>类别</th><th>字段</th><th>说明</th></tr>
+ *   <tr><td>标识</td><td>id, gridPrice, upId, downId</td><td>编号、价格、双向链表指针</td></tr>
+ *   <tr><td>多仓订单</td><td>hasLongOrder, longOrderId, longTraderParam</td><td>是否有挂单、订单ID、完整参数</td></tr>
+ *   <tr><td>空仓订单</td><td>hasShortOrder, shortOrderId, shortTraderParam</td><td>是否有挂单、订单ID、完整参数</td></tr>
+ *   <tr><td>止盈</td><td>longTakeProfitOrderId, shortTakeProfitOrderId</td><td>止盈条件单ID</td></tr>
+ * </table>
+ *
+ * <h3>6 个全局 O(1) 索引</h3>
+ * <pre>
+ *   INDEX                 → findById(int)               ID → 元素
+ *   PRICE_INDEX           → findByPrice(BigDecimal)     价格 → 元素
+ *   LONG_ORDER_ID_INDEX   → findByLongOrderId(String)   多仓挂单ID → 元素
+ *   SHORT_ORDER_ID_INDEX  → findByShortOrderId(String)  空仓挂单ID → 元素
+ *   LONG_TP_ORDER_ID_INDEX→ findByLongTakeProfitOrderId(String)  多止盈ID → 元素
+ *   SHORT_TP_ORDER_ID_INDEX→ findByShortTakeProfitOrderId(String) 空止盈ID → 元素
+ * </pre>
+ * 索引通过 {@link #rebuildIndex(List)}(全量重建)或 {@link #refreshIndices()}(增量刷新)
+ * 维护,每次操作后自动打印全量网格状态到控制台。
+ *
+ * <h3>何时填充 TraderParam</h3>
+ * 初始化时 {@code updateGridElements()} 为每个元素预填充 longTraderParam 和 shortTraderParam
+ * (含 direction/entryPrice/takeProfitPrice/quantity),订单ID字段在挂单成功后由
+ * {@link GateGridTradeService} 的 4 个辅助方法写入。
  *
  * <h3>使用示例</h3>
  * <pre>
- *   GridElement element = GridElement.builder()
- *       .id(1)
- *       .gridPrice(new BigDecimal("2300.0"))
- *       .build();
- *
- *   // 挂多仓单
- *   TraderParam longParam = TraderParam.builder()
- *       .direction(TraderParam.Direction.LONG)
- *       .entryPrice(new BigDecimal("2293.7"))
- *       .takeProfitPrice(new BigDecimal("2301.6"))
- *       .build();
- *   element.setHasLongOrder(true);
- *   element.setLongTraderParam(longParam);
+ *   GridElement elem = GridElement.findById(1);
+ *   boolean hasLong = elem.isHasLongOrder();           // 是否挂了多单
+ *   BigDecimal tpPrice = elem.getLongTraderParam().getTakeProfitPrice();  // 止盈价
+ *   elem.getUp();     // 上一个网格(O(1))
+ *   elem.getDown();   // 下一个网格(O(1))
  * </pre>
  *
  * @author Administrator
@@ -65,6 +87,10 @@
     private String longTakeProfitOrderId;
     /** 空仓止盈订单 ID */
     private String shortTakeProfitOrderId;
+    /** 多仓止损订单 ID */
+    private String longStopLossOrderId;
+    /** 空仓止损订单 ID */
+    private String shortStopLossOrderId;
 
     /** 全局 ID 索引,由 {@link GateConfig#setGridElements(List)} 触发重建,O(1) 查找 */
     private static final Map<Integer, GridElement> INDEX = new ConcurrentHashMap<>();
@@ -78,6 +104,10 @@
     private static final Map<String, GridElement> LONG_TP_ORDER_ID_INDEX = new ConcurrentHashMap<>();
     /** 全局空仓止盈订单 ID 索引,由 {@link GateConfig#setGridElements(List)} 触发重建,O(1) 查找 */
     private static final Map<String, GridElement> SHORT_TP_ORDER_ID_INDEX = new ConcurrentHashMap<>();
+    /** 全局多仓止损订单 ID 索引 */
+    private static final Map<String, GridElement> LONG_SL_ORDER_ID_INDEX = new ConcurrentHashMap<>();
+    /** 全局空仓止损订单 ID 索引 */
+    private static final Map<String, GridElement> SHORT_SL_ORDER_ID_INDEX = new ConcurrentHashMap<>();
 
     /**
      * 根据 ID 快速查找网格元素(O(1))。
@@ -140,6 +170,20 @@
     }
 
     /**
+     * 根据多仓止损订单 ID 快速查找网格元素(O(1))。
+     */
+    public static GridElement findByLongStopLossOrderId(String orderId) {
+        return LONG_SL_ORDER_ID_INDEX.get(orderId);
+    }
+
+    /**
+     * 根据空仓止损订单 ID 快速查找网格元素(O(1))。
+     */
+    public static GridElement findByShortStopLossOrderId(String orderId) {
+        return SHORT_SL_ORDER_ID_INDEX.get(orderId);
+    }
+
+    /**
      * 从列表中重建全局 ID 索引和价格索引。
      * 由 {@link GateConfig#setGridElements(List)} 在每次列表变更后调用。
      */
@@ -150,10 +194,13 @@
         SHORT_ORDER_ID_INDEX.clear();
         LONG_TP_ORDER_ID_INDEX.clear();
         SHORT_TP_ORDER_ID_INDEX.clear();
+        LONG_SL_ORDER_ID_INDEX.clear();
+        SHORT_SL_ORDER_ID_INDEX.clear();
         for (GridElement e : elements) {
             INDEX.put(e.getId(), e);
             putDynamicIndices(e);
         }
+        logAll();
     }
 
     /**
@@ -168,9 +215,59 @@
         SHORT_ORDER_ID_INDEX.clear();
         LONG_TP_ORDER_ID_INDEX.clear();
         SHORT_TP_ORDER_ID_INDEX.clear();
+        LONG_SL_ORDER_ID_INDEX.clear();
+        SHORT_SL_ORDER_ID_INDEX.clear();
         for (GridElement e : INDEX.values()) {
             putDynamicIndices(e);
         }
+        logAll();
+    }
+
+    /**
+     * 打印全部网格数据到日志。
+     */
+    public static void logAll() {
+        List<GridElement> sorted = new ArrayList<>(INDEX.values());
+        sorted.sort((a, b) -> Integer.compare(a.getId(), b.getId()));
+        StringBuilder sb = new StringBuilder("\n========== 网格数据 ==========\n");
+        for (GridElement e : sorted) {
+            if (e.isHasLongOrder() || e.isHasShortOrder()
+                    || e.getLongStopLossOrderId() != null || e.getShortStopLossOrderId() != null){
+                sb.append(String.format(
+                        "  ID=%4d  价格=%s  up=%s  down=%s  多仓=%s(%s)  空仓=%s(%s)  多止盈=%s  空止盈=%s  多止损=%s  空止损=%s\n",
+                        e.getId(),
+                        e.getGridPrice(),
+                        e.getUpId(),
+                        e.getDownId(),
+                        e.isHasLongOrder() ? "有" : "无",
+                        e.getLongOrderId() != null ? e.getLongOrderId() : "-",
+                        e.isHasShortOrder() ? "有" : "无",
+                        e.getShortOrderId() != null ? e.getShortOrderId() : "-",
+                        e.getLongTakeProfitOrderId() != null ? e.getLongTakeProfitOrderId() : "-",
+                        e.getShortTakeProfitOrderId() != null ? e.getShortTakeProfitOrderId() : "-",
+                        e.getLongStopLossOrderId() != null ? e.getLongStopLossOrderId() : "-",
+                        e.getShortStopLossOrderId() != null ? e.getShortStopLossOrderId() : "-"
+                ));
+            }
+        }
+        sb.append(String.format(
+                "------------------------------------------------------------\n" +
+                "  索引统计: ID=%d  价格=%d  多仓订单ID=%d  空仓订单ID=%d  多止盈ID=%d  空止盈ID=%d  多止损ID=%d  空止损ID=%d\n",
+                INDEX.size(),
+                PRICE_INDEX.size(),
+                LONG_ORDER_ID_INDEX.size(),
+                SHORT_ORDER_ID_INDEX.size(),
+                LONG_TP_ORDER_ID_INDEX.size(),
+                SHORT_TP_ORDER_ID_INDEX.size(),
+                LONG_SL_ORDER_ID_INDEX.size(),
+                SHORT_SL_ORDER_ID_INDEX.size()
+        ));
+        sb.append(String.format("  多仓订单ID索引: %s\n", LONG_ORDER_ID_INDEX.keySet()));
+        sb.append(String.format("  空仓订单ID索引: %s\n", SHORT_ORDER_ID_INDEX.keySet()));
+        sb.append(String.format("  多止盈ID索引: %s\n", LONG_TP_ORDER_ID_INDEX.keySet()));
+        sb.append(String.format("  空止盈ID索引: %s\n", SHORT_TP_ORDER_ID_INDEX.keySet()));
+        sb.append("================================\n");
+        System.out.println(sb);
     }
 
     /**
@@ -179,10 +276,10 @@
      *
      * @return 已挂多仓条件单的 GridElement 列表
      */
-    public static List<GridElement> findAllLongOrders(BigDecimal currentPrice) {
+    public static List<GridElement> findAllShortOrders(BigDecimal currentPrice) {
         List<GridElement> result = new ArrayList<>();
         for (GridElement e : INDEX.values()) {
-            if (e.isHasLongOrder() && e.getGridPrice().compareTo(currentPrice) < 0) {
+            if (e.isHasShortOrder() && e.getGridPrice().compareTo(currentPrice) < 0 && e.getShortTakeProfitOrderId() == null) {
                 result.add(e);
             }
         }
@@ -195,10 +292,10 @@
      *
      * @return 已挂空仓条件单的 GridElement 列表
      */
-    public static List<GridElement> findAllShortOrders(BigDecimal currentPrice) {
+    public static List<GridElement> findAllLongOrders(BigDecimal currentPrice) {
         List<GridElement> result = new ArrayList<>();
         for (GridElement e : INDEX.values()) {
-            if (e.isHasShortOrder() && e.getGridPrice().compareTo(currentPrice) < 0) {
+            if (e.isHasLongOrder() && e.getGridPrice().compareTo(currentPrice) > 0 && e.getLongTakeProfitOrderId() == null) {
                 result.add(e);
             }
         }
@@ -218,6 +315,12 @@
         }
         if (e.getShortTakeProfitOrderId() != null) {
             SHORT_TP_ORDER_ID_INDEX.put(e.getShortTakeProfitOrderId(), e);
+        }
+        if (e.getLongStopLossOrderId() != null) {
+            LONG_SL_ORDER_ID_INDEX.put(e.getLongStopLossOrderId(), e);
+        }
+        if (e.getShortStopLossOrderId() != null) {
+            SHORT_SL_ORDER_ID_INDEX.put(e.getShortStopLossOrderId(), e);
         }
     }
 
@@ -248,6 +351,8 @@
         this.shortOrderId = builder.shortOrderId;
         this.longTakeProfitOrderId = builder.longTakeProfitOrderId;
         this.shortTakeProfitOrderId = builder.shortTakeProfitOrderId;
+        this.longStopLossOrderId = builder.longStopLossOrderId;
+        this.shortStopLossOrderId = builder.shortStopLossOrderId;
     }
 
     // ==================== 网格层级编号 ====================
@@ -334,6 +439,20 @@
     /** 设置空仓止盈订单 ID */
     public void setShortTakeProfitOrderId(String shortTakeProfitOrderId) { this.shortTakeProfitOrderId = shortTakeProfitOrderId; }
 
+    // ==================== 多仓止损订单 ID ====================
+
+    /** @return 多仓止损订单 ID */
+    public String getLongStopLossOrderId() { return longStopLossOrderId; }
+    /** 设置多仓止损订单 ID */
+    public void setLongStopLossOrderId(String longStopLossOrderId) { this.longStopLossOrderId = longStopLossOrderId; }
+
+    // ==================== 空仓止损订单 ID ====================
+
+    /** @return 空仓止损订单 ID */
+    public String getShortStopLossOrderId() { return shortStopLossOrderId; }
+    /** 设置空仓止损订单 ID */
+    public void setShortStopLossOrderId(String shortStopLossOrderId) { this.shortStopLossOrderId = shortStopLossOrderId; }
+
     public static Builder builder() {
         return new Builder();
     }
@@ -372,6 +491,10 @@
         private String longTakeProfitOrderId;
         /** 空仓止盈订单 ID */
         private String shortTakeProfitOrderId;
+        /** 多仓止损订单 ID */
+        private String longStopLossOrderId;
+        /** 空仓止损订单 ID */
+        private String shortStopLossOrderId;
 
         /** 设置网格层级编号 */
         public Builder id(int id) { this.id = id; return this; }
@@ -397,6 +520,10 @@
         public Builder longTakeProfitOrderId(String longTakeProfitOrderId) { this.longTakeProfitOrderId = longTakeProfitOrderId; return this; }
         /** 设置空仓止盈订单 ID */
         public Builder shortTakeProfitOrderId(String shortTakeProfitOrderId) { this.shortTakeProfitOrderId = shortTakeProfitOrderId; return this; }
+        /** 设置多仓止损订单 ID */
+        public Builder longStopLossOrderId(String longStopLossOrderId) { this.longStopLossOrderId = longStopLossOrderId; return this; }
+        /** 设置空仓止损订单 ID */
+        public Builder shortStopLossOrderId(String shortStopLossOrderId) { this.shortStopLossOrderId = shortStopLossOrderId; return this; }
 
         public GridElement build() {
             return new GridElement(this);

--
Gitblit v1.9.1