108 lines
3.4 KiB
Go
108 lines
3.4 KiB
Go
|
package factory
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"go-admin/common/global"
|
|||
|
|
"go-admin/common/interfaces"
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// ExchangeFactory 交易所工厂
|
|||
|
|
type ExchangeFactory struct {
|
|||
|
|
clients map[string]interfaces.ExchangeInterface
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewExchangeFactory 创建交易所工厂实例
|
|||
|
|
func NewExchangeFactory() *ExchangeFactory {
|
|||
|
|
return &ExchangeFactory{
|
|||
|
|
clients: make(map[string]interfaces.ExchangeInterface),
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetExchange 获取指定交易所的客户端
|
|||
|
|
func (f *ExchangeFactory) GetExchange(exchangeType string) (interfaces.ExchangeInterface, error) {
|
|||
|
|
exchangeType = strings.ToLower(strings.TrimSpace(exchangeType))
|
|||
|
|
|
|||
|
|
// 检查是否已经创建了该交易所的客户端
|
|||
|
|
if client, exists := f.clients[exchangeType]; exists {
|
|||
|
|
return client, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 根据交易所类型创建对应的客户端
|
|||
|
|
var client interfaces.ExchangeInterface
|
|||
|
|
var err error
|
|||
|
|
|
|||
|
|
switch exchangeType {
|
|||
|
|
case global.EXCHANGE_BINANCE:
|
|||
|
|
client, err = f.createBinanceClient()
|
|||
|
|
case global.EXCHANGE_BITGET:
|
|||
|
|
client, err = f.createBitgetClient()
|
|||
|
|
case global.EXCHANGE_OKEX:
|
|||
|
|
return nil, fmt.Errorf("OKEx integration not implemented yet")
|
|||
|
|
case global.EXCHANGE_GATE:
|
|||
|
|
return nil, fmt.Errorf("Gate.io integration not implemented yet")
|
|||
|
|
case global.EXCHANGE_COINBASE:
|
|||
|
|
return nil, fmt.Errorf("Coinbase integration not implemented yet")
|
|||
|
|
case global.EXCHANGE_BITFINEX:
|
|||
|
|
return nil, fmt.Errorf("Bitfinex integration not implemented yet")
|
|||
|
|
case global.EXCHANGE_BITMEX:
|
|||
|
|
return nil, fmt.Errorf("BitMEX integration not implemented yet")
|
|||
|
|
default:
|
|||
|
|
return nil, fmt.Errorf("unsupported exchange type: %s", exchangeType)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("failed to create %s client: %v", exchangeType, err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 缓存客户端实例
|
|||
|
|
f.clients[exchangeType] = client
|
|||
|
|
|
|||
|
|
return client, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// createBinanceClient 创建Binance客户端包装器
|
|||
|
|
func (f *ExchangeFactory) createBinanceClient() (interfaces.ExchangeInterface, error) {
|
|||
|
|
// Binance适配器尚未完全实现，暴时返回错误
|
|||
|
|
return nil, fmt.Errorf("Binance adapter not fully implemented yet")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// createBitgetClient 创建Bitget客户端
|
|||
|
|
func (f *ExchangeFactory) createBitgetClient() (interfaces.ExchangeInterface, error) {
|
|||
|
|
// 使用默认空参数创建Bitget客户端（公开接口不需要认证）
|
|||
|
|
|
|||
|
|
return nil, fmt.Errorf("Bitget adapter not fully implemented yet")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetSupportedExchanges 获取支持的交易所列表
|
|||
|
|
func (f *ExchangeFactory) GetSupportedExchanges() []string {
|
|||
|
|
return []string{
|
|||
|
|
// global.EXCHANGE_BINANCE, // Binance适配器尚未完全实现
|
|||
|
|
global.EXCHANGE_BITGET,
|
|||
|
|
// 其他交易所可以在未来添加
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RegisterExchange 注册自定义交易所客户端
|
|||
|
|
func (f *ExchangeFactory) RegisterExchange(exchangeType string, client interfaces.ExchangeInterface) {
|
|||
|
|
f.clients[strings.ToLower(exchangeType)] = client
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ClearCache 清理缓存的客户端
|
|||
|
|
func (f *ExchangeFactory) ClearCache() {
|
|||
|
|
for _, client := range f.clients {
|
|||
|
|
if closer, ok := client.(interface{ Close() error }); ok {
|
|||
|
|
closer.Close()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
f.clients = make(map[string]interfaces.ExchangeInterface)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 全局工厂实例
|
|||
|
|
var DefaultExchangeFactory = NewExchangeFactory()
|
|||
|
|
|
|||
|
|
// GetExchangeClient 获取交易所客户端的便捷函数
|
|||
|
|
func GetExchangeClient(exchangeType string) (interfaces.ExchangeInterface, error) {
|
|||
|
|
return DefaultExchangeFactory.GetExchange(exchangeType)
|
|||
|
|
}
|