lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-10 09:11:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

xalign: add withdraw detection notification

Browse Source
This commit is contained in:
c9s 2024-08-01 16:58:07 +08:00
parent 600d81049e
commit 089e69a221
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
1 changed files with 20 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
pkg/strategy/xalign/strategy.go
View File

@ -9,6 +9,7 @@ import (
"time"
"github.com/sirupsen/logrus"
"golang.org/x/time/rate"
"github.com/c9s/bbgo/pkg/bbgo"
"github.com/c9s/bbgo/pkg/core"
@ -20,6 +21,8 @@ const ID = "xalign"
var log = logrus.WithField("strategy", ID)
var activeTransferNotificationLimiter = rate.NewLimiter(rate.Every(5*time.Minute), 1)
func init() {
bbgo.RegisterStrategy(ID, &Strategy{})
}
@ -47,6 +50,10 @@ type Strategy struct {
Duration types.Duration `json:"for"`
MaxAmounts map[string]fixedpoint.Value `json:"maxAmounts"`
SlackNotify bool `json:"slackNotify"`
SlackNotifyMentions []string `json:"slackNotifyMentions"`
SlackNotifyThresholdAmount fixedpoint.Value `json:"slackNotifyThresholdAmount,omitempty"`
faultBalanceRecords map[string][]TimeBalance
sessions map[string]*bbgo.ExchangeSession
@ -114,7 +121,10 @@ func (s *Strategy) aggregateBalances(
return totalBalances, sessionBalances
}
func (s *Strategy) detectActiveTransfers(ctx context.Context, sessions map[string]*bbgo.ExchangeSession) (bool, error) {
func (s *Strategy) detectActiveWithdraw(
ctx context.Context,
sessions map[string]*bbgo.ExchangeSession,
) (*types.Withdraw, error) {
var err2 error
until := time.Now()
since := until.Add(-time.Hour * 24)
@ -134,12 +144,12 @@ func (s *Strategy) detectActiveTransfers(ctx context.Context, sessions map[strin
for _, withdraw := range withdraws {
switch withdraw.Status {
case types.WithdrawStatusProcessing, types.WithdrawStatusSent, types.WithdrawStatusAwaitingApproval:
return true, nil
return &withdraw, nil
}
}
}
return false, err2
return nil, err2
}
func (s *Strategy) selectSessionForCurrency(
@ -405,7 +415,6 @@ func (s *Strategy) recordBalance(totalBalances types.BalanceMap) {
}
func (s *Strategy) align(ctx context.Context, sessions map[string]*bbgo.ExchangeSession) {
for sessionName, session := range sessions {
ob, ok := s.orderBooks[sessionName]
if !ok {
@ -414,16 +423,20 @@ func (s *Strategy) align(ctx context.Context, sessions map[string]*bbgo.Exchange
}
if ok {
if err := ob.GracefulCancel(ctx, session.Exchange); err != nil {
log.WithError(err).Errorf("can not cancel order")
log.WithError(err).Errorf("unable to cancel order")
}
}
}
foundActiveTransfer, err := s.detectActiveTransfers(ctx, sessions)
pendingWithdraw, err := s.detectActiveWithdraw(ctx, sessions)
if err != nil {
log.WithError(err).Errorf("unable to check active transfers")
} else if foundActiveTransfer {
} else if pendingWithdraw != nil {
log.Warnf("found active transfer, skip balance align check")
if activeTransferNotificationLimiter.Allow() {
bbgo.Notify("Found active withdraw, skip balance align", pendingWithdraw)
}
return
}