mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-25 08:15:15 +00:00
306 lines
9.5 KiB
Go
306 lines
9.5 KiB
Go
package binance
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/adshao/go-binance/v2"
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/c9s/bbgo/pkg/fixedpoint"
|
|
"github.com/c9s/bbgo/pkg/types"
|
|
"github.com/c9s/bbgo/pkg/util"
|
|
)
|
|
|
|
func toGlobalIsolatedUserAsset(userAsset binance.IsolatedUserAsset) types.IsolatedUserAsset {
|
|
return types.IsolatedUserAsset{
|
|
Asset: userAsset.Asset,
|
|
Borrowed: fixedpoint.MustNewFromString(userAsset.Borrowed),
|
|
Free: fixedpoint.MustNewFromString(userAsset.Free),
|
|
Interest: fixedpoint.MustNewFromString(userAsset.Interest),
|
|
Locked: fixedpoint.MustNewFromString(userAsset.Locked),
|
|
NetAsset: fixedpoint.MustNewFromString(userAsset.NetAsset),
|
|
NetAssetOfBtc: fixedpoint.MustNewFromString(userAsset.NetAssetOfBtc),
|
|
BorrowEnabled: userAsset.BorrowEnabled,
|
|
RepayEnabled: userAsset.RepayEnabled,
|
|
TotalAsset: fixedpoint.MustNewFromString(userAsset.TotalAsset),
|
|
}
|
|
}
|
|
|
|
func toGlobalIsolatedMarginAsset(asset binance.IsolatedMarginAsset) types.IsolatedMarginAsset {
|
|
return types.IsolatedMarginAsset{
|
|
Symbol: asset.Symbol,
|
|
QuoteAsset: toGlobalIsolatedUserAsset(asset.QuoteAsset),
|
|
BaseAsset: toGlobalIsolatedUserAsset(asset.BaseAsset),
|
|
IsolatedCreated: asset.IsolatedCreated,
|
|
MarginLevel: fixedpoint.MustNewFromString(asset.MarginLevel),
|
|
MarginLevelStatus: asset.MarginLevelStatus,
|
|
MarginRatio: fixedpoint.MustNewFromString(asset.MarginRatio),
|
|
IndexPrice: fixedpoint.MustNewFromString(asset.IndexPrice),
|
|
LiquidatePrice: fixedpoint.MustNewFromString(asset.LiquidatePrice),
|
|
LiquidateRate: fixedpoint.MustNewFromString(asset.LiquidateRate),
|
|
TradeEnabled: false,
|
|
}
|
|
}
|
|
|
|
func toGlobalIsolatedMarginAssets(assets []binance.IsolatedMarginAsset) (retAssets []types.IsolatedMarginAsset) {
|
|
for _, asset := range assets {
|
|
retAssets = append(retAssets, toGlobalIsolatedMarginAsset(asset))
|
|
}
|
|
|
|
return retAssets
|
|
}
|
|
|
|
func toGlobalIsolatedMarginAccount(account *binance.IsolatedMarginAccount) *types.IsolatedMarginAccount {
|
|
return &types.IsolatedMarginAccount{
|
|
TotalAssetOfBTC: fixedpoint.MustNewFromString(account.TotalNetAssetOfBTC),
|
|
TotalLiabilityOfBTC: fixedpoint.MustNewFromString(account.TotalLiabilityOfBTC),
|
|
TotalNetAssetOfBTC: fixedpoint.MustNewFromString(account.TotalNetAssetOfBTC),
|
|
Assets: toGlobalIsolatedMarginAssets(account.Assets),
|
|
}
|
|
}
|
|
|
|
func toGlobalMarginUserAssets(userAssets []binance.UserAsset) (retAssets []types.MarginUserAsset) {
|
|
for _, asset := range userAssets {
|
|
retAssets = append(retAssets, types.MarginUserAsset{
|
|
Asset: asset.Asset,
|
|
Borrowed: fixedpoint.MustNewFromString(asset.Borrowed),
|
|
Free: fixedpoint.MustNewFromString(asset.Free),
|
|
Interest: fixedpoint.MustNewFromString(asset.Interest),
|
|
Locked: fixedpoint.MustNewFromString(asset.Locked),
|
|
NetAsset: fixedpoint.MustNewFromString(asset.NetAsset),
|
|
})
|
|
}
|
|
|
|
return retAssets
|
|
}
|
|
|
|
func toGlobalMarginAccount(account *binance.MarginAccount) *types.MarginAccount {
|
|
return &types.MarginAccount{
|
|
BorrowEnabled: account.BorrowEnabled,
|
|
MarginLevel: fixedpoint.MustNewFromString(account.MarginLevel),
|
|
TotalAssetOfBTC: fixedpoint.MustNewFromString(account.TotalAssetOfBTC),
|
|
TotalLiabilityOfBTC: fixedpoint.MustNewFromString(account.TotalLiabilityOfBTC),
|
|
TotalNetAssetOfBTC: fixedpoint.MustNewFromString(account.TotalNetAssetOfBTC),
|
|
TradeEnabled: account.TradeEnabled,
|
|
TransferEnabled: account.TransferEnabled,
|
|
UserAssets: toGlobalMarginUserAssets(account.UserAssets),
|
|
}
|
|
}
|
|
|
|
func toGlobalTicker(stats *binance.PriceChangeStats) types.Ticker {
|
|
return types.Ticker{
|
|
Volume: util.MustParseFloat(stats.Volume),
|
|
Last: util.MustParseFloat(stats.LastPrice),
|
|
Open: util.MustParseFloat(stats.OpenPrice),
|
|
High: util.MustParseFloat(stats.HighPrice),
|
|
Low: util.MustParseFloat(stats.LowPrice),
|
|
Buy: util.MustParseFloat(stats.BidPrice),
|
|
Sell: util.MustParseFloat(stats.AskPrice),
|
|
Time: time.Unix(0, stats.CloseTime*int64(time.Millisecond)),
|
|
}
|
|
}
|
|
|
|
func toLocalOrderType(orderType types.OrderType) (binance.OrderType, error) {
|
|
switch orderType {
|
|
|
|
case types.OrderTypeLimitMaker:
|
|
return binance.OrderTypeLimitMaker, nil
|
|
|
|
case types.OrderTypeLimit:
|
|
return binance.OrderTypeLimit, nil
|
|
|
|
case types.OrderTypeStopLimit:
|
|
return binance.OrderTypeStopLossLimit, nil
|
|
|
|
case types.OrderTypeStopMarket:
|
|
return binance.OrderTypeStopLoss, nil
|
|
|
|
case types.OrderTypeMarket:
|
|
return binance.OrderTypeMarket, nil
|
|
}
|
|
|
|
return "", fmt.Errorf("can not convert to local order, order type %s not supported", orderType)
|
|
}
|
|
|
|
func toGlobalOrders(binanceOrders []*binance.Order) (orders []types.Order, err error) {
|
|
for _, binanceOrder := range binanceOrders {
|
|
order, err := toGlobalOrder(binanceOrder, false)
|
|
if err != nil {
|
|
return orders, err
|
|
}
|
|
|
|
orders = append(orders, *order)
|
|
}
|
|
|
|
return orders, err
|
|
}
|
|
|
|
func toGlobalOrder(binanceOrder *binance.Order, isMargin bool) (*types.Order, error) {
|
|
return &types.Order{
|
|
SubmitOrder: types.SubmitOrder{
|
|
ClientOrderID: binanceOrder.ClientOrderID,
|
|
Symbol: binanceOrder.Symbol,
|
|
Side: toGlobalSideType(binanceOrder.Side),
|
|
Type: toGlobalOrderType(binanceOrder.Type),
|
|
Quantity: util.MustParseFloat(binanceOrder.OrigQuantity),
|
|
Price: util.MustParseFloat(binanceOrder.Price),
|
|
TimeInForce: string(binanceOrder.TimeInForce),
|
|
},
|
|
Exchange: types.ExchangeBinance,
|
|
IsWorking: binanceOrder.IsWorking,
|
|
OrderID: uint64(binanceOrder.OrderID),
|
|
Status: toGlobalOrderStatus(binanceOrder.Status),
|
|
ExecutedQuantity: util.MustParseFloat(binanceOrder.ExecutedQuantity),
|
|
CreationTime: types.Time(millisecondTime(binanceOrder.Time)),
|
|
UpdateTime: types.Time(millisecondTime(binanceOrder.UpdateTime)),
|
|
IsMargin: isMargin,
|
|
IsIsolated: binanceOrder.IsIsolated,
|
|
}, nil
|
|
}
|
|
|
|
func millisecondTime(t int64) time.Time {
|
|
return time.Unix(0, t*int64(time.Millisecond))
|
|
}
|
|
|
|
func ToGlobalTrade(t binance.TradeV3, isMargin bool) (*types.Trade, error) {
|
|
// skip trade ID that is the same. however this should not happen
|
|
var side types.SideType
|
|
if t.IsBuyer {
|
|
side = types.SideTypeBuy
|
|
} else {
|
|
side = types.SideTypeSell
|
|
}
|
|
|
|
price, err := strconv.ParseFloat(t.Price, 64)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "price parse error, price: %+v", t.Price)
|
|
}
|
|
|
|
quantity, err := strconv.ParseFloat(t.Quantity, 64)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "quantity parse error, quantity: %+v", t.Quantity)
|
|
}
|
|
|
|
var quoteQuantity = 0.0
|
|
if len(t.QuoteQuantity) > 0 {
|
|
quoteQuantity, err = strconv.ParseFloat(t.QuoteQuantity, 64)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "quote quantity parse error, quoteQuantity: %+v", t.QuoteQuantity)
|
|
}
|
|
} else {
|
|
quoteQuantity = price * quantity
|
|
}
|
|
|
|
fee, err := strconv.ParseFloat(t.Commission, 64)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "commission parse error, commission: %+v", t.Commission)
|
|
}
|
|
|
|
return &types.Trade{
|
|
ID: t.ID,
|
|
OrderID: uint64(t.OrderID),
|
|
Price: price,
|
|
Symbol: t.Symbol,
|
|
Exchange: "binance",
|
|
Quantity: quantity,
|
|
QuoteQuantity: quoteQuantity,
|
|
Side: side,
|
|
IsBuyer: t.IsBuyer,
|
|
IsMaker: t.IsMaker,
|
|
Fee: fee,
|
|
FeeCurrency: t.CommissionAsset,
|
|
Time: types.Time(millisecondTime(t.Time)),
|
|
IsMargin: isMargin,
|
|
IsIsolated: t.IsIsolated,
|
|
}, nil
|
|
}
|
|
|
|
func toGlobalSideType(side binance.SideType) types.SideType {
|
|
switch side {
|
|
case binance.SideTypeBuy:
|
|
return types.SideTypeBuy
|
|
|
|
case binance.SideTypeSell:
|
|
return types.SideTypeSell
|
|
|
|
default:
|
|
log.Errorf("can not convert binance side type, unknown side type: %q", side)
|
|
return ""
|
|
}
|
|
}
|
|
|
|
func toGlobalOrderType(orderType binance.OrderType) types.OrderType {
|
|
switch orderType {
|
|
|
|
case binance.OrderTypeLimit,
|
|
binance.OrderTypeLimitMaker, binance.OrderTypeTakeProfitLimit:
|
|
return types.OrderTypeLimit
|
|
|
|
case binance.OrderTypeMarket:
|
|
return types.OrderTypeMarket
|
|
|
|
case binance.OrderTypeStopLossLimit:
|
|
return types.OrderTypeStopLimit
|
|
|
|
case binance.OrderTypeStopLoss:
|
|
return types.OrderTypeStopMarket
|
|
|
|
default:
|
|
log.Errorf("unsupported order type: %v", orderType)
|
|
return ""
|
|
}
|
|
}
|
|
|
|
func toGlobalOrderStatus(orderStatus binance.OrderStatusType) types.OrderStatus {
|
|
switch orderStatus {
|
|
case binance.OrderStatusTypeNew:
|
|
return types.OrderStatusNew
|
|
|
|
case binance.OrderStatusTypeRejected:
|
|
return types.OrderStatusRejected
|
|
|
|
case binance.OrderStatusTypeCanceled:
|
|
return types.OrderStatusCanceled
|
|
|
|
case binance.OrderStatusTypePartiallyFilled:
|
|
return types.OrderStatusPartiallyFilled
|
|
|
|
case binance.OrderStatusTypeFilled:
|
|
return types.OrderStatusFilled
|
|
}
|
|
|
|
return types.OrderStatus(orderStatus)
|
|
}
|
|
|
|
// ConvertTrades converts the binance v3 trade into the global trade type
|
|
func ConvertTrades(remoteTrades []*binance.TradeV3) (trades []types.Trade, err error) {
|
|
for _, t := range remoteTrades {
|
|
trade, err := ToGlobalTrade(*t, false)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "binance v3 trade parse error, trade: %+v", *t)
|
|
}
|
|
|
|
trades = append(trades, *trade)
|
|
}
|
|
|
|
return trades, err
|
|
}
|
|
|
|
func convertSubscription(s types.Subscription) string {
|
|
// binance uses lower case symbol name,
|
|
// for kline, it's "<symbol>@kline_<interval>"
|
|
// for depth, it's "<symbol>@depth OR <symbol>@depth@100ms"
|
|
switch s.Channel {
|
|
case types.KLineChannel:
|
|
return fmt.Sprintf("%s@%s_%s", strings.ToLower(s.Symbol), s.Channel, s.Options.String())
|
|
|
|
case types.BookChannel:
|
|
return fmt.Sprintf("%s@depth", strings.ToLower(s.Symbol))
|
|
}
|
|
|
|
return fmt.Sprintf("%s@%s", strings.ToLower(s.Symbol), s.Channel)
|
|
}
|