package batch

import (
	"context"
	"strconv"
	"time"

	"golang.org/x/time/rate"

	"github.com/c9s/bbgo/pkg/types"
)

type ClosedOrderBatchQuery struct {
	types.ExchangeTradeHistoryService
}

func (q *ClosedOrderBatchQuery) Query(ctx context.Context, symbol string, startTime, endTime time.Time, lastOrderID uint64) (c chan types.Order, errC chan error) {
	query := &AsyncTimeRangedBatchQuery{
		Type:    types.Order{},
		Limiter: rate.NewLimiter(rate.Every(5*time.Second), 2),
		Q: func(startTime, endTime time.Time) (interface{}, error) {
			orders, err := q.ExchangeTradeHistoryService.QueryClosedOrders(ctx, symbol, startTime, endTime, lastOrderID)
			return orders, err
		},
		T: func(obj interface{}) time.Time {
			return time.Time(obj.(types.Order).CreationTime)
		},
		ID: func(obj interface{}) string {
			order := obj.(types.Order)
			if order.OrderID > lastOrderID {
				lastOrderID = order.OrderID
			}
			return strconv.FormatUint(order.OrderID, 10)
		},
	}

	c = make(chan types.Order, 100)
	errC = query.Query(ctx, c, startTime, endTime)
	return c, errC
}