418 lines
13 KiB
Go
418 lines
13 KiB
Go
|
package bitgetservice
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"go-admin/common/global"
|
|||
|
|
"go-admin/common/helper"
|
|||
|
|
"go-admin/config"
|
|||
|
|
"go-admin/models"
|
|||
|
|
"go-admin/pkg/utility"
|
|||
|
|
"strconv"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"github.com/bytedance/sonic"
|
|||
|
|
log "github.com/go-admin-team/go-admin-core/logger"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// BitgetRestApi Bitget REST API服务
|
|||
|
|
type BitgetRestApi struct {
|
|||
|
|
httpClient *helper.BitgetClient
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewBitgetRestApi 创建新的Bitget REST API实例
|
|||
|
|
func NewBitgetRestApi() *BitgetRestApi {
|
|||
|
|
// 从配置中获取代理设置
|
|||
|
|
proxyUrl := config.ExtConfig.ProxyUrl
|
|||
|
|
|
|||
|
|
// 使用代理配置创建HTTP客户端
|
|||
|
|
httpClient, err := helper.NewBitgetClient("", "", "", proxyUrl)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("创建Bitget HTTP客户端失败", err)
|
|||
|
|
// 如果代理创建失败，使用无代理的客户端
|
|||
|
|
httpClient, _ = helper.NewBitgetClient("", "", "", "")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &BitgetRestApi{
|
|||
|
|
httpClient: httpClient,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewBitgetSimpleClient 创建简单的Bitget客户端（兼容性函数）
|
|||
|
|
func NewBitgetSimpleClient() *BitgetClient {
|
|||
|
|
// 从配置中获取代理设置
|
|||
|
|
proxyUrl := config.ExtConfig.ProxyUrl
|
|||
|
|
|
|||
|
|
client, err := NewBitgetClient("", "", "", proxyUrl)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("创建Bitget简单客户端失败", err)
|
|||
|
|
// 如果代理创建失败，使用无代理的客户端
|
|||
|
|
client, _ = NewBitgetClient("", "", "", "")
|
|||
|
|
}
|
|||
|
|
return client
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// bitgetSymbolResponse 通用的符号响应结构体
|
|||
|
|
type bitgetSymbolResponse struct {
|
|||
|
|
Code string `json:"code"`
|
|||
|
|
Msg string `json:"msg"`
|
|||
|
|
Data []struct {
|
|||
|
|
Symbol string `json:"symbol"`
|
|||
|
|
BaseCoin string `json:"baseCoin"`
|
|||
|
|
QuoteCoin string `json:"quoteCoin"`
|
|||
|
|
Status string `json:"status"`
|
|||
|
|
MinTradeAmount string `json:"minTradeAmount"`
|
|||
|
|
MaxTradeAmount string `json:"maxTradeAmount"`
|
|||
|
|
TakerFeeRate string `json:"takerFeeRate"`
|
|||
|
|
MakerFeeRate string `json:"makerFeeRate"`
|
|||
|
|
PricePrecision string `json:"pricePrecision"`
|
|||
|
|
QuantityPrecision string `json:"quantityPrecision"`
|
|||
|
|
} `json:"data"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getSymbolsFromAPI 通用的符号获取方法
|
|||
|
|
// symbolType: "spot" 现货, "futures" 合约
|
|||
|
|
func (api *BitgetRestApi) getSymbolsFromAPI(symbolType string) ([]BitgetSymbol, error) {
|
|||
|
|
var url string
|
|||
|
|
var errorPrefix string
|
|||
|
|
|
|||
|
|
switch symbolType {
|
|||
|
|
case "spot":
|
|||
|
|
url = "/api/v2/spot/public/symbols"
|
|||
|
|
errorPrefix = "现货"
|
|||
|
|
case "futures":
|
|||
|
|
url = "/api/v2/mix/market/contracts?productType=USDT-FUTURES"
|
|||
|
|
errorPrefix = "合约"
|
|||
|
|
default:
|
|||
|
|
return nil, fmt.Errorf("不支持的符号类型: %s", symbolType)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送API请求
|
|||
|
|
resp, _, err := api.httpClient.SendSpotRequestAuth(url, "GET", map[string]interface{}{})
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("获取bitget %s交易所信息失败: %v", errorPrefix, err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析响应数据
|
|||
|
|
var response bitgetSymbolResponse
|
|||
|
|
if err := sonic.Unmarshal(resp, &response); err != nil {
|
|||
|
|
return nil, fmt.Errorf("解析bitget %s交易所信息失败: %v", errorPrefix, err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查API响应状态
|
|||
|
|
if response.Code != "00000" {
|
|||
|
|
return nil, fmt.Errorf("bitget API错误: %s", response.Msg)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 转换数据格式
|
|||
|
|
return api.convertToSymbols(response.Data), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// convertToSymbols 将API响应数据转换为BitgetSymbol切片
|
|||
|
|
func (api *BitgetRestApi) convertToSymbols(data []struct {
|
|||
|
|
Symbol string `json:"symbol"`
|
|||
|
|
BaseCoin string `json:"baseCoin"`
|
|||
|
|
QuoteCoin string `json:"quoteCoin"`
|
|||
|
|
Status string `json:"status"`
|
|||
|
|
MinTradeAmount string `json:"minTradeAmount"`
|
|||
|
|
MaxTradeAmount string `json:"maxTradeAmount"`
|
|||
|
|
TakerFeeRate string `json:"takerFeeRate"`
|
|||
|
|
MakerFeeRate string `json:"makerFeeRate"`
|
|||
|
|
PricePrecision string `json:"pricePrecision"`
|
|||
|
|
QuantityPrecision string `json:"quantityPrecision"`
|
|||
|
|
}) []BitgetSymbol {
|
|||
|
|
symbols := make([]BitgetSymbol, 0, len(data))
|
|||
|
|
|
|||
|
|
for _, item := range data {
|
|||
|
|
if item.Status == "online" {
|
|||
|
|
priceDigit, _ := strconv.Atoi(item.PricePrecision)
|
|||
|
|
quantileDigit, _ := strconv.Atoi(item.QuantityPrecision)
|
|||
|
|
|
|||
|
|
symbol := BitgetSymbol{
|
|||
|
|
Symbol: item.Symbol,
|
|||
|
|
BaseAsset: item.BaseCoin,
|
|||
|
|
QuoteAsset: item.QuoteCoin,
|
|||
|
|
Status: item.Status,
|
|||
|
|
MinQty: item.MinTradeAmount,
|
|||
|
|
MaxQty: item.MaxTradeAmount,
|
|||
|
|
PriceDigit: priceDigit,
|
|||
|
|
QuantityDigit: quantileDigit,
|
|||
|
|
}
|
|||
|
|
symbols = append(symbols, symbol)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return symbols
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetSpotSymbols 获取现货交易对信息
|
|||
|
|
func (api *BitgetRestApi) GetSpotSymbols() ([]BitgetSymbol, error) {
|
|||
|
|
return api.getSymbolsFromAPI("spot")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetFutSymbols 获取合约交易对信息
|
|||
|
|
func (api *BitgetRestApi) GetFutSymbols() ([]BitgetSymbol, error) {
|
|||
|
|
return api.getSymbolsFromAPI("futures")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// symbolConfig 符号配置结构体
|
|||
|
|
type symbolConfig struct {
|
|||
|
|
symbolType string
|
|||
|
|
quoteAssetSymbols []string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// buildTradeSetFromSymbols 通用的TradeSet构建方法
|
|||
|
|
func buildTradeSetFromSymbols(symbols []BitgetSymbol, config symbolConfig) (map[string]models.TradeSet, []string) {
|
|||
|
|
tradeSets := make(map[string]models.TradeSet, len(symbols))
|
|||
|
|
deleteSymbols := make([]string, 0)
|
|||
|
|
|
|||
|
|
for _, item := range symbols {
|
|||
|
|
if utility.ContainsStr(config.quoteAssetSymbols, item.QuoteAsset) && item.Status == "online" {
|
|||
|
|
tradeSet := models.TradeSet{
|
|||
|
|
Coin: item.BaseAsset,
|
|||
|
|
Currency: item.QuoteAsset,
|
|||
|
|
PriceDigit: item.PriceDigit,
|
|||
|
|
AmountDigit: item.QuantityDigit,
|
|||
|
|
MinQty: utility.StringAsFloat(item.MinQty),
|
|||
|
|
}
|
|||
|
|
tradeSets[item.Symbol] = tradeSet
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return tradeSets, deleteSymbols
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetSpotSymbols 获取现货交易对并构建TradeSet映射
|
|||
|
|
func GetSpotSymbols() (map[string]models.TradeSet, []string, error) {
|
|||
|
|
api := NewBitgetRestApi()
|
|||
|
|
symbols, err := api.GetSpotSymbols()
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("获取bitget规范信息失败", err)
|
|||
|
|
return make(map[string]models.TradeSet, 0), []string{}, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
config := symbolConfig{
|
|||
|
|
symbolType: "spot",
|
|||
|
|
quoteAssetSymbols: []string{"USDT", "ETH", "BTC", "SOL", "BNB", "DOGE"},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
tradeSets, deleteSymbols := buildTradeSetFromSymbols(symbols, config)
|
|||
|
|
return tradeSets, deleteSymbols, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetFutSymbols 获取合约交易对并构建TradeSet映射
|
|||
|
|
func GetFutSymbols() (map[string]models.TradeSet, []string, error) {
|
|||
|
|
api := NewBitgetRestApi()
|
|||
|
|
symbols, err := api.GetFutSymbols()
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("获取bitget合约规范信息失败", err)
|
|||
|
|
return make(map[string]models.TradeSet, 0), []string{}, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
config := symbolConfig{
|
|||
|
|
symbolType: "futures",
|
|||
|
|
quoteAssetSymbols: []string{"USDT"},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
tradeSets, deleteSymbols := buildTradeSetFromSymbols(symbols, config)
|
|||
|
|
return tradeSets, deleteSymbols, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getTicker24hData 获取24小时行情数据的通用方法
|
|||
|
|
func (api *BitgetRestApi) getTicker24hData(url, tickerType, description string, tradeSet *map[string]models.TradeSet) ([]string, error) {
|
|||
|
|
// 定义响应数据结构
|
|||
|
|
type tickerResponse struct {
|
|||
|
|
Code string `json:"code"`
|
|||
|
|
Msg string `json:"msg"`
|
|||
|
|
Data []struct {
|
|||
|
|
Symbol string `json:"symbol"`
|
|||
|
|
High24h string `json:"high24h"`
|
|||
|
|
Low24h string `json:"low24h"`
|
|||
|
|
Close string `json:"lastPr"`
|
|||
|
|
Open string `json:"open"`
|
|||
|
|
Volume24h string `json:"baseVolume"` // 基础币成交量
|
|||
|
|
QuoteVol string `json:"usdtVolume"` // USDT成交量
|
|||
|
|
Change24h string `json:"changeUtc24h"` // UTC 24h涨跌幅
|
|||
|
|
Change string `json:"change24h"` // 24h涨跌幅
|
|||
|
|
} `json:"data"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送API请求
|
|||
|
|
resp, _, err := api.httpClient.SendSpotRequestAuth(url, "GET", map[string]interface{}{})
|
|||
|
|
if err != nil {
|
|||
|
|
return []string{}, fmt.Errorf("获取bitget %s 24h行情失败: %v", description, err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析响应数据
|
|||
|
|
var response tickerResponse
|
|||
|
|
if err := sonic.Unmarshal(resp, &response); err != nil {
|
|||
|
|
return []string{}, fmt.Errorf("解析bitget %s 24h行情失败: %v", description, err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查API响应状态
|
|||
|
|
if response.Code != "00000" {
|
|||
|
|
return []string{}, fmt.Errorf("bitget %s API错误: %s", description, response.Msg)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 处理行情数据
|
|||
|
|
deleteSymbols := make([]string, 0)
|
|||
|
|
processedCount := 0
|
|||
|
|
errorCount := 0
|
|||
|
|
|
|||
|
|
for _, item := range response.Data {
|
|||
|
|
// 构建缓存键
|
|||
|
|
key := fmt.Sprintf(tickerType, global.EXCHANGE_BITGET, item.Symbol)
|
|||
|
|
|
|||
|
|
// 检查交易对是否存在
|
|||
|
|
symbol, exists := (*tradeSet)[item.Symbol]
|
|||
|
|
if !exists {
|
|||
|
|
// 清理不存在的交易对缓存
|
|||
|
|
if err := helper.DefaultRedis.DeleteString(key); err != nil {
|
|||
|
|
log.Warn("删除无效缓存失败: %s, error: %v", key, err)
|
|||
|
|
}
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析交易对获取coin和currency
|
|||
|
|
var coin, currency string
|
|||
|
|
if strings.HasSuffix(item.Symbol, symbol.Currency) {
|
|||
|
|
coin = strings.TrimSuffix(item.Symbol, symbol.Currency)
|
|||
|
|
currency = symbol.Currency
|
|||
|
|
} else {
|
|||
|
|
deleteSymbols = append(deleteSymbols, item.Symbol)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 构建交易对数据
|
|||
|
|
symbol.Coin = coin
|
|||
|
|
symbol.Currency = currency
|
|||
|
|
symbol.OpenPrice = utility.StringAsFloat(item.Open)
|
|||
|
|
symbol.PriceChange = utility.StringAsFloat(item.Change)
|
|||
|
|
symbol.LowPrice = item.Low24h
|
|||
|
|
symbol.HighPrice = item.High24h
|
|||
|
|
symbol.Volume = item.Volume24h
|
|||
|
|
symbol.QuoteVolume = item.QuoteVol
|
|||
|
|
symbol.LastPrice = item.Close
|
|||
|
|
|
|||
|
|
// 验证数据有效性
|
|||
|
|
if symbol.Coin == "" || symbol.Currency == "" ||
|
|||
|
|
utility.StringToFloat64(symbol.QuoteVolume) <= 0 ||
|
|||
|
|
symbol.LastPrice == "" || utility.StringToFloat64(symbol.LastPrice) <= 0 {
|
|||
|
|
if err := helper.DefaultRedis.DeleteString(key); err != nil {
|
|||
|
|
log.Warn("删除无效数据缓存失败: %s, error: %v", key, err)
|
|||
|
|
}
|
|||
|
|
deleteSymbols = append(deleteSymbols, item.Symbol)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 缓存数据
|
|||
|
|
val, err := sonic.Marshal(symbol)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("序列化交易对数据失败: %s, error: %v", item.Symbol, err)
|
|||
|
|
errorCount++
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err := helper.DefaultRedis.SetString(key, string(val)); err != nil {
|
|||
|
|
log.Error("缓存交易对失败: %s, error: %v", item.Symbol, err)
|
|||
|
|
errorCount++
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
processedCount++
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
log.Info("处理%s行情数据完成: 成功=%d, 错误=%d, 删除=%d",
|
|||
|
|
description, processedCount, errorCount, len(deleteSymbols))
|
|||
|
|
|
|||
|
|
return deleteSymbols, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetSpotTicker24h 获取现货24小时行情变更并缓存
|
|||
|
|
func (api *BitgetRestApi) GetSpotTicker24h(tradeSet *map[string]models.TradeSet) ([]string, error) {
|
|||
|
|
return api.getTicker24hData("/api/v2/spot/market/tickers", global.TICKER_SPOT, "现货", tradeSet)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetFutTicker24h 获取U本位合约24小时行情变更并缓存
|
|||
|
|
func (api *BitgetRestApi) GetFutTicker24h(tradeSet *map[string]models.TradeSet) ([]string, error) {
|
|||
|
|
return api.getTicker24hData("/api/v2/mix/market/tickers?productType=USDT-FUTURES", global.TICKER_FUTURES, "U本位合约", tradeSet)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// BitgetSymbol Bitget交易对信息
|
|||
|
|
type BitgetSymbol struct {
|
|||
|
|
Symbol string `json:"symbol"`
|
|||
|
|
BaseCoin string `json:"baseCoin"`
|
|||
|
|
QuoteCoin string `json:"quoteCoin"`
|
|||
|
|
BaseAsset string `json:"baseAsset"`
|
|||
|
|
QuoteAsset string `json:"quoteAsset"`
|
|||
|
|
Status string `json:"status"`
|
|||
|
|
MinQty string `json:"minQty"`
|
|||
|
|
MaxQty string `json:"maxQty"`
|
|||
|
|
PriceDigit int `json:"priceDigit"`
|
|||
|
|
QuantityDigit int `json:"quantityDigit"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetAndReloadSymbols 获取并重新加载交易对信息
|
|||
|
|
// symbolType 1-现货 2-合约
|
|||
|
|
func GetAndReloadSymbols(data *map[string]models.TradeSet, symbolType int) error {
|
|||
|
|
var tradeSets map[string]models.TradeSet
|
|||
|
|
var err error
|
|||
|
|
|
|||
|
|
switch symbolType {
|
|||
|
|
case 1:
|
|||
|
|
tradeSets, _, err = GetSpotSymbols()
|
|||
|
|
case 2:
|
|||
|
|
tradeSets, _, err = GetFutSymbols()
|
|||
|
|
default:
|
|||
|
|
return fmt.Errorf("无效的symbolType: %d", symbolType)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
*data = tradeSets
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// InitSymbolsTicker24h 初始化24小时行情数据
|
|||
|
|
// symbolType 1-现货 2-合约
|
|||
|
|
func InitSymbolsTicker24h(data *map[string]models.TradeSet, symbolType int) ([]string, error) {
|
|||
|
|
api := NewBitgetRestApi()
|
|||
|
|
|
|||
|
|
switch symbolType {
|
|||
|
|
case 1:
|
|||
|
|
return api.GetSpotTicker24h(data)
|
|||
|
|
case 2:
|
|||
|
|
return api.GetFutTicker24h(data)
|
|||
|
|
default:
|
|||
|
|
return []string{}, fmt.Errorf("无效的symbolType: %d", symbolType)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetSpotTicker24 获取单个交易对24h行情
|
|||
|
|
func (api *BitgetRestApi) GetSpotTicker24(symbol string, data *models.Ticker24, tradeSet *models.TradeSet) error {
|
|||
|
|
key := fmt.Sprintf(global.TICKER_SPOT, global.EXCHANGE_BITGET, symbol)
|
|||
|
|
val, err := helper.DefaultRedis.GetString(key)
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
err = sonic.Unmarshal([]byte(val), tradeSet)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if tradeSet.Coin != "" {
|
|||
|
|
data.HighPrice = tradeSet.HighPrice
|
|||
|
|
data.ChangePercent = fmt.Sprintf("%g", tradeSet.PriceChange)
|
|||
|
|
data.LastPrice = tradeSet.LastPrice
|
|||
|
|
data.LowPrice = tradeSet.LowPrice
|
|||
|
|
data.OpenPrice = fmt.Sprintf("%g", tradeSet.OpenPrice)
|
|||
|
|
data.QuoteVolume = tradeSet.QuoteVolume
|
|||
|
|
data.Volume = tradeSet.Volume
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
}
|