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

deposit2transfer: fix mutex lock

Browse Source
This commit is contained in:
c9s 2023-08-08 12:38:23 +08:00
parent 073c4562fd
commit 4a28843a0a
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
2 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config/deposit2transfer.yaml
View File

@ -9,4 +9,4 @@ exchangeStrategies:
## assets are the assets you want to transfer into the margin account
assets:
- BTC
- USDT

19
pkg/strategy/deposit2transfer/strategy.go
View File

@ -39,7 +39,7 @@ type Strategy struct {
Assets []string `json:"assets"`
Interval types.Interval `json:"interval"`
Interval types.Duration `json:"interval"`
marginTransferService marginTransferService
depositHistoryService types.ExchangeTransferService
@ -56,8 +56,8 @@ func (s *Strategy) ID() string {
func (s *Strategy) Subscribe(session *bbgo.ExchangeSession) {}
func (s *Strategy) Defaults() error {
if s.Interval == "" {
s.Interval = types.Interval1m
if s.Interval == 0 {
s.Interval = types.Duration(5 * time.Minute)
}
return nil
@ -87,7 +87,9 @@ func (s *Strategy) Run(ctx context.Context, orderExecutor bbgo.OrderExecutor, se
return errDepositHistoryNotSupport
}
go s.tickWatcher(ctx, s.Interval.Duration())
session.UserDataStream.OnStart(func() {
go s.tickWatcher(ctx, s.Interval.Duration())
})
return nil
}
@ -97,6 +99,7 @@ func (s *Strategy) tickWatcher(ctx context.Context, interval time.Duration) {
defer ticker.Stop()
s.checkDeposits(ctx)
for {
select {
case <-ctx.Done():
@ -121,6 +124,7 @@ func (s *Strategy) checkDeposits(ctx context.Context) {
if len(succeededDeposits) == 0 {
log.Infof("no %s deposit found", asset)
continue
}
for _, d := range succeededDeposits {
@ -163,10 +167,10 @@ func (s *Strategy) scanDepositHistory(ctx context.Context, asset string, duratio
})
s.mu.Lock()
defer s.mu.Lock()
defer s.mu.Unlock()
for _, deposit := range deposits {
log.Infof("scanning deposit: %+v", deposit)
log.Infof("checking deposit: %+v", deposit)
if deposit.Asset != asset {
continue
@ -177,9 +181,10 @@ func (s *Strategy) scanDepositHistory(ctx context.Context, asset string, duratio
s.watchingDeposits[deposit.TransactionID] = deposit
} else {
switch deposit.Status {
case types.DepositSuccess:
// ignore all deposits that are already success
log.Infof("ignored succeess deposit: %s", deposit.TransactionID)
log.Infof("ignored succeess deposit: %s %+v", deposit.TransactionID, deposit)
continue
case types.DepositCredited, types.DepositPending: