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

fix used time field for withdraw

Browse Source
This commit is contained in:
c9s 2021-03-11 16:44:43 +08:00
parent b25671c864
commit 8e85274876
2 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pkg/exchange/max/exchange.go
View File

@ -504,7 +504,7 @@ func (e *Exchange) QueryWithdrawHistory(ctx context.Context, asset string, since
startTime = endTime
} else {
// its in descending order, so we get the first record
startTime = time.Unix(withdraws[0].UpdatedAt, 0)
startTime = time.Unix(withdraws[0].CreatedAt, 0)
}
}
@ -560,7 +560,7 @@ func (e *Exchange) QueryDepositHistory(ctx context.Context, asset string, since,
if len(deposits) < limit {
startTime = endTime
} else {
startTime = time.Unix(deposits[0].UpdatedAt, 0)
startTime = time.Unix(deposits[0].CreatedAt, 0)
}
}

18
pkg/service/withdraw.go
View File

@ -13,10 +13,12 @@ type WithdrawService struct {
DB *sqlx.DB
}
// Sync syncs the withdraw records into db
func (s *WithdrawService) Sync(ctx context.Context, ex types.Exchange) error {
txnIDs := map[string]struct{}{}
records, err := s.QueryLast(ex.Name(), 10)
// query descending
records, err := s.QueryLast(ex.Name(), 100)
if err != nil {
return err
}
@ -30,15 +32,25 @@ func (s *WithdrawService) Sync(ctx context.Context, ex types.Exchange) error {
return ErrNotImplemented
}
withdraws, err := transferApi.QueryWithdrawHistory(ctx, "", records[0].ApplyTime.Time(), time.Now())
since := time.Time{}
if len(records) > 0 {
since = records[len(records)-1].ApplyTime.Time()
}
// asset "" means all assets
withdraws, err := transferApi.QueryWithdrawHistory(ctx, "", since, time.Now())
if err != nil {
return err
}
for _, withdraw := range withdraws {
if _, exists := txnIDs[withdraw.TransactionID] ; exists {
if _, exists := txnIDs[withdraw.TransactionID]; exists {
continue
}
if err := s.Insert(withdraw); err != nil {
return err
}
}
return nil