lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-22 06:53:52 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Merge pull request #1786 from c9s/chiahung/xalign/detect-pending-deposit

Browse Source 
FEATURE: [xalign] detect active depoit. if found, skip align
This commit is contained in:
kbearXD 2024-10-21 16:42:12 +08:00 committed by GitHub
commit 859be58113
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 46 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -156,6 +156,38 @@ func (s *Strategy) detectActiveWithdraw(
return nil, err2
}
func (s *Strategy) detectActiveDeposit(
ctx context.Context,
sessions map[string]*bbgo.ExchangeSession,
) (*types.Deposit, error) {
var err2 error
until := time.Now()
since := until.Add(-time.Hour * 24)
for _, session := range sessions {
transferService, ok := session.Exchange.(types.ExchangeTransferHistoryService)
if !ok {
continue
}
deposits, err := transferService.QueryDepositHistory(ctx, "", since, until)
if err != nil {
log.WithError(err).Errorf("unable to query deposit history")
err2 = err
continue
}
for _, deposit := range deposits {
log.Infof("checking deposit status: %s", deposit.String())
switch deposit.Status {
case types.DepositPending:
return &deposit, nil
}
}
}
return nil, err2
}
func (s *Strategy) selectSessionForCurrency(
ctx context.Context, sessions map[string]*bbgo.ExchangeSession, currency string, changeQuantity fixedpoint.Value,
) (*bbgo.ExchangeSession, *types.SubmitOrder) {
@ -439,9 +471,9 @@ func (s *Strategy) align(ctx context.Context, sessions map[string]*bbgo.Exchange
pendingWithdraw, err := s.detectActiveWithdraw(ctx, sessions)
if err != nil {
log.WithError(err).Errorf("unable to check active transfers")
log.WithError(err).Errorf("unable to check active transfers (withdraw)")
} else if pendingWithdraw != nil {
log.Warnf("found active transfer, skip balance align check")
log.Warnf("found active transfer (withdraw), skip balance align check")
if activeTransferNotificationLimiter.Allow() {
bbgo.Notify("Found active withdraw, skip balance align", pendingWithdraw)
@ -449,6 +481,18 @@ func (s *Strategy) align(ctx context.Context, sessions map[string]*bbgo.Exchange
return
}
pendingDeposit, err := s.detectActiveDeposit(ctx, sessions)
if err != nil {
log.WithError(err).Errorf("unable to check active transfers (deposit)")
} else if pendingDeposit != nil {
log.Warnf("found active transfer (deposit), skip balance align check")
if activeTransferNotificationLimiter.Allow() {
bbgo.Notify("Found active deposit, skip balance align", pendingDeposit)
}
return
}
totalBalances, sessionBalances := s.aggregateBalances(ctx, sessions)
_ = sessionBalances