rewrite deposit sync service

pkg/service/deposit.go

@@ -4,8 +4,10 @@ import (
 	"context"
 	"time"
 
+	sq "github.com/Masterminds/squirrel"
 	"github.com/jmoiron/sqlx"
 
+	"github.com/c9s/bbgo/pkg/exchange/batch"
 	"github.com/c9s/bbgo/pkg/types"
 )
 
@@ -14,41 +16,49 @@ type DepositService struct {
 }
 
 // Sync syncs the withdraw records into db
-func (s *DepositService) Sync(ctx context.Context, ex types.Exchange) error {
+func (s *DepositService) Sync(ctx context.Context, ex types.Exchange, startTime time.Time) error {
-	txnIDs := map[string]struct{}{}
+	isMargin, isFutures, isIsolated, _ := getExchangeAttributes(ex)
-
+	if isMargin || isFutures || isIsolated {
-	// query descending
+		// only works in spot
-	records, err := s.QueryLast(ex.Name(), 10)
+		return nil
-	if err != nil {
-		return err
-	}
-
-	for _, record := range records {
-		txnIDs[record.TransactionID] = struct{}{}
 	}
 
 	transferApi, ok := ex.(types.ExchangeTransferService)
 	if !ok {
-		return ErrNotImplemented
+		return nil
 	}
 
-	since := time.Time{}
+	tasks := []SyncTask{
-	if len(records) > 0 {
+		{
-		since = records[len(records)-1].Time.Time()
+			Type:   types.Deposit{},
+			Select: SelectLastDeposits(ex.Name(), 100),
+			BatchQuery: func(ctx context.Context, startTime, endTime time.Time) (interface{}, chan error) {
+				query := &batch.DepositBatchQuery{
+					ExchangeTransferService: transferApi,
+				}
+				return query.Query(ctx, "", startTime, endTime)
+			},
+			Time: func(obj interface{}) time.Time {
+				return obj.(types.Deposit).Time.Time()
+			},
+			ID: func(obj interface{}) string {
+				deposit := obj.(types.Deposit)
+				return deposit.TransactionID
+			},
+			Filter: func(obj interface{}) bool {
+				deposit := obj.(types.Deposit)
+				if len(deposit.TransactionID) == 0 {
+					return false
 				}
 
-	// asset "" means all assets
+				return true
-	deposits, err := transferApi.QueryDepositHistory(ctx, "", since, time.Now())
+			},
-	if err != nil {
+			LogInsert: true,
-		return err
+		},
 	}
 
-	for _, deposit := range deposits {
+	for _, sel := range tasks {
-		if _, exists := txnIDs[deposit.TransactionID]; exists {
+		if err := sel.execute(ctx, s.DB, startTime); err != nil {
-			continue
-		}
-
-		if err := s.Insert(deposit); err != nil {
 			return err
 		}
 	}
@@ -104,3 +114,14 @@ func (s *DepositService) Insert(deposit types.Deposit) error {
 	_, err := s.DB.NamedExec(sql, deposit)
 	return err
 }
+
+
+func SelectLastDeposits(ex types.ExchangeName, limit uint64) sq.SelectBuilder {
+	return sq.Select("*").
+		From("deposits").
+		Where(sq.And{
+			sq.Eq{"exchange": ex},
+		}).
+		OrderBy("time DESC").
+		Limit(limit)
+}