mirror of
https://github.com/laoxong/nofx.git
synced 2026-06-04 09:58:22 +08:00
f4ece051e7
* refactor: 简化交易动作,移除 update_stop_loss/update_take_profit/partial_close - 移除 Decision 结构体中的 NewStopLoss, NewTakeProfit, ClosePercentage 字段 - 删除 executeUpdateStopLossWithRecord, executeUpdateTakeProfitWithRecord, executePartialCloseWithRecord 函数 - 简化 logger 中的 partial_close 聚合逻辑 - 更新 AI prompt 和验证逻辑,只保留 6 个核心动作 - 清理相关测试代码 保留的交易动作: open_long, open_short, close_long, close_short, hold, wait * refactor: 移除 AI学习与反思 模块 - 删除前端 AILearning.tsx 组件和相关引用 - 删除后端 /performance API 接口 - 删除 logger 中 AnalyzePerformance、calculateSharpeRatio 等函数 - 删除 PerformanceAnalysis、TradeOutcome、SymbolPerformance 等结构体 - 删除 Context 中的 Performance 字段 - 移除 AI prompt 中夏普比率自我进化相关内容 - 清理 i18n 翻译文件中的相关条目 该模块基于磁盘存储计算,经常出错,做减法移除 * refactor: 将数据库操作统一迁移到 store 包 - 新增 store/ 包,统一管理所有数据库操作 - store.go: 主 Store 结构,懒加载各子模块 - user.go, ai_model.go, exchange.go, trader.go 等子模块 - 支持加密/解密函数注入 (SetCryptoFuncs) - 更新 main.go 使用 store.New() 替代 config.NewDatabase() - 更新 api/server.go 使用 *store.Store 替代 *config.Database - 更新 manager/trader_manager.go: - 新增 LoadTradersFromStore, LoadUserTradersFromStore 方法 - 删除旧版 LoadUserTraders, LoadTraderByID, loadSingleTrader 等方法 - 移除 nofx/config 依赖 - 删除 config/database.go 和 config/database_test.go - 更新 api/server_test.go 使用 store.Trader 类型 - 清理 logger/ 包中未使用的 telegram 相关代码 * refactor: unify encryption key management via .env - Remove redundant EncryptionManager and SecureStorage - Simplify CryptoService to load keys from environment variables only - RSA_PRIVATE_KEY: RSA private key for client-server encryption - DATA_ENCRYPTION_KEY: AES-256 key for database encryption - JWT_SECRET: JWT signing key for authentication - Update start.sh to auto-generate missing keys on first run - Remove secrets/ directory and file-based key storage - Delete obsolete encryption setup scripts - Update .env.example with all required keys * refactor: unify logger usage across mcp package - Add MCPLogger adapter in logger package to implement mcp.Logger interface - Update mcp/config.go to use global logger by default - Remove redundant defaultLogger from mcp/logger.go - Keep noopLogger for testing purposes * chore: remove leftover test RSA key file * chore: remove unused bootstrap package * refactor: unify logging to use logger package instead of fmt/log - Replace all fmt.Print/log.Print calls with logger package - Add auto-initialization in logger package init() for test compatibility - Update main.go to initialize logger at startup - Migrate all packages: api, backtest, config, decision, manager, market, store, trader * refactor: rename database file from config.db to data.db - Update main.go, start.sh, docker-compose.yml - Update migration script and documentation - Update .gitignore and translations * fix: add RSA_PRIVATE_KEY to docker-compose environment * fix: add registration_enabled to /api/config response * fix: Fix navigation between login and register pages Use window.location.href instead of react-router's navigate() to fix the issue where URL changes but the page doesn't reload due to App.tsx using custom route state management. * fix: Switch SQLite from WAL to DELETE mode for Docker compatibility WAL mode causes data sync issues with Docker bind mounts on macOS due to incompatible file locking mechanisms between the container and host. DELETE mode (traditional journaling) ensures data is written directly to the main database file. * refactor: Remove default user from database initialization The default user was a legacy placeholder that is no longer needed now that proper user registration is in place. * feat: Add order tracking system with centralized status sync - Add trader_orders table for tracking all order lifecycle - Implement GetOrderStatus interface for all exchanges (Binance, Bybit, Hyperliquid, Aster, Lighter) - Create OrderSyncManager for centralized order status polling - Add trading statistics (Sharpe ratio, win rate, profit factor) to AI context - Include recent completed orders in AI decision input - Remove per-order goroutine polling in favor of global sync manager * feat: Add TradingView K-line chart to dashboard - Create TradingViewChart component with exchange/symbol selectors - Support Binance, Bybit, OKX, Coinbase, Kraken, KuCoin exchanges - Add popular symbols quick selection - Support multiple timeframes (1m to 1W) - Add fullscreen mode - Integrate with Dashboard page below equity chart - Add i18n translations for zh/en * refactor: Replace separate charts with tabbed ChartTabs component - Create ChartTabs component with tab switching between equity curve and K-line - Add embedded mode support for EquityChart and TradingViewChart - User can now switch between account equity and market chart in same area * fix: Use ChartTabs in App.tsx and fix embedded mode in EquityChart - Replace EquityChart with ChartTabs in App.tsx (the actual dashboard renderer) - Fix EquityChart embedded mode for error and empty data states - Rename interval state to timeInterval to avoid shadowing window.setInterval - Add debug logging to ChartTabs component * feat: Add position tracking system for accurate trade history - Add trader_positions table to track complete open/close trades - Add PositionSyncManager to detect manual closes via polling - Record position on open, update on close with PnL calculation - Use positions table for trading stats and recent trades (replacing orders table) - Fix TradingView chart symbol format (add .P suffix for futures) - Fix DecisionCard wait/hold action color (gray instead of red) - Auto-append USDT suffix for custom symbol input * update ---------
959 lines
28 KiB
Go
959 lines
28 KiB
Go
package trader
|
||
|
||
import (
|
||
"context"
|
||
"crypto/rand"
|
||
"encoding/hex"
|
||
"fmt"
|
||
"nofx/hook"
|
||
"nofx/logger"
|
||
"strconv"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
|
||
"github.com/adshao/go-binance/v2/futures"
|
||
)
|
||
|
||
// getBrOrderID 生成唯一订单ID(合约专用)
|
||
// 格式: x-{BR_ID}{TIMESTAMP}{RANDOM}
|
||
// 合约限制32字符,统一使用此限制以保持一致性
|
||
// 使用纳秒时间戳+随机数确保全局唯一性(冲突概率 < 10^-20)
|
||
func getBrOrderID() string {
|
||
brID := "KzrpZaP9" // 合约br ID
|
||
|
||
// 计算可用空间: 32 - len("x-KzrpZaP9") = 32 - 11 = 21字符
|
||
// 分配: 13位时间戳 + 8位随机数 = 21字符(完美利用)
|
||
timestamp := time.Now().UnixNano() % 10000000000000 // 13位纳秒时间戳
|
||
|
||
// 生成4字节随机数(8位十六进制)
|
||
randomBytes := make([]byte, 4)
|
||
rand.Read(randomBytes)
|
||
randomHex := hex.EncodeToString(randomBytes)
|
||
|
||
// 格式: x-KzrpZaP9{13位时间戳}{8位随机}
|
||
// 示例: x-KzrpZaP91234567890123abcdef12 (正好31字符)
|
||
orderID := fmt.Sprintf("x-%s%d%s", brID, timestamp, randomHex)
|
||
|
||
// 确保不超过32字符限制(理论上正好31字符)
|
||
if len(orderID) > 32 {
|
||
orderID = orderID[:32]
|
||
}
|
||
|
||
return orderID
|
||
}
|
||
|
||
// FuturesTrader 币安合约交易器
|
||
type FuturesTrader struct {
|
||
client *futures.Client
|
||
|
||
// 余额缓存
|
||
cachedBalance map[string]interface{}
|
||
balanceCacheTime time.Time
|
||
balanceCacheMutex sync.RWMutex
|
||
|
||
// 持仓缓存
|
||
cachedPositions []map[string]interface{}
|
||
positionsCacheTime time.Time
|
||
positionsCacheMutex sync.RWMutex
|
||
|
||
// 缓存有效期(15秒)
|
||
cacheDuration time.Duration
|
||
}
|
||
|
||
// NewFuturesTrader 创建合约交易器
|
||
func NewFuturesTrader(apiKey, secretKey string, userId string) *FuturesTrader {
|
||
client := futures.NewClient(apiKey, secretKey)
|
||
|
||
hookRes := hook.HookExec[hook.NewBinanceTraderResult](hook.NEW_BINANCE_TRADER, userId, client)
|
||
if hookRes != nil && hookRes.GetResult() != nil {
|
||
client = hookRes.GetResult()
|
||
}
|
||
|
||
// 同步时间,避免 Timestamp ahead 错误
|
||
syncBinanceServerTime(client)
|
||
trader := &FuturesTrader{
|
||
client: client,
|
||
cacheDuration: 15 * time.Second, // 15秒缓存
|
||
}
|
||
|
||
// 设置双向持仓模式(Hedge Mode)
|
||
// 这是必需的,因为代码中使用了 PositionSide (LONG/SHORT)
|
||
if err := trader.setDualSidePosition(); err != nil {
|
||
logger.Infof("⚠️ 设置双向持仓模式失败: %v (如果已是双向模式则忽略此警告)", err)
|
||
}
|
||
|
||
return trader
|
||
}
|
||
|
||
// setDualSidePosition 设置双向持仓模式(初始化时调用)
|
||
func (t *FuturesTrader) setDualSidePosition() error {
|
||
// 尝试设置双向持仓模式
|
||
err := t.client.NewChangePositionModeService().
|
||
DualSide(true). // true = 双向持仓(Hedge Mode)
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
// 如果错误信息包含"No need to change",说明已经是双向持仓模式
|
||
if strings.Contains(err.Error(), "No need to change position side") {
|
||
logger.Infof(" ✓ 账户已是双向持仓模式(Hedge Mode)")
|
||
return nil
|
||
}
|
||
// 其他错误则返回(但在调用方不会中断初始化)
|
||
return err
|
||
}
|
||
|
||
logger.Infof(" ✓ 账户已切换为双向持仓模式(Hedge Mode)")
|
||
logger.Infof(" ℹ️ 双向持仓模式允许同时持有多单和空单")
|
||
return nil
|
||
}
|
||
|
||
// syncBinanceServerTime 同步币安服务器时间,确保请求时间戳合法
|
||
func syncBinanceServerTime(client *futures.Client) {
|
||
serverTime, err := client.NewServerTimeService().Do(context.Background())
|
||
if err != nil {
|
||
logger.Infof("⚠️ 同步币安服务器时间失败: %v", err)
|
||
return
|
||
}
|
||
|
||
now := time.Now().UnixMilli()
|
||
offset := now - serverTime
|
||
client.TimeOffset = offset
|
||
logger.Infof("⏱ 已同步币安服务器时间,偏移 %dms", offset)
|
||
}
|
||
|
||
// GetBalance 获取账户余额(带缓存)
|
||
func (t *FuturesTrader) GetBalance() (map[string]interface{}, error) {
|
||
// 先检查缓存是否有效
|
||
t.balanceCacheMutex.RLock()
|
||
if t.cachedBalance != nil && time.Since(t.balanceCacheTime) < t.cacheDuration {
|
||
cacheAge := time.Since(t.balanceCacheTime)
|
||
t.balanceCacheMutex.RUnlock()
|
||
logger.Infof("✓ 使用缓存的账户余额(缓存时间: %.1f秒前)", cacheAge.Seconds())
|
||
return t.cachedBalance, nil
|
||
}
|
||
t.balanceCacheMutex.RUnlock()
|
||
|
||
// 缓存过期或不存在,调用API
|
||
logger.Infof("🔄 缓存过期,正在调用币安API获取账户余额...")
|
||
account, err := t.client.NewGetAccountService().Do(context.Background())
|
||
if err != nil {
|
||
logger.Infof("❌ 币安API调用失败: %v", err)
|
||
return nil, fmt.Errorf("获取账户信息失败: %w", err)
|
||
}
|
||
|
||
result := make(map[string]interface{})
|
||
result["totalWalletBalance"], _ = strconv.ParseFloat(account.TotalWalletBalance, 64)
|
||
result["availableBalance"], _ = strconv.ParseFloat(account.AvailableBalance, 64)
|
||
result["totalUnrealizedProfit"], _ = strconv.ParseFloat(account.TotalUnrealizedProfit, 64)
|
||
|
||
logger.Infof("✓ 币安API返回: 总余额=%s, 可用=%s, 未实现盈亏=%s",
|
||
account.TotalWalletBalance,
|
||
account.AvailableBalance,
|
||
account.TotalUnrealizedProfit)
|
||
|
||
// 更新缓存
|
||
t.balanceCacheMutex.Lock()
|
||
t.cachedBalance = result
|
||
t.balanceCacheTime = time.Now()
|
||
t.balanceCacheMutex.Unlock()
|
||
|
||
return result, nil
|
||
}
|
||
|
||
// GetPositions 获取所有持仓(带缓存)
|
||
func (t *FuturesTrader) GetPositions() ([]map[string]interface{}, error) {
|
||
// 先检查缓存是否有效
|
||
t.positionsCacheMutex.RLock()
|
||
if t.cachedPositions != nil && time.Since(t.positionsCacheTime) < t.cacheDuration {
|
||
cacheAge := time.Since(t.positionsCacheTime)
|
||
t.positionsCacheMutex.RUnlock()
|
||
logger.Infof("✓ 使用缓存的持仓信息(缓存时间: %.1f秒前)", cacheAge.Seconds())
|
||
return t.cachedPositions, nil
|
||
}
|
||
t.positionsCacheMutex.RUnlock()
|
||
|
||
// 缓存过期或不存在,调用API
|
||
logger.Infof("🔄 缓存过期,正在调用币安API获取持仓信息...")
|
||
positions, err := t.client.NewGetPositionRiskService().Do(context.Background())
|
||
if err != nil {
|
||
return nil, fmt.Errorf("获取持仓失败: %w", err)
|
||
}
|
||
|
||
var result []map[string]interface{}
|
||
for _, pos := range positions {
|
||
posAmt, _ := strconv.ParseFloat(pos.PositionAmt, 64)
|
||
if posAmt == 0 {
|
||
continue // 跳过无持仓的
|
||
}
|
||
|
||
posMap := make(map[string]interface{})
|
||
posMap["symbol"] = pos.Symbol
|
||
posMap["positionAmt"], _ = strconv.ParseFloat(pos.PositionAmt, 64)
|
||
posMap["entryPrice"], _ = strconv.ParseFloat(pos.EntryPrice, 64)
|
||
posMap["markPrice"], _ = strconv.ParseFloat(pos.MarkPrice, 64)
|
||
posMap["unRealizedProfit"], _ = strconv.ParseFloat(pos.UnRealizedProfit, 64)
|
||
posMap["leverage"], _ = strconv.ParseFloat(pos.Leverage, 64)
|
||
posMap["liquidationPrice"], _ = strconv.ParseFloat(pos.LiquidationPrice, 64)
|
||
|
||
// 判断方向
|
||
if posAmt > 0 {
|
||
posMap["side"] = "long"
|
||
} else {
|
||
posMap["side"] = "short"
|
||
}
|
||
|
||
result = append(result, posMap)
|
||
}
|
||
|
||
// 更新缓存
|
||
t.positionsCacheMutex.Lock()
|
||
t.cachedPositions = result
|
||
t.positionsCacheTime = time.Now()
|
||
t.positionsCacheMutex.Unlock()
|
||
|
||
return result, nil
|
||
}
|
||
|
||
// SetMarginMode 设置仓位模式
|
||
func (t *FuturesTrader) SetMarginMode(symbol string, isCrossMargin bool) error {
|
||
var marginType futures.MarginType
|
||
if isCrossMargin {
|
||
marginType = futures.MarginTypeCrossed
|
||
} else {
|
||
marginType = futures.MarginTypeIsolated
|
||
}
|
||
|
||
// 尝试设置仓位模式
|
||
err := t.client.NewChangeMarginTypeService().
|
||
Symbol(symbol).
|
||
MarginType(marginType).
|
||
Do(context.Background())
|
||
|
||
marginModeStr := "全仓"
|
||
if !isCrossMargin {
|
||
marginModeStr = "逐仓"
|
||
}
|
||
|
||
if err != nil {
|
||
// 如果错误信息包含"No need to change",说明仓位模式已经是目标值
|
||
if contains(err.Error(), "No need to change margin type") {
|
||
logger.Infof(" ✓ %s 仓位模式已是 %s", symbol, marginModeStr)
|
||
return nil
|
||
}
|
||
// 如果有持仓,无法更改仓位模式,但不影响交易
|
||
if contains(err.Error(), "Margin type cannot be changed if there exists position") {
|
||
logger.Infof(" ⚠️ %s 有持仓,无法更改仓位模式,继续使用当前模式", symbol)
|
||
return nil
|
||
}
|
||
// 检测多资产模式(错误码 -4168)
|
||
if contains(err.Error(), "Multi-Assets mode") || contains(err.Error(), "-4168") || contains(err.Error(), "4168") {
|
||
logger.Infof(" ⚠️ %s 检测到多资产模式,强制使用全仓模式", symbol)
|
||
logger.Infof(" 💡 提示:如需使用逐仓模式,请在币安关闭多资产模式")
|
||
return nil
|
||
}
|
||
// 检测统一账户 API(Portfolio Margin)
|
||
if contains(err.Error(), "unified") || contains(err.Error(), "portfolio") || contains(err.Error(), "Portfolio") {
|
||
logger.Infof(" ❌ %s 检测到统一账户 API,无法进行合约交易", symbol)
|
||
return fmt.Errorf("请使用「现货与合约交易」API 权限,不要使用「统一账户 API」")
|
||
}
|
||
logger.Infof(" ⚠️ 设置仓位模式失败: %v", err)
|
||
// 不返回错误,让交易继续
|
||
return nil
|
||
}
|
||
|
||
logger.Infof(" ✓ %s 仓位模式已设置为 %s", symbol, marginModeStr)
|
||
return nil
|
||
}
|
||
|
||
// SetLeverage 设置杠杆(智能判断+冷却期)
|
||
func (t *FuturesTrader) SetLeverage(symbol string, leverage int) error {
|
||
// 先尝试获取当前杠杆(从持仓信息)
|
||
currentLeverage := 0
|
||
positions, err := t.GetPositions()
|
||
if err == nil {
|
||
for _, pos := range positions {
|
||
if pos["symbol"] == symbol {
|
||
if lev, ok := pos["leverage"].(float64); ok {
|
||
currentLeverage = int(lev)
|
||
break
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// 如果当前杠杆已经是目标杠杆,跳过
|
||
if currentLeverage == leverage && currentLeverage > 0 {
|
||
logger.Infof(" ✓ %s 杠杆已是 %dx,无需切换", symbol, leverage)
|
||
return nil
|
||
}
|
||
|
||
// 切换杠杆
|
||
_, err = t.client.NewChangeLeverageService().
|
||
Symbol(symbol).
|
||
Leverage(leverage).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
// 如果错误信息包含"No need to change",说明杠杆已经是目标值
|
||
if contains(err.Error(), "No need to change") {
|
||
logger.Infof(" ✓ %s 杠杆已是 %dx", symbol, leverage)
|
||
return nil
|
||
}
|
||
return fmt.Errorf("设置杠杆失败: %w", err)
|
||
}
|
||
|
||
logger.Infof(" ✓ %s 杠杆已切换为 %dx", symbol, leverage)
|
||
|
||
// 切换杠杆后等待5秒(避免冷却期错误)
|
||
logger.Infof(" ⏱ 等待5秒冷却期...")
|
||
time.Sleep(5 * time.Second)
|
||
|
||
return nil
|
||
}
|
||
|
||
// OpenLong 开多仓
|
||
func (t *FuturesTrader) OpenLong(symbol string, quantity float64, leverage int) (map[string]interface{}, error) {
|
||
// 先取消该币种的所有委托单(清理旧的止损止盈单)
|
||
if err := t.CancelAllOrders(symbol); err != nil {
|
||
logger.Infof(" ⚠ 取消旧委托单失败(可能没有委托单): %v", err)
|
||
}
|
||
|
||
// 设置杠杆
|
||
if err := t.SetLeverage(symbol, leverage); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 注意:仓位模式应该由调用方(AutoTrader)在开仓前通过 SetMarginMode 设置
|
||
|
||
// 格式化数量到正确精度
|
||
quantityStr, err := t.FormatQuantity(symbol, quantity)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// ✅ 检查格式化后的数量是否为 0(防止四舍五入导致的错误)
|
||
quantityFloat, parseErr := strconv.ParseFloat(quantityStr, 64)
|
||
if parseErr != nil || quantityFloat <= 0 {
|
||
return nil, fmt.Errorf("开仓数量过小,格式化后为 0 (原始: %.8f → 格式化: %s)。建议增加开仓金额或选择价格更低的币种", quantity, quantityStr)
|
||
}
|
||
|
||
// ✅ 检查最小名义价值(Binance 要求至少 10 USDT)
|
||
if err := t.CheckMinNotional(symbol, quantityFloat); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 创建市价买入订单(使用br ID)
|
||
order, err := t.client.NewCreateOrderService().
|
||
Symbol(symbol).
|
||
Side(futures.SideTypeBuy).
|
||
PositionSide(futures.PositionSideTypeLong).
|
||
Type(futures.OrderTypeMarket).
|
||
Quantity(quantityStr).
|
||
NewClientOrderID(getBrOrderID()).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
return nil, fmt.Errorf("开多仓失败: %w", err)
|
||
}
|
||
|
||
logger.Infof("✓ 开多仓成功: %s 数量: %s", symbol, quantityStr)
|
||
logger.Infof(" 订单ID: %d", order.OrderID)
|
||
|
||
result := make(map[string]interface{})
|
||
result["orderId"] = order.OrderID
|
||
result["symbol"] = order.Symbol
|
||
result["status"] = order.Status
|
||
return result, nil
|
||
}
|
||
|
||
// OpenShort 开空仓
|
||
func (t *FuturesTrader) OpenShort(symbol string, quantity float64, leverage int) (map[string]interface{}, error) {
|
||
// 先取消该币种的所有委托单(清理旧的止损止盈单)
|
||
if err := t.CancelAllOrders(symbol); err != nil {
|
||
logger.Infof(" ⚠ 取消旧委托单失败(可能没有委托单): %v", err)
|
||
}
|
||
|
||
// 设置杠杆
|
||
if err := t.SetLeverage(symbol, leverage); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 注意:仓位模式应该由调用方(AutoTrader)在开仓前通过 SetMarginMode 设置
|
||
|
||
// 格式化数量到正确精度
|
||
quantityStr, err := t.FormatQuantity(symbol, quantity)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// ✅ 检查格式化后的数量是否为 0(防止四舍五入导致的错误)
|
||
quantityFloat, parseErr := strconv.ParseFloat(quantityStr, 64)
|
||
if parseErr != nil || quantityFloat <= 0 {
|
||
return nil, fmt.Errorf("开仓数量过小,格式化后为 0 (原始: %.8f → 格式化: %s)。建议增加开仓金额或选择价格更低的币种", quantity, quantityStr)
|
||
}
|
||
|
||
// ✅ 检查最小名义价值(Binance 要求至少 10 USDT)
|
||
if err := t.CheckMinNotional(symbol, quantityFloat); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 创建市价卖出订单(使用br ID)
|
||
order, err := t.client.NewCreateOrderService().
|
||
Symbol(symbol).
|
||
Side(futures.SideTypeSell).
|
||
PositionSide(futures.PositionSideTypeShort).
|
||
Type(futures.OrderTypeMarket).
|
||
Quantity(quantityStr).
|
||
NewClientOrderID(getBrOrderID()).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
return nil, fmt.Errorf("开空仓失败: %w", err)
|
||
}
|
||
|
||
logger.Infof("✓ 开空仓成功: %s 数量: %s", symbol, quantityStr)
|
||
logger.Infof(" 订单ID: %d", order.OrderID)
|
||
|
||
result := make(map[string]interface{})
|
||
result["orderId"] = order.OrderID
|
||
result["symbol"] = order.Symbol
|
||
result["status"] = order.Status
|
||
return result, nil
|
||
}
|
||
|
||
// CloseLong 平多仓
|
||
func (t *FuturesTrader) CloseLong(symbol string, quantity float64) (map[string]interface{}, error) {
|
||
// 如果数量为0,获取当前持仓数量
|
||
if quantity == 0 {
|
||
positions, err := t.GetPositions()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
for _, pos := range positions {
|
||
if pos["symbol"] == symbol && pos["side"] == "long" {
|
||
quantity = pos["positionAmt"].(float64)
|
||
break
|
||
}
|
||
}
|
||
|
||
if quantity == 0 {
|
||
return nil, fmt.Errorf("没有找到 %s 的多仓", symbol)
|
||
}
|
||
}
|
||
|
||
// 格式化数量
|
||
quantityStr, err := t.FormatQuantity(symbol, quantity)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 创建市价卖出订单(平多,使用br ID)
|
||
order, err := t.client.NewCreateOrderService().
|
||
Symbol(symbol).
|
||
Side(futures.SideTypeSell).
|
||
PositionSide(futures.PositionSideTypeLong).
|
||
Type(futures.OrderTypeMarket).
|
||
Quantity(quantityStr).
|
||
NewClientOrderID(getBrOrderID()).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
return nil, fmt.Errorf("平多仓失败: %w", err)
|
||
}
|
||
|
||
logger.Infof("✓ 平多仓成功: %s 数量: %s", symbol, quantityStr)
|
||
|
||
// 平仓后取消该币种的所有挂单(止损止盈单)
|
||
if err := t.CancelAllOrders(symbol); err != nil {
|
||
logger.Infof(" ⚠ 取消挂单失败: %v", err)
|
||
}
|
||
|
||
result := make(map[string]interface{})
|
||
result["orderId"] = order.OrderID
|
||
result["symbol"] = order.Symbol
|
||
result["status"] = order.Status
|
||
return result, nil
|
||
}
|
||
|
||
// CloseShort 平空仓
|
||
func (t *FuturesTrader) CloseShort(symbol string, quantity float64) (map[string]interface{}, error) {
|
||
// 如果数量为0,获取当前持仓数量
|
||
if quantity == 0 {
|
||
positions, err := t.GetPositions()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
for _, pos := range positions {
|
||
if pos["symbol"] == symbol && pos["side"] == "short" {
|
||
quantity = -pos["positionAmt"].(float64) // 空仓数量是负的,取绝对值
|
||
break
|
||
}
|
||
}
|
||
|
||
if quantity == 0 {
|
||
return nil, fmt.Errorf("没有找到 %s 的空仓", symbol)
|
||
}
|
||
}
|
||
|
||
// 格式化数量
|
||
quantityStr, err := t.FormatQuantity(symbol, quantity)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 创建市价买入订单(平空,使用br ID)
|
||
order, err := t.client.NewCreateOrderService().
|
||
Symbol(symbol).
|
||
Side(futures.SideTypeBuy).
|
||
PositionSide(futures.PositionSideTypeShort).
|
||
Type(futures.OrderTypeMarket).
|
||
Quantity(quantityStr).
|
||
NewClientOrderID(getBrOrderID()).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
return nil, fmt.Errorf("平空仓失败: %w", err)
|
||
}
|
||
|
||
logger.Infof("✓ 平空仓成功: %s 数量: %s", symbol, quantityStr)
|
||
|
||
// 平仓后取消该币种的所有挂单(止损止盈单)
|
||
if err := t.CancelAllOrders(symbol); err != nil {
|
||
logger.Infof(" ⚠ 取消挂单失败: %v", err)
|
||
}
|
||
|
||
result := make(map[string]interface{})
|
||
result["orderId"] = order.OrderID
|
||
result["symbol"] = order.Symbol
|
||
result["status"] = order.Status
|
||
return result, nil
|
||
}
|
||
|
||
// CancelStopLossOrders 仅取消止损单(不影响止盈单)
|
||
func (t *FuturesTrader) CancelStopLossOrders(symbol string) error {
|
||
// 获取该币种的所有未完成订单
|
||
orders, err := t.client.NewListOpenOrdersService().
|
||
Symbol(symbol).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
return fmt.Errorf("获取未完成订单失败: %w", err)
|
||
}
|
||
|
||
// 过滤出止损单并取消(取消所有方向的止损单,包括LONG和SHORT)
|
||
canceledCount := 0
|
||
var cancelErrors []error
|
||
for _, order := range orders {
|
||
orderType := order.Type
|
||
|
||
// 只取消止损订单(不取消止盈订单)
|
||
if orderType == futures.OrderTypeStopMarket || orderType == futures.OrderTypeStop {
|
||
_, err := t.client.NewCancelOrderService().
|
||
Symbol(symbol).
|
||
OrderID(order.OrderID).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
errMsg := fmt.Sprintf("订单ID %d: %v", order.OrderID, err)
|
||
cancelErrors = append(cancelErrors, fmt.Errorf("%s", errMsg))
|
||
logger.Infof(" ⚠ 取消止损单失败: %s", errMsg)
|
||
continue
|
||
}
|
||
|
||
canceledCount++
|
||
logger.Infof(" ✓ 已取消止损单 (订单ID: %d, 类型: %s, 方向: %s)", order.OrderID, orderType, order.PositionSide)
|
||
}
|
||
}
|
||
|
||
if canceledCount == 0 && len(cancelErrors) == 0 {
|
||
logger.Infof(" ℹ %s 没有止损单需要取消", symbol)
|
||
} else if canceledCount > 0 {
|
||
logger.Infof(" ✓ 已取消 %s 的 %d 个止损单", symbol, canceledCount)
|
||
}
|
||
|
||
// 如果所有取消都失败了,返回错误
|
||
if len(cancelErrors) > 0 && canceledCount == 0 {
|
||
return fmt.Errorf("取消止损单失败: %v", cancelErrors)
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// CancelTakeProfitOrders 仅取消止盈单(不影响止损单)
|
||
func (t *FuturesTrader) CancelTakeProfitOrders(symbol string) error {
|
||
// 获取该币种的所有未完成订单
|
||
orders, err := t.client.NewListOpenOrdersService().
|
||
Symbol(symbol).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
return fmt.Errorf("获取未完成订单失败: %w", err)
|
||
}
|
||
|
||
// 过滤出止盈单并取消(取消所有方向的止盈单,包括LONG和SHORT)
|
||
canceledCount := 0
|
||
var cancelErrors []error
|
||
for _, order := range orders {
|
||
orderType := order.Type
|
||
|
||
// 只取消止盈订单(不取消止损订单)
|
||
if orderType == futures.OrderTypeTakeProfitMarket || orderType == futures.OrderTypeTakeProfit {
|
||
_, err := t.client.NewCancelOrderService().
|
||
Symbol(symbol).
|
||
OrderID(order.OrderID).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
errMsg := fmt.Sprintf("订单ID %d: %v", order.OrderID, err)
|
||
cancelErrors = append(cancelErrors, fmt.Errorf("%s", errMsg))
|
||
logger.Infof(" ⚠ 取消止盈单失败: %s", errMsg)
|
||
continue
|
||
}
|
||
|
||
canceledCount++
|
||
logger.Infof(" ✓ 已取消止盈单 (订单ID: %d, 类型: %s, 方向: %s)", order.OrderID, orderType, order.PositionSide)
|
||
}
|
||
}
|
||
|
||
if canceledCount == 0 && len(cancelErrors) == 0 {
|
||
logger.Infof(" ℹ %s 没有止盈单需要取消", symbol)
|
||
} else if canceledCount > 0 {
|
||
logger.Infof(" ✓ 已取消 %s 的 %d 个止盈单", symbol, canceledCount)
|
||
}
|
||
|
||
// 如果所有取消都失败了,返回错误
|
||
if len(cancelErrors) > 0 && canceledCount == 0 {
|
||
return fmt.Errorf("取消止盈单失败: %v", cancelErrors)
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// CancelAllOrders 取消该币种的所有挂单
|
||
func (t *FuturesTrader) CancelAllOrders(symbol string) error {
|
||
err := t.client.NewCancelAllOpenOrdersService().
|
||
Symbol(symbol).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
return fmt.Errorf("取消挂单失败: %w", err)
|
||
}
|
||
|
||
logger.Infof(" ✓ 已取消 %s 的所有挂单", symbol)
|
||
return nil
|
||
}
|
||
|
||
// CancelStopOrders 取消该币种的止盈/止损单(用于调整止盈止损位置)
|
||
func (t *FuturesTrader) CancelStopOrders(symbol string) error {
|
||
// 获取该币种的所有未完成订单
|
||
orders, err := t.client.NewListOpenOrdersService().
|
||
Symbol(symbol).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
return fmt.Errorf("获取未完成订单失败: %w", err)
|
||
}
|
||
|
||
// 过滤出止盈止损单并取消
|
||
canceledCount := 0
|
||
for _, order := range orders {
|
||
orderType := order.Type
|
||
|
||
// 只取消止损和止盈订单
|
||
if orderType == futures.OrderTypeStopMarket ||
|
||
orderType == futures.OrderTypeTakeProfitMarket ||
|
||
orderType == futures.OrderTypeStop ||
|
||
orderType == futures.OrderTypeTakeProfit {
|
||
|
||
_, err := t.client.NewCancelOrderService().
|
||
Symbol(symbol).
|
||
OrderID(order.OrderID).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
logger.Infof(" ⚠ 取消订单 %d 失败: %v", order.OrderID, err)
|
||
continue
|
||
}
|
||
|
||
canceledCount++
|
||
logger.Infof(" ✓ 已取消 %s 的止盈/止损单 (订单ID: %d, 类型: %s)",
|
||
symbol, order.OrderID, orderType)
|
||
}
|
||
}
|
||
|
||
if canceledCount == 0 {
|
||
logger.Infof(" ℹ %s 没有止盈/止损单需要取消", symbol)
|
||
} else {
|
||
logger.Infof(" ✓ 已取消 %s 的 %d 个止盈/止损单", symbol, canceledCount)
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// GetMarketPrice 获取市场价格
|
||
func (t *FuturesTrader) GetMarketPrice(symbol string) (float64, error) {
|
||
prices, err := t.client.NewListPricesService().Symbol(symbol).Do(context.Background())
|
||
if err != nil {
|
||
return 0, fmt.Errorf("获取价格失败: %w", err)
|
||
}
|
||
|
||
if len(prices) == 0 {
|
||
return 0, fmt.Errorf("未找到价格")
|
||
}
|
||
|
||
price, err := strconv.ParseFloat(prices[0].Price, 64)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
|
||
return price, nil
|
||
}
|
||
|
||
// CalculatePositionSize 计算仓位大小
|
||
func (t *FuturesTrader) CalculatePositionSize(balance, riskPercent, price float64, leverage int) float64 {
|
||
riskAmount := balance * (riskPercent / 100.0)
|
||
positionValue := riskAmount * float64(leverage)
|
||
quantity := positionValue / price
|
||
return quantity
|
||
}
|
||
|
||
// SetStopLoss 设置止损单
|
||
func (t *FuturesTrader) SetStopLoss(symbol string, positionSide string, quantity, stopPrice float64) error {
|
||
var side futures.SideType
|
||
var posSide futures.PositionSideType
|
||
|
||
if positionSide == "LONG" {
|
||
side = futures.SideTypeSell
|
||
posSide = futures.PositionSideTypeLong
|
||
} else {
|
||
side = futures.SideTypeBuy
|
||
posSide = futures.PositionSideTypeShort
|
||
}
|
||
|
||
// 格式化数量
|
||
quantityStr, err := t.FormatQuantity(symbol, quantity)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
_, err = t.client.NewCreateOrderService().
|
||
Symbol(symbol).
|
||
Side(side).
|
||
PositionSide(posSide).
|
||
Type(futures.OrderTypeStopMarket).
|
||
StopPrice(fmt.Sprintf("%.8f", stopPrice)).
|
||
Quantity(quantityStr).
|
||
WorkingType(futures.WorkingTypeContractPrice).
|
||
ClosePosition(true).
|
||
NewClientOrderID(getBrOrderID()).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
return fmt.Errorf("设置止损失败: %w", err)
|
||
}
|
||
|
||
logger.Infof(" 止损价设置: %.4f", stopPrice)
|
||
return nil
|
||
}
|
||
|
||
// SetTakeProfit 设置止盈单
|
||
func (t *FuturesTrader) SetTakeProfit(symbol string, positionSide string, quantity, takeProfitPrice float64) error {
|
||
var side futures.SideType
|
||
var posSide futures.PositionSideType
|
||
|
||
if positionSide == "LONG" {
|
||
side = futures.SideTypeSell
|
||
posSide = futures.PositionSideTypeLong
|
||
} else {
|
||
side = futures.SideTypeBuy
|
||
posSide = futures.PositionSideTypeShort
|
||
}
|
||
|
||
// 格式化数量
|
||
quantityStr, err := t.FormatQuantity(symbol, quantity)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
_, err = t.client.NewCreateOrderService().
|
||
Symbol(symbol).
|
||
Side(side).
|
||
PositionSide(posSide).
|
||
Type(futures.OrderTypeTakeProfitMarket).
|
||
StopPrice(fmt.Sprintf("%.8f", takeProfitPrice)).
|
||
Quantity(quantityStr).
|
||
WorkingType(futures.WorkingTypeContractPrice).
|
||
ClosePosition(true).
|
||
NewClientOrderID(getBrOrderID()).
|
||
Do(context.Background())
|
||
|
||
if err != nil {
|
||
return fmt.Errorf("设置止盈失败: %w", err)
|
||
}
|
||
|
||
logger.Infof(" 止盈价设置: %.4f", takeProfitPrice)
|
||
return nil
|
||
}
|
||
|
||
// GetMinNotional 获取最小名义价值(Binance要求)
|
||
func (t *FuturesTrader) GetMinNotional(symbol string) float64 {
|
||
// 使用保守的默认值 10 USDT,确保订单能够通过交易所验证
|
||
return 10.0
|
||
}
|
||
|
||
// CheckMinNotional 检查订单是否满足最小名义价值要求
|
||
func (t *FuturesTrader) CheckMinNotional(symbol string, quantity float64) error {
|
||
price, err := t.GetMarketPrice(symbol)
|
||
if err != nil {
|
||
return fmt.Errorf("获取市价失败: %w", err)
|
||
}
|
||
|
||
notionalValue := quantity * price
|
||
minNotional := t.GetMinNotional(symbol)
|
||
|
||
if notionalValue < minNotional {
|
||
return fmt.Errorf(
|
||
"订单金额 %.2f USDT 低于最小要求 %.2f USDT (数量: %.4f, 价格: %.4f)",
|
||
notionalValue, minNotional, quantity, price,
|
||
)
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// GetSymbolPrecision 获取交易对的数量精度
|
||
func (t *FuturesTrader) GetSymbolPrecision(symbol string) (int, error) {
|
||
exchangeInfo, err := t.client.NewExchangeInfoService().Do(context.Background())
|
||
if err != nil {
|
||
return 0, fmt.Errorf("获取交易规则失败: %w", err)
|
||
}
|
||
|
||
for _, s := range exchangeInfo.Symbols {
|
||
if s.Symbol == symbol {
|
||
// 从LOT_SIZE filter获取精度
|
||
for _, filter := range s.Filters {
|
||
if filter["filterType"] == "LOT_SIZE" {
|
||
stepSize := filter["stepSize"].(string)
|
||
precision := calculatePrecision(stepSize)
|
||
logger.Infof(" %s 数量精度: %d (stepSize: %s)", symbol, precision, stepSize)
|
||
return precision, nil
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
logger.Infof(" ⚠ %s 未找到精度信息,使用默认精度3", symbol)
|
||
return 3, nil // 默认精度为3
|
||
}
|
||
|
||
// calculatePrecision 从stepSize计算精度
|
||
func calculatePrecision(stepSize string) int {
|
||
// 去除尾部的0
|
||
stepSize = trimTrailingZeros(stepSize)
|
||
|
||
// 查找小数点
|
||
dotIndex := -1
|
||
for i := 0; i < len(stepSize); i++ {
|
||
if stepSize[i] == '.' {
|
||
dotIndex = i
|
||
break
|
||
}
|
||
}
|
||
|
||
// 如果没有小数点或小数点在最后,精度为0
|
||
if dotIndex == -1 || dotIndex == len(stepSize)-1 {
|
||
return 0
|
||
}
|
||
|
||
// 返回小数点后的位数
|
||
return len(stepSize) - dotIndex - 1
|
||
}
|
||
|
||
// trimTrailingZeros 去除尾部的0
|
||
func trimTrailingZeros(s string) string {
|
||
// 如果没有小数点,直接返回
|
||
if !stringContains(s, ".") {
|
||
return s
|
||
}
|
||
|
||
// 从后向前遍历,去除尾部的0
|
||
for len(s) > 0 && s[len(s)-1] == '0' {
|
||
s = s[:len(s)-1]
|
||
}
|
||
|
||
// 如果最后一位是小数点,也去掉
|
||
if len(s) > 0 && s[len(s)-1] == '.' {
|
||
s = s[:len(s)-1]
|
||
}
|
||
|
||
return s
|
||
}
|
||
|
||
// FormatQuantity 格式化数量到正确的精度
|
||
func (t *FuturesTrader) FormatQuantity(symbol string, quantity float64) (string, error) {
|
||
precision, err := t.GetSymbolPrecision(symbol)
|
||
if err != nil {
|
||
// 如果获取失败,使用默认格式
|
||
return fmt.Sprintf("%.3f", quantity), nil
|
||
}
|
||
|
||
format := fmt.Sprintf("%%.%df", precision)
|
||
return fmt.Sprintf(format, quantity), nil
|
||
}
|
||
|
||
// 辅助函数
|
||
func contains(s, substr string) bool {
|
||
return len(s) >= len(substr) && stringContains(s, substr)
|
||
}
|
||
|
||
func stringContains(s, substr string) bool {
|
||
for i := 0; i <= len(s)-len(substr); i++ {
|
||
if s[i:i+len(substr)] == substr {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// GetOrderStatus 获取订单状态
|
||
func (t *FuturesTrader) GetOrderStatus(symbol string, orderID string) (map[string]interface{}, error) {
|
||
// 将 orderID 转换为 int64
|
||
orderIDInt, err := strconv.ParseInt(orderID, 10, 64)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("无效的订单ID: %s", orderID)
|
||
}
|
||
|
||
order, err := t.client.NewGetOrderService().
|
||
Symbol(symbol).
|
||
OrderID(orderIDInt).
|
||
Do(context.Background())
|
||
if err != nil {
|
||
return nil, fmt.Errorf("获取订单状态失败: %w", err)
|
||
}
|
||
|
||
// 解析成交价格
|
||
avgPrice, _ := strconv.ParseFloat(order.AvgPrice, 64)
|
||
executedQty, _ := strconv.ParseFloat(order.ExecutedQuantity, 64)
|
||
|
||
result := map[string]interface{}{
|
||
"orderId": order.OrderID,
|
||
"symbol": order.Symbol,
|
||
"status": string(order.Status),
|
||
"avgPrice": avgPrice,
|
||
"executedQty": executedQty,
|
||
"side": string(order.Side),
|
||
"type": string(order.Type),
|
||
"time": order.Time,
|
||
"updateTime": order.UpdateTime,
|
||
}
|
||
|
||
// 币安合约的手续费需要通过 GetUserTrades 获取,这里暂时不获取
|
||
// 后续可以通过 WebSocket 或单独查询获取
|
||
result["commission"] = 0.0
|
||
|
||
return result, nil
|
||
}
|