From 9e2cb4bd7f4bee42ad4cf88c32478b7f981b7b87 Mon Sep 17 00:00:00 2001 From: c9s Date: Tue, 12 Nov 2024 15:48:17 +0800 Subject: [PATCH] integrate livenote and slack alert into deposit2transfer --- pkg/exchange/binance/exchange.go | 1 + pkg/strategy/deposit2transfer/strategy.go | 53 +++++++++++++++++++---- pkg/types/deposit.go | 36 +++++++++++++-- 3 files changed, 78 insertions(+), 12 deletions(-) diff --git a/pkg/exchange/binance/exchange.go b/pkg/exchange/binance/exchange.go index 9fb017dde..320e62830 100644 --- a/pkg/exchange/binance/exchange.go +++ b/pkg/exchange/binance/exchange.go @@ -654,6 +654,7 @@ func (e *Exchange) QueryDepositHistory(ctx context.Context, asset string, since, RawStatus: strconv.Itoa(int(d.Status)), UnlockConfirm: d.UnlockConfirm, Confirmation: d.ConfirmTimes, + Network: d.Network, }) } diff --git a/pkg/strategy/deposit2transfer/strategy.go b/pkg/strategy/deposit2transfer/strategy.go index 5b779020b..ae01da938 100644 --- a/pkg/strategy/deposit2transfer/strategy.go +++ b/pkg/strategy/deposit2transfer/strategy.go @@ -14,6 +14,7 @@ import ( "github.com/c9s/bbgo/pkg/bbgo" "github.com/c9s/bbgo/pkg/exchange/retry" "github.com/c9s/bbgo/pkg/fixedpoint" + "github.com/c9s/bbgo/pkg/livenote" "github.com/c9s/bbgo/pkg/types" ) @@ -40,6 +41,10 @@ func init() { bbgo.RegisterStrategy(ID, &Strategy{}) } +type SlackAlert struct { + Mentions []string `json:"mentions"` +} + type Strategy struct { Environment *bbgo.Environment @@ -48,6 +53,8 @@ type Strategy struct { Interval types.Duration `json:"interval"` TransferDelay types.Duration `json:"transferDelay"` + SlackAlert *SlackAlert `json:"slackAlert"` + marginTransferService marginTransferService depositHistoryService types.ExchangeTransferService @@ -137,7 +144,7 @@ func (s *Strategy) tickWatcher(ctx context.Context, interval time.Duration) { } func (s *Strategy) checkDeposits(ctx context.Context) { - accountLimiter := rate.NewLimiter(rate.Every(3*time.Second), 1) + accountLimiter := rate.NewLimiter(rate.Every(5*time.Second), 1) for _, asset := range s.Assets { logger := s.logger.WithField("asset", asset) @@ -204,16 +211,39 @@ func (s *Strategy) checkDeposits(ctx context.Context) { d.Amount.String(), d.Asset, amount.String(), d.Asset) + if s.SlackAlert != nil { + bbgo.PostLiveNote(&d, + livenote.Comment(fmt.Sprintf("Transferring deposit asset %s %s into the margin account", amount.String(), d.Asset)), + ) + } + err2 := retry.GeneralBackoff(ctx, func() error { return s.marginTransferService.TransferMarginAccountAsset(ctx, d.Asset, amount, types.TransferIn) }) if err2 != nil { logger.WithError(err2).Errorf("unable to transfer deposit asset into the margin account") + + if s.SlackAlert != nil { + bbgo.PostLiveNote(&d, + livenote.Comment(fmt.Sprintf("Margin account transfer error: %+v", err2)), + ) + } } } } } +func (s *Strategy) addWatchingDeposit(deposit types.Deposit) { + s.watchingDeposits[deposit.TransactionID] = deposit + + if s.SlackAlert != nil { + bbgo.PostLiveNote(&deposit, + livenote.CompareObject(true), + livenote.OneTimeMention(s.SlackAlert.Mentions...), + ) + } +} + func (s *Strategy) scanDepositHistory(ctx context.Context, asset string, duration time.Duration) ([]types.Deposit, error) { logger := s.logger.WithField("asset", asset) logger.Debugf("scanning %s deposit history...", asset) @@ -239,6 +269,7 @@ func (s *Strategy) scanDepositHistory(ctx context.Context, asset string, duratio s.mu.Lock() defer s.mu.Unlock() + // update the watching deposits for _, deposit := range deposits { logger.Debugf("checking deposit: %+v", deposit) @@ -246,18 +277,22 @@ func (s *Strategy) scanDepositHistory(ctx context.Context, asset string, duratio continue } + // if the deposit record is already in the watch list, update it if _, ok := s.watchingDeposits[deposit.TransactionID]; ok { - // if the deposit record is in the watch list, update it - s.watchingDeposits[deposit.TransactionID] = deposit + s.addWatchingDeposit(deposit) } else { + // if the deposit record is not in the watch list, we need to check the status + // here the deposit is outside the watching list switch deposit.Status { case types.DepositSuccess: + // if the deposit is in success status, we need to check if it's newer than the latest deposit time + // this usually happens when the deposit is credited to the account very quickly if depositTime, ok := s.lastAssetDepositTimes[asset]; ok { // if it's newer than the latest deposit time, then we just add it the monitoring list if deposit.Time.After(depositTime) { logger.Infof("adding new success deposit: %s", deposit.TransactionID) - s.watchingDeposits[deposit.TransactionID] = deposit + s.addWatchingDeposit(deposit) } } else { // ignore all initial deposits that are already in success status @@ -266,7 +301,7 @@ func (s *Strategy) scanDepositHistory(ctx context.Context, asset string, duratio case types.DepositCredited, types.DepositPending: logger.Infof("adding pending deposit: %s", deposit.TransactionID) - s.watchingDeposits[deposit.TransactionID] = deposit + s.addWatchingDeposit(deposit) } } } @@ -281,10 +316,12 @@ func (s *Strategy) scanDepositHistory(ctx context.Context, asset string, duratio } var succeededDeposits []types.Deposit - for _, deposit := range s.watchingDeposits { - if deposit.Status == types.DepositSuccess { - logger.Infof("found pending -> success deposit: %+v", deposit) + // find the succeeded deposits + for _, deposit := range s.watchingDeposits { + switch deposit.Status { + case types.DepositSuccess: + logger.Infof("found pending -> success deposit: %+v", deposit) current, required := deposit.GetCurrentConfirmation() if required > 0 && deposit.UnlockConfirm > 0 && current < deposit.UnlockConfirm { logger.Infof("deposit %s unlock confirm %d is not reached, current: %d, required: %d, skip this round", deposit.TransactionID, deposit.UnlockConfirm, current, required) diff --git a/pkg/types/deposit.go b/pkg/types/deposit.go index bb2fa9de3..dcd58e9c6 100644 --- a/pkg/types/deposit.go +++ b/pkg/types/deposit.go @@ -46,6 +46,8 @@ type Deposit struct { // Confirmation format = "current/required", for example: "7/16" Confirmation string `json:"confirmation"` + + Network string `json:"network,omitempty"` } func (d Deposit) GetCurrentConfirmation() (current int, required int) { @@ -109,9 +111,35 @@ func (d *Deposit) SlackAttachment() slack.Attachment { }) } + if len(d.Confirmation) > 0 { + text := d.Confirmation + if d.UnlockConfirm > 0 { + text = fmt.Sprintf("%s (unlock %d)", d.Confirmation, d.UnlockConfirm) + } + fields = append(fields, slack.AttachmentField{ + Title: "Confirmation", + Value: text, + Short: false, + }) + } + + if len(d.Network) > 0 { + fields = append(fields, slack.AttachmentField{ + Title: "Network", + Value: d.Network, + Short: false, + }) + } + + fields = append(fields, slack.AttachmentField{ + Title: "Amount", + Value: d.Amount.String() + " " + d.Asset, + Short: false, + }) + return slack.Attachment{ Color: depositStatusSlackColor(d.Status), - Title: fmt.Sprintf("Deposit %s %s To %s", d.Amount.String(), d.Asset, d.Address), + Title: fmt.Sprintf("Deposit %s %s To %s (%s)", d.Amount.String(), d.Asset, d.Address, d.Exchange), // TitleLink: "", Pretext: "", Text: "", @@ -119,9 +147,9 @@ func (d *Deposit) SlackAttachment() slack.Attachment { // ServiceIcon: "", // FromURL: "", // OriginalURL: "", - Fields: fields, - Footer: fmt.Sprintf("Apply Time: %s", d.Time.Time().Format(time.RFC3339)), - // FooterIcon: "", + Fields: fields, + Footer: fmt.Sprintf("Apply Time: %s", d.Time.Time().Format(time.RFC3339)), + FooterIcon: ExchangeFooterIcon(d.Exchange), } }