bbgo/pkg/strategy/autobuy/strategy.go

package autobuy

import (
	"context"
	"fmt"
	"sync"

	"github.com/robfig/cron/v3"
	"github.com/sirupsen/logrus"

	"git.qtrade.icu/lychiyu/bbgo/pkg/bbgo"
	"git.qtrade.icu/lychiyu/bbgo/pkg/fixedpoint"
	indicatorv2 "git.qtrade.icu/lychiyu/bbgo/pkg/indicator/v2"
	"git.qtrade.icu/lychiyu/bbgo/pkg/strategy/common"
	"git.qtrade.icu/lychiyu/bbgo/pkg/types"
)

const ID = "autobuy"

var log = logrus.WithField("strategy", ID)

func init() {
	bbgo.RegisterStrategy(ID, &Strategy{})
}

type Strategy struct {
	*common.Strategy

	Environment *bbgo.Environment
	Market      types.Market

	Symbol         string                  `json:"symbol"`
	Schedule       string                  `json:"schedule"`
	MinBaseBalance fixedpoint.Value        `json:"minBaseBalance"`
	OrderType      types.OrderType         `json:"orderType"`
	PriceType      types.PriceType         `json:"priceType"`
	Bollinger      *types.BollingerSetting `json:"bollinger"`
	DryRun         bool                    `json:"dryRun"`

	// Deprecated, to be replaced by MinBaseBalance
	Threshold fixedpoint.Value `json:"threshold"`

	bbgo.QuantityOrAmount

	boll *indicatorv2.BOLLStream
	cron *cron.Cron
}

func (s *Strategy) Initialize() error {
	if s.Strategy == nil {
		s.Strategy = &common.Strategy{}
	}

	if s.Threshold.Sign() > 0 && s.MinBaseBalance.IsZero() {
		s.MinBaseBalance = s.Threshold
	}
	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) Validate() error {
	if s.Symbol == "" {
		return fmt.Errorf("symbol is required")
	}

	if s.Schedule == "" {
		return fmt.Errorf("schedule is required")
	}

	if s.MinBaseBalance.Sign() <= 0 {
		return fmt.Errorf("minBaseBalance must be greater than 0")
	}

	if err := s.QuantityOrAmount.Validate(); err != nil {
		return err
	}

	if s.Bollinger != nil {
		if s.Bollinger.Interval == "" {
			return fmt.Errorf("bollinger interval is required")
		}

		if s.Bollinger.BandWidth <= 0 {
			return fmt.Errorf("bollinger band width must be greater than 0")
		}
	}
	return nil
}

func (s *Strategy) Defaults() error {
	if s.OrderType == "" {
		s.OrderType = types.OrderTypeLimit
	}

	if s.PriceType == "" {
		s.PriceType = types.PriceTypeMaker
	}
	return nil
}

func (s *Strategy) Subscribe(session *bbgo.ExchangeSession) {
	if s.Bollinger != nil {
		session.Subscribe(types.KLineChannel, s.Symbol, types.SubscribeOptions{Interval: s.Bollinger.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())

	if s.Bollinger != nil {
		s.boll = session.Indicators(s.Symbol).BOLL(s.Bollinger.IntervalWindow, s.Bollinger.BandWidth)
	}

	s.OrderExecutor.ActiveMakerOrders().OnFilled(func(order types.Order) {
		s.autobuy(ctx)
	})

	// the shutdown handler, you can cancel all orders
	bbgo.OnShutdown(ctx, func(ctx context.Context, wg *sync.WaitGroup) {
		defer wg.Done()
		s.cancelOrders(ctx)
	})

	s.cron = cron.New()
	s.cron.AddFunc(s.Schedule, func() {
		s.autobuy(ctx)
	})
	s.cron.Start()

	return nil
}

func (s *Strategy) cancelOrders(ctx context.Context) {
	if err := s.OrderExecutor.GracefulCancel(ctx); err != nil {
		log.WithError(err).Errorf("failed to cancel orders")
	}
}

func (s *Strategy) autobuy(ctx context.Context) {
	s.cancelOrders(ctx)

	balance, ok := s.Session.GetAccount().Balance(s.Market.BaseCurrency)
	if !ok {
		log.Errorf("%s balance not found", s.Market.BaseCurrency)
		return
	}
	log.Infof("balance: %s", balance.String())

	ticker, err := s.Session.Exchange.QueryTicker(ctx, s.Symbol)
	if err != nil {
		log.WithError(err).Errorf("failed to query ticker")
		return
	}

	side := types.SideTypeBuy
	price := s.PriceType.GetPrice(ticker, side)

	if s.boll != nil && price.Float64() > s.boll.UpBand.Last(0) {
		log.Infof("price %s is higher than upper band %f, skip", price.String(), s.boll.UpBand.Last(0))
		return
	}

	if balance.Available.Compare(s.MinBaseBalance) > 0 {
		log.Infof("balance %s is higher than minBaseBalance %s", balance.Available.String(), s.MinBaseBalance.String())
		return
	}
	log.Infof("balance %s is lower than minBaseBalance %s", balance.Available.String(), s.MinBaseBalance.String())

	quantity := s.CalculateQuantity(price)
	submitOrder := types.SubmitOrder{
		Symbol:   s.Symbol,
		Side:     side,
		Type:     s.OrderType,
		Quantity: quantity,
		Price:    price,
	}

	if s.DryRun {
		log.Infof("dry run, skip")
		return
	}

	log.Infof("submitting order: %s", submitOrder.String())
	_, err = s.OrderExecutor.SubmitOrders(ctx, submitOrder)
	if err != nil {
		log.WithError(err).Errorf("submit order error")
	}
}