From 39b09846041021c5bd0fea92d981ccb3490638d1 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Fri, 08 May 2026 23:56:43 +0800
Subject: [PATCH] fix(gateApi): 修复双向持仓模式下的杠杆设置功能

---
 src/main/java/com/xcong/excoin/modules/gateApi/wsHandler/AbstractPrivateChannelHandler.java |  146 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 146 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/gateApi/wsHandler/AbstractPrivateChannelHandler.java b/src/main/java/com/xcong/excoin/modules/gateApi/wsHandler/AbstractPrivateChannelHandler.java
new file mode 100644
index 0000000..f3173d2
--- /dev/null
+++ b/src/main/java/com/xcong/excoin/modules/gateApi/wsHandler/AbstractPrivateChannelHandler.java
@@ -0,0 +1,146 @@
+package com.xcong.excoin.modules.gateApi.wsHandler;
+
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
+import com.xcong.excoin.modules.gateApi.GateGridTradeService;
+import lombok.extern.slf4j.Slf4j;
+import org.java_websocket.client.WebSocketClient;
+
+import javax.crypto.Mac;
+import javax.crypto.spec.SecretKeySpec;
+import java.nio.charset.StandardCharsets;
+
+/**
+ * 私有频道处理器的抽象基类。
+ *
+ * <h3>封装内容</h3>
+ * <ul>
+ *   <li>HMAC-SHA512 签名计算(UTF-8 编码)</li>
+ *   <li>认证请求 JSON 构建(id/time/channel/payload/auth)</li>
+ *   <li>subscribe / unsubscribe 的默认实现(含签名)</li>
+ *   <li>用户 ID 获取(从 {@link GateGridTradeService#getUserId()})</li>
+ * </ul>
+ *
+ * <h3>签名算法</h3>
+ * {@code SIGN = Hex(HmacSHA512(secret_utf8, "channel={channel}&event={event}&time={timeSec}"_utf8))}
+ *
+ * <h3>子类</h3>
+ * {@link com.xcong.excoin.modules.gateApi.wsHandler.handler.PositionsChannelHandler}、
+ * {@link com.xcong.excoin.modules.gateApi.wsHandler.handler.PositionClosesChannelHandler}
+ *
+ * @author Administrator
+ */
+@Slf4j
+public abstract class AbstractPrivateChannelHandler implements GateChannelHandler {
+
+    private static final char[] HEX_ARRAY = "0123456789abcdef".toCharArray();
+
+    private final String channelName;
+    private final String apiKey;
+    private final String apiSecret;
+    private final String contract;
+    private final GateGridTradeService gridTradeService;
+
+    public AbstractPrivateChannelHandler(String channelName,
+                                          String apiKey, String apiSecret,
+                                          String contract,
+                                          GateGridTradeService gridTradeService) {
+        this.channelName = channelName;
+        this.apiKey = apiKey;
+        this.apiSecret = apiSecret;
+        this.contract = contract;
+        this.gridTradeService = gridTradeService;
+    }
+
+    @Override
+    public String getChannelName() { return channelName; }
+
+    /**
+     * 发送带签名的订阅请求。
+     * payload: [userId, contract],auth: {method:"api_key", KEY, SIGN}
+     */
+    @Override
+    public void subscribe(WebSocketClient ws) {
+        long timeSec = System.currentTimeMillis() / 1000;
+        JSONObject msg = buildAuthRequest("subscribe", buildUid(), timeSec);
+        ws.send(msg.toJSONString());
+        log.info("[{}] 订阅成功, 合约:{}", channelName, contract);
+    }
+
+    /**
+     * 发送带签名的取消订阅请求,与 subscribe 对称。
+     */
+    @Override
+    public void unsubscribe(WebSocketClient ws) {
+        long timeSec = System.currentTimeMillis() / 1000;
+        JSONObject msg = new JSONObject();
+        msg.put("id", timeSec * 1000000 + (System.currentTimeMillis() % 1000));
+        msg.put("time", timeSec);
+        msg.put("channel", channelName);
+        msg.put("event", "unsubscribe");
+        JSONArray payload = new JSONArray();
+        payload.add(contract);
+        msg.put("payload", payload);
+        JSONObject auth = new JSONObject();
+        auth.put("method", "api_key");
+        auth.put("KEY", apiKey);
+        auth.put("SIGN", hs512Sign("unsubscribe", timeSec));
+        msg.put("auth", auth);
+        ws.send(msg.toJSONString());
+        log.info("[{}] 取消订阅成功, 合约:{}", channelName, contract);
+    }
+
+    protected GateGridTradeService getGridTradeService() { return gridTradeService; }
+    protected String getContract() { return contract; }
+
+    /**
+     * 从策略服务获取用户 ID,用于私有频道订阅的 payload[0]。
+     */
+    private String buildUid() {
+        return gridTradeService != null && gridTradeService.getUserId() != null
+                ? String.valueOf(gridTradeService.getUserId()) : "";
+    }
+
+    /**
+     * 构建认证请求 JSON。包含 id、time、channel、event、payload[auth_user_id, contract]、auth 字段。
+     */
+    private JSONObject buildAuthRequest(String event, String uid, long timeSec) {
+        JSONObject msg = new JSONObject();
+        msg.put("id", timeSec * 1000000 + (System.currentTimeMillis() % 1000));
+        msg.put("time", timeSec);
+        msg.put("channel", channelName);
+        msg.put("event", event);
+        JSONArray payload = new JSONArray();
+        payload.add(uid);
+        payload.add(contract);
+        msg.put("payload", payload);
+        JSONObject auth = new JSONObject();
+        auth.put("method", "api_key");
+        auth.put("KEY", apiKey);
+        auth.put("SIGN", hs512Sign(event, timeSec));
+        msg.put("auth", auth);
+        return msg;
+    }
+
+    /**
+     * HMAC-SHA512 签名,使用 UTF-8 编码。
+     */
+    private String hs512Sign(String event, long timeSec) {
+        try {
+            String message = "channel=" + channelName + "&event=" + event + "&time=" + timeSec;
+            Mac mac = Mac.getInstance("HmacSHA512");
+            SecretKeySpec spec = new SecretKeySpec(apiSecret.getBytes(StandardCharsets.UTF_8), "HmacSHA512");
+            mac.init(spec);
+            byte[] hash = mac.doFinal(message.getBytes(StandardCharsets.UTF_8));
+            StringBuilder hex = new StringBuilder(hash.length * 2);
+            for (byte b : hash) {
+                hex.append(HEX_ARRAY[(b >> 4) & 0xF]);
+                hex.append(HEX_ARRAY[b & 0xF]);
+            }
+            return hex.toString();
+        } catch (Exception e) {
+            log.error("[{}] 签名计算失败", channelName, e);
+            return "";
+        }
+    }
+}

--
Gitblit v1.9.1