mirror of
https://github.com/c9s/bbgo.git
synced 2024-09-20 16:21:09 +00:00
104 lines
2.5 KiB
Go
104 lines
2.5 KiB
Go
package pnl
|
|
|
|
import (
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/c9s/bbgo/pkg/fixedpoint"
|
|
"github.com/c9s/bbgo/pkg/types"
|
|
)
|
|
|
|
type AverageCostCalculator struct {
|
|
TradingFeeCurrency string
|
|
Market types.Market
|
|
}
|
|
|
|
func (c *AverageCostCalculator) Calculate(symbol string, trades []types.Trade, currentPrice fixedpoint.Value) *AverageCostPnlReport {
|
|
// copy trades, so that we can truncate it.
|
|
var bidVolume = fixedpoint.Zero
|
|
var askVolume = fixedpoint.Zero
|
|
var feeUSD = fixedpoint.Zero
|
|
|
|
if len(trades) == 0 {
|
|
return &AverageCostPnlReport{
|
|
Symbol: symbol,
|
|
Market: c.Market,
|
|
LastPrice: currentPrice,
|
|
NumTrades: 0,
|
|
BuyVolume: bidVolume,
|
|
SellVolume: askVolume,
|
|
FeeInUSD: feeUSD,
|
|
}
|
|
}
|
|
|
|
var currencyFees = map[string]fixedpoint.Value{}
|
|
|
|
var position = types.NewPositionFromMarket(c.Market)
|
|
position.SetFeeRate(types.ExchangeFee{
|
|
// binance vip 0 uses 0.075%
|
|
MakerFeeRate: fixedpoint.NewFromFloat(0.075 * 0.01),
|
|
TakerFeeRate: fixedpoint.NewFromFloat(0.075 * 0.01),
|
|
})
|
|
|
|
// TODO: configure the exchange fee rate here later
|
|
// position.SetExchangeFeeRate()
|
|
var totalProfit fixedpoint.Value
|
|
var totalNetProfit fixedpoint.Value
|
|
|
|
var tradeIDs = map[uint64]types.Trade{}
|
|
|
|
for _, trade := range trades {
|
|
if _, exists := tradeIDs[trade.ID]; exists {
|
|
log.Warnf("duplicated trade: %+v", trade)
|
|
continue
|
|
}
|
|
|
|
if trade.Symbol != symbol {
|
|
continue
|
|
}
|
|
|
|
profit, netProfit, madeProfit := position.AddTrade(trade)
|
|
if madeProfit {
|
|
totalProfit = totalProfit.Add(profit)
|
|
totalNetProfit = totalNetProfit.Add(netProfit)
|
|
}
|
|
|
|
if trade.IsBuyer {
|
|
bidVolume = bidVolume.Add(trade.Quantity)
|
|
} else {
|
|
askVolume = askVolume.Add(trade.Quantity)
|
|
}
|
|
|
|
if _, ok := currencyFees[trade.FeeCurrency]; !ok {
|
|
currencyFees[trade.FeeCurrency] = trade.Fee
|
|
} else {
|
|
currencyFees[trade.FeeCurrency] = currencyFees[trade.FeeCurrency].Add(trade.Fee)
|
|
}
|
|
|
|
tradeIDs[trade.ID] = trade
|
|
}
|
|
|
|
unrealizedProfit := currentPrice.Sub(position.AverageCost).
|
|
Mul(position.GetBase())
|
|
|
|
return &AverageCostPnlReport{
|
|
Symbol: symbol,
|
|
Market: c.Market,
|
|
LastPrice: currentPrice,
|
|
NumTrades: len(trades),
|
|
StartTime: time.Time(trades[0].Time),
|
|
|
|
BuyVolume: bidVolume,
|
|
SellVolume: askVolume,
|
|
|
|
Stock: position.GetBase(),
|
|
Profit: totalProfit,
|
|
NetProfit: totalNetProfit,
|
|
UnrealizedProfit: unrealizedProfit,
|
|
AverageCost: position.AverageCost,
|
|
FeeInUSD: totalProfit.Sub(totalNetProfit),
|
|
CurrencyFees: currencyFees,
|
|
}
|
|
}
|