58 lines
1.6 KiB
Go
58 lines
1.6 KiB
Go
|
package backtest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"git.qtrade.icu/lychiyu/bbgo/pkg/bbgo"
|
||
|
|
"git.qtrade.icu/lychiyu/bbgo/pkg/fixedpoint"
|
||
|
|
"git.qtrade.icu/lychiyu/bbgo/pkg/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
type FeeModeFunction func(order *types.Order, market *types.Market, feeRate fixedpoint.Value) (fee fixedpoint.Value, feeCurrency string)
|
||
|
|
|
||
|
|
func feeModeFunctionToken(order *types.Order, _ *types.Market, feeRate fixedpoint.Value) (fee fixedpoint.Value, feeCurrency string) {
|
||
|
|
quoteQuantity := order.Quantity.Mul(order.Price)
|
||
|
|
feeCurrency = FeeToken
|
||
|
|
fee = quoteQuantity.Mul(feeRate)
|
||
|
|
return fee, feeCurrency
|
||
|
|
}
|
||
|
|
|
||
|
|
func feeModeFunctionNative(order *types.Order, market *types.Market, feeRate fixedpoint.Value) (fee fixedpoint.Value, feeCurrency string) {
|
||
|
|
switch order.Side {
|
||
|
|
|
||
|
|
case types.SideTypeBuy:
|
||
|
|
fee = order.Quantity.Mul(feeRate)
|
||
|
|
feeCurrency = market.BaseCurrency
|
||
|
|
|
||
|
|
case types.SideTypeSell:
|
||
|
|
quoteQuantity := order.Quantity.Mul(order.Price)
|
||
|
|
fee = quoteQuantity.Mul(feeRate)
|
||
|
|
feeCurrency = market.QuoteCurrency
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
return fee, feeCurrency
|
||
|
|
}
|
||
|
|
|
||
|
|
func feeModeFunctionQuote(order *types.Order, market *types.Market, feeRate fixedpoint.Value) (fee fixedpoint.Value, feeCurrency string) {
|
||
|
|
feeCurrency = market.QuoteCurrency
|
||
|
|
quoteQuantity := order.Quantity.Mul(order.Price)
|
||
|
|
fee = quoteQuantity.Mul(feeRate)
|
||
|
|
return fee, feeCurrency
|
||
|
|
}
|
||
|
|
|
||
|
|
func getFeeModeFunction(feeMode bbgo.BacktestFeeMode) FeeModeFunction {
|
||
|
|
switch feeMode {
|
||
|
|
|
||
|
|
case bbgo.BacktestFeeModeNative:
|
||
|
|
return feeModeFunctionNative
|
||
|
|
|
||
|
|
case bbgo.BacktestFeeModeQuote:
|
||
|
|
return feeModeFunctionQuote
|
||
|
|
|
||
|
|
case bbgo.BacktestFeeModeToken:
|
||
|
|
return feeModeFunctionToken
|
||
|
|
|
||
|
|
default:
|
||
|
|
return feeModeFunctionQuote
|
||
|
|
}
|
||
|
|
}
|