mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-16 20:13:52 +00:00
492 lines
14 KiB
Go
492 lines
14 KiB
Go
package scmaker
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math"
|
|
"sync"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/c9s/bbgo/pkg/bbgo"
|
|
"github.com/c9s/bbgo/pkg/fixedpoint"
|
|
. "github.com/c9s/bbgo/pkg/indicator/v2"
|
|
"github.com/c9s/bbgo/pkg/strategy/common"
|
|
"github.com/c9s/bbgo/pkg/types"
|
|
)
|
|
|
|
const ID = "scmaker"
|
|
|
|
type advancedOrderCancelApi interface {
|
|
CancelAllOrders(ctx context.Context) ([]types.Order, error)
|
|
CancelOrdersBySymbol(ctx context.Context, symbol string) ([]types.Order, error)
|
|
}
|
|
|
|
type BollingerConfig struct {
|
|
Interval types.Interval `json:"interval"`
|
|
Window int `json:"window"`
|
|
K float64 `json:"k"`
|
|
}
|
|
|
|
func init() {
|
|
bbgo.RegisterStrategy(ID, &Strategy{})
|
|
}
|
|
|
|
// Strategy scmaker is a stable coin market maker
|
|
type Strategy struct {
|
|
*common.Strategy
|
|
|
|
Environment *bbgo.Environment
|
|
Market types.Market
|
|
|
|
Symbol string `json:"symbol"`
|
|
|
|
NumOfLiquidityLayers int `json:"numOfLiquidityLayers"`
|
|
|
|
LiquidityUpdateInterval types.Interval `json:"liquidityUpdateInterval"`
|
|
PriceRangeBollinger *BollingerConfig `json:"priceRangeBollinger"`
|
|
StrengthInterval types.Interval `json:"strengthInterval"`
|
|
|
|
AdjustmentUpdateInterval types.Interval `json:"adjustmentUpdateInterval"`
|
|
|
|
MidPriceEMA *types.IntervalWindow `json:"midPriceEMA"`
|
|
LiquiditySlideRule *bbgo.SlideRule `json:"liquidityScale"`
|
|
LiquidityLayerTickSize fixedpoint.Value `json:"liquidityLayerTickSize"`
|
|
LiquiditySkew fixedpoint.Value `json:"liquiditySkew"`
|
|
|
|
MaxExposure fixedpoint.Value `json:"maxExposure"`
|
|
|
|
MinProfit fixedpoint.Value `json:"minProfit"`
|
|
|
|
liquidityOrderBook, adjustmentOrderBook *bbgo.ActiveOrderBook
|
|
book *types.StreamOrderBook
|
|
|
|
liquidityScale bbgo.Scale
|
|
|
|
// indicators
|
|
ewma *EWMAStream
|
|
boll *BOLLStream
|
|
intensity *IntensityStream
|
|
}
|
|
|
|
func (s *Strategy) Initialize() error {
|
|
s.Strategy = &common.Strategy{}
|
|
return nil
|
|
}
|
|
|
|
func (s *Strategy) ID() string {
|
|
return ID
|
|
}
|
|
|
|
func (s *Strategy) InstanceID() string {
|
|
return fmt.Sprintf("%s:%s", ID, s.Symbol)
|
|
}
|
|
|
|
func (s *Strategy) Subscribe(session *bbgo.ExchangeSession) {
|
|
session.Subscribe(types.BookChannel, s.Symbol, types.SubscribeOptions{})
|
|
session.Subscribe(types.KLineChannel, s.Symbol, types.SubscribeOptions{Interval: s.AdjustmentUpdateInterval})
|
|
session.Subscribe(types.KLineChannel, s.Symbol, types.SubscribeOptions{Interval: s.LiquidityUpdateInterval})
|
|
|
|
if s.MidPriceEMA != nil {
|
|
session.Subscribe(types.KLineChannel, s.Symbol, types.SubscribeOptions{Interval: s.MidPriceEMA.Interval})
|
|
}
|
|
}
|
|
|
|
func (s *Strategy) Run(ctx context.Context, _ bbgo.OrderExecutor, session *bbgo.ExchangeSession) error {
|
|
s.Strategy.Initialize(ctx, s.Environment, session, s.Market, ID, s.InstanceID())
|
|
|
|
s.book = types.NewStreamBook(s.Symbol)
|
|
s.book.BindStream(session.MarketDataStream)
|
|
|
|
s.liquidityOrderBook = bbgo.NewActiveOrderBook(s.Symbol)
|
|
s.liquidityOrderBook.BindStream(session.UserDataStream)
|
|
|
|
s.adjustmentOrderBook = bbgo.NewActiveOrderBook(s.Symbol)
|
|
s.adjustmentOrderBook.BindStream(session.UserDataStream)
|
|
|
|
scale, err := s.LiquiditySlideRule.Scale()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := scale.Solve(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if cancelApi, ok := session.Exchange.(advancedOrderCancelApi); ok {
|
|
_, _ = cancelApi.CancelOrdersBySymbol(ctx, s.Symbol)
|
|
}
|
|
|
|
s.liquidityScale = scale
|
|
|
|
s.initializeMidPriceEMA(session)
|
|
s.initializePriceRangeBollinger(session)
|
|
s.initializeIntensityIndicator(session)
|
|
|
|
session.UserDataStream.OnStart(func() {
|
|
s.placeLiquidityOrders(ctx)
|
|
})
|
|
|
|
session.MarketDataStream.OnKLineClosed(func(k types.KLine) {
|
|
if k.Interval == s.AdjustmentUpdateInterval {
|
|
s.placeAdjustmentOrders(ctx)
|
|
}
|
|
|
|
if k.Interval == s.LiquidityUpdateInterval {
|
|
s.placeLiquidityOrders(ctx)
|
|
}
|
|
})
|
|
|
|
bbgo.OnShutdown(ctx, func(ctx context.Context, wg *sync.WaitGroup) {
|
|
defer wg.Done()
|
|
|
|
err := s.liquidityOrderBook.GracefulCancel(ctx, s.Session.Exchange)
|
|
logErr(err, "unable to cancel liquidity orders")
|
|
|
|
err = s.adjustmentOrderBook.GracefulCancel(ctx, s.Session.Exchange)
|
|
logErr(err, "unable to cancel adjustment orders")
|
|
})
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Strategy) preloadKLines(
|
|
inc *KLineStream, session *bbgo.ExchangeSession, symbol string, interval types.Interval,
|
|
) {
|
|
if store, ok := session.MarketDataStore(symbol); ok {
|
|
if kLinesData, ok := store.KLinesOfInterval(interval); ok {
|
|
for _, k := range *kLinesData {
|
|
inc.EmitUpdate(k)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (s *Strategy) initializeMidPriceEMA(session *bbgo.ExchangeSession) {
|
|
kLines := KLines(session.MarketDataStream, s.Symbol, s.MidPriceEMA.Interval)
|
|
s.ewma = EWMA2(ClosePrices(kLines), s.MidPriceEMA.Window)
|
|
|
|
s.preloadKLines(kLines, session, s.Symbol, s.MidPriceEMA.Interval)
|
|
}
|
|
|
|
func (s *Strategy) initializeIntensityIndicator(session *bbgo.ExchangeSession) {
|
|
kLines := KLines(session.MarketDataStream, s.Symbol, s.StrengthInterval)
|
|
s.intensity = Intensity(kLines, 10)
|
|
|
|
s.preloadKLines(kLines, session, s.Symbol, s.StrengthInterval)
|
|
}
|
|
|
|
func (s *Strategy) initializePriceRangeBollinger(session *bbgo.ExchangeSession) {
|
|
kLines := KLines(session.MarketDataStream, s.Symbol, s.PriceRangeBollinger.Interval)
|
|
closePrices := ClosePrices(kLines)
|
|
s.boll = BOLL(closePrices, s.PriceRangeBollinger.Window, s.PriceRangeBollinger.K)
|
|
|
|
s.preloadKLines(kLines, session, s.Symbol, s.PriceRangeBollinger.Interval)
|
|
}
|
|
|
|
func (s *Strategy) placeAdjustmentOrders(ctx context.Context) {
|
|
_ = s.adjustmentOrderBook.GracefulCancel(ctx, s.Session.Exchange)
|
|
|
|
if s.Position.IsDust() {
|
|
return
|
|
}
|
|
|
|
ticker, err := s.Session.Exchange.QueryTicker(ctx, s.Symbol)
|
|
if logErr(err, "unable to query ticker") {
|
|
return
|
|
}
|
|
|
|
if _, err := s.Session.UpdateAccount(ctx); err != nil {
|
|
logErr(err, "unable to update account")
|
|
return
|
|
}
|
|
|
|
baseBal, _ := s.Session.Account.Balance(s.Market.BaseCurrency)
|
|
quoteBal, _ := s.Session.Account.Balance(s.Market.QuoteCurrency)
|
|
|
|
var adjOrders []types.SubmitOrder
|
|
|
|
posSize := s.Position.Base.Abs()
|
|
tickSize := s.Market.TickSize
|
|
|
|
if s.Position.IsShort() {
|
|
price := profitProtectedPrice(types.SideTypeBuy, s.Position.AverageCost, ticker.Sell.Add(tickSize.Neg()), s.Session.MakerFeeRate, s.MinProfit)
|
|
quoteQuantity := fixedpoint.Min(price.Mul(posSize), quoteBal.Available)
|
|
bidQuantity := quoteQuantity.Div(price)
|
|
|
|
if s.Market.IsDustQuantity(bidQuantity, price) {
|
|
return
|
|
}
|
|
|
|
adjOrders = append(adjOrders, types.SubmitOrder{
|
|
Symbol: s.Symbol,
|
|
Type: types.OrderTypeLimitMaker,
|
|
Side: types.SideTypeBuy,
|
|
Price: price,
|
|
Quantity: bidQuantity,
|
|
Market: s.Market,
|
|
TimeInForce: types.TimeInForceGTC,
|
|
})
|
|
} else if s.Position.IsLong() {
|
|
price := profitProtectedPrice(types.SideTypeSell, s.Position.AverageCost, ticker.Buy.Add(tickSize), s.Session.MakerFeeRate, s.MinProfit)
|
|
askQuantity := fixedpoint.Min(posSize, baseBal.Available)
|
|
|
|
if s.Market.IsDustQuantity(askQuantity, price) {
|
|
return
|
|
}
|
|
|
|
adjOrders = append(adjOrders, types.SubmitOrder{
|
|
Symbol: s.Symbol,
|
|
Type: types.OrderTypeLimitMaker,
|
|
Side: types.SideTypeSell,
|
|
Price: price,
|
|
Quantity: askQuantity,
|
|
Market: s.Market,
|
|
TimeInForce: types.TimeInForceGTC,
|
|
})
|
|
}
|
|
|
|
createdOrders, err := s.OrderExecutor.SubmitOrders(ctx, adjOrders...)
|
|
if logErr(err, "unable to place liquidity orders") {
|
|
return
|
|
}
|
|
|
|
s.adjustmentOrderBook.Add(createdOrders...)
|
|
}
|
|
|
|
func (s *Strategy) placeLiquidityOrders(ctx context.Context) {
|
|
ticker, err := s.Session.Exchange.QueryTicker(ctx, s.Symbol)
|
|
if logErr(err, "unable to query ticker") {
|
|
return
|
|
}
|
|
|
|
if s.IsHalted(ticker.Time) {
|
|
log.Warn("circuitBreakRiskControl: trading halted")
|
|
return
|
|
}
|
|
|
|
err = s.liquidityOrderBook.GracefulCancel(ctx, s.Session.Exchange)
|
|
if logErr(err, "unable to cancel orders") {
|
|
return
|
|
}
|
|
|
|
if ticker.Buy.IsZero() && ticker.Sell.IsZero() {
|
|
ticker.Sell = ticker.Last.Add(s.Market.TickSize)
|
|
ticker.Buy = ticker.Last.Sub(s.Market.TickSize)
|
|
} else if ticker.Buy.IsZero() {
|
|
ticker.Buy = ticker.Sell.Sub(s.Market.TickSize)
|
|
} else if ticker.Sell.IsZero() {
|
|
ticker.Sell = ticker.Buy.Add(s.Market.TickSize)
|
|
}
|
|
|
|
if _, err := s.Session.UpdateAccount(ctx); err != nil {
|
|
logErr(err, "unable to update account")
|
|
return
|
|
}
|
|
|
|
baseBal, _ := s.Session.Account.Balance(s.Market.BaseCurrency)
|
|
quoteBal, _ := s.Session.Account.Balance(s.Market.QuoteCurrency)
|
|
|
|
spread := ticker.Sell.Sub(ticker.Buy)
|
|
tickSize := fixedpoint.Max(s.LiquidityLayerTickSize, s.Market.TickSize)
|
|
|
|
midPriceEMA := s.ewma.Last(0)
|
|
midPrice := fixedpoint.NewFromFloat(midPriceEMA)
|
|
|
|
bandWidth := s.boll.Last(0)
|
|
|
|
log.Infof("spread: %f mid price ema: %f boll band width: %f", spread.Float64(), midPriceEMA, bandWidth)
|
|
|
|
sum := s.liquidityScale.Sum(1.0)
|
|
askSum := sum
|
|
bidSum := sum
|
|
|
|
log.Infof("liquidity sum: %f / %f", askSum, bidSum)
|
|
|
|
skew := s.LiquiditySkew.Float64()
|
|
useSkew := !s.LiquiditySkew.IsZero()
|
|
if useSkew {
|
|
askSum = sum / skew
|
|
bidSum = sum * skew
|
|
log.Infof("adjusted liqudity skew: %f / %f", askSum, bidSum)
|
|
}
|
|
|
|
var bidPrices []fixedpoint.Value
|
|
var askPrices []fixedpoint.Value
|
|
|
|
// calculate and collect prices
|
|
for i := 0; i <= s.NumOfLiquidityLayers; i++ {
|
|
fi := fixedpoint.NewFromInt(int64(i))
|
|
sp := tickSize.Mul(fi)
|
|
spreadBidPrice := midPrice.Sub(sp)
|
|
spreadAskPrice := midPrice.Add(sp)
|
|
|
|
bidPrice := ticker.Buy
|
|
askPrice := ticker.Sell
|
|
|
|
if i == s.NumOfLiquidityLayers {
|
|
bwf := fixedpoint.NewFromFloat(bandWidth)
|
|
bidPrice = fixedpoint.Min(midPrice.Add(bwf.Neg()), spreadBidPrice)
|
|
askPrice = fixedpoint.Max(midPrice.Add(bwf), spreadAskPrice)
|
|
} else if i > 0 {
|
|
bidPrice = spreadBidPrice
|
|
askPrice = spreadAskPrice
|
|
}
|
|
|
|
if i > 0 && bidPrice.Compare(ticker.Buy) > 0 {
|
|
bidPrice = ticker.Buy.Sub(sp)
|
|
}
|
|
|
|
if i > 0 && askPrice.Compare(ticker.Sell) < 0 {
|
|
askPrice = ticker.Sell.Add(sp)
|
|
}
|
|
|
|
bidPrice = s.Market.TruncatePrice(bidPrice)
|
|
askPrice = s.Market.TruncatePrice(askPrice)
|
|
|
|
bidPrices = append(bidPrices, bidPrice)
|
|
askPrices = append(askPrices, askPrice)
|
|
}
|
|
|
|
availableBase := baseBal.Available
|
|
availableQuote := quoteBal.Available
|
|
|
|
// check max exposure
|
|
if s.MaxExposure.Sign() > 0 {
|
|
availableQuote = fixedpoint.Min(availableQuote, s.MaxExposure)
|
|
|
|
baseQuoteValue := availableBase.Mul(ticker.Sell)
|
|
if baseQuoteValue.Compare(s.MaxExposure) > 0 {
|
|
availableBase = s.MaxExposure.Div(ticker.Sell)
|
|
}
|
|
}
|
|
|
|
makerQuota := &bbgo.QuotaTransaction{}
|
|
makerQuota.QuoteAsset.Add(availableQuote)
|
|
makerQuota.BaseAsset.Add(availableBase)
|
|
|
|
log.Infof("balances before liq orders: %s, %s",
|
|
baseBal.String(),
|
|
quoteBal.String())
|
|
|
|
if !s.Position.IsDust() {
|
|
if s.Position.IsLong() {
|
|
availableBase = availableBase.Sub(s.Position.Base)
|
|
availableBase = s.Market.RoundDownQuantityByPrecision(availableBase)
|
|
} else if s.Position.IsShort() {
|
|
posSizeInQuote := s.Position.Base.Mul(ticker.Sell)
|
|
availableQuote = availableQuote.Sub(posSizeInQuote)
|
|
}
|
|
}
|
|
|
|
askX := availableBase.Float64() / askSum
|
|
bidX := availableQuote.Float64() / (bidSum * (fixedpoint.Sum(bidPrices).Float64()))
|
|
|
|
askX = math.Trunc(askX*1e8) / 1e8
|
|
bidX = math.Trunc(bidX*1e8) / 1e8
|
|
|
|
var liqOrders []types.SubmitOrder
|
|
for i := 0; i <= s.NumOfLiquidityLayers; i++ {
|
|
bidQuantity := fixedpoint.NewFromFloat(s.liquidityScale.Call(float64(i)) * bidX)
|
|
askQuantity := fixedpoint.NewFromFloat(s.liquidityScale.Call(float64(i)) * askX)
|
|
bidPrice := bidPrices[i]
|
|
askPrice := askPrices[i]
|
|
|
|
log.Infof("liqudity layer #%d %f/%f = %f/%f", i, askPrice.Float64(), bidPrice.Float64(), askQuantity.Float64(), bidQuantity.Float64())
|
|
|
|
placeBuy := true
|
|
placeSell := true
|
|
averageCost := s.Position.AverageCost
|
|
// when long position, do not place sell orders below the average cost
|
|
if !s.Position.IsDust() {
|
|
if s.Position.IsLong() && askPrice.Compare(averageCost) < 0 {
|
|
placeSell = false
|
|
}
|
|
|
|
if s.Position.IsShort() && bidPrice.Compare(averageCost) > 0 {
|
|
placeBuy = false
|
|
}
|
|
}
|
|
|
|
quoteQuantity := bidQuantity.Mul(bidPrice)
|
|
|
|
if s.Market.IsDustQuantity(bidQuantity, bidPrice) || !makerQuota.QuoteAsset.Lock(quoteQuantity) {
|
|
placeBuy = false
|
|
}
|
|
|
|
if s.Market.IsDustQuantity(askQuantity, askPrice) || !makerQuota.BaseAsset.Lock(askQuantity) {
|
|
placeSell = false
|
|
}
|
|
|
|
if placeBuy {
|
|
liqOrders = append(liqOrders, types.SubmitOrder{
|
|
Symbol: s.Symbol,
|
|
Side: types.SideTypeBuy,
|
|
Type: types.OrderTypeLimitMaker,
|
|
Quantity: bidQuantity,
|
|
Price: bidPrice,
|
|
Market: s.Market,
|
|
TimeInForce: types.TimeInForceGTC,
|
|
})
|
|
}
|
|
|
|
if placeSell {
|
|
liqOrders = append(liqOrders, types.SubmitOrder{
|
|
Symbol: s.Symbol,
|
|
Side: types.SideTypeSell,
|
|
Type: types.OrderTypeLimitMaker,
|
|
Quantity: askQuantity,
|
|
Price: askPrice,
|
|
Market: s.Market,
|
|
TimeInForce: types.TimeInForceGTC,
|
|
})
|
|
}
|
|
}
|
|
|
|
makerQuota.Commit()
|
|
|
|
createdOrders, err := s.OrderExecutor.SubmitOrders(ctx, liqOrders...)
|
|
if logErr(err, "unable to place liquidity orders") {
|
|
return
|
|
}
|
|
|
|
s.liquidityOrderBook.Add(createdOrders...)
|
|
log.Infof("%d liq orders are placed successfully", len(liqOrders))
|
|
}
|
|
|
|
func profitProtectedPrice(
|
|
side types.SideType, averageCost, price, feeRate, minProfit fixedpoint.Value,
|
|
) fixedpoint.Value {
|
|
switch side {
|
|
case types.SideTypeSell:
|
|
minProfitPrice := averageCost.Add(
|
|
averageCost.Mul(feeRate.Add(minProfit)))
|
|
return fixedpoint.Max(minProfitPrice, price)
|
|
|
|
case types.SideTypeBuy:
|
|
minProfitPrice := averageCost.Sub(
|
|
averageCost.Mul(feeRate.Add(minProfit)))
|
|
return fixedpoint.Min(minProfitPrice, price)
|
|
|
|
}
|
|
return price
|
|
}
|
|
|
|
func logErr(err error, msgAndArgs ...interface{}) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
|
|
if len(msgAndArgs) == 0 {
|
|
log.WithError(err).Error(err.Error())
|
|
} else if len(msgAndArgs) == 1 {
|
|
msg := msgAndArgs[0].(string)
|
|
log.WithError(err).Error(msg)
|
|
} else if len(msgAndArgs) > 1 {
|
|
msg := msgAndArgs[0].(string)
|
|
log.WithError(err).Errorf(msg, msgAndArgs[1:]...)
|
|
}
|
|
|
|
return true
|
|
}
|