38 lines
991 B
Go
38 lines
991 B
Go
|
package batch
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"strconv"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"git.qtrade.icu/lychiyu/qbtrade/pkg/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
type KLineBatchQuery struct {
|
||
|
|
types.Exchange
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *KLineBatchQuery) Query(ctx context.Context, symbol string, interval types.Interval, startTime, endTime time.Time) (c chan types.KLine, errC chan error) {
|
||
|
|
query := &AsyncTimeRangedBatchQuery{
|
||
|
|
Type: types.KLine{},
|
||
|
|
Limiter: nil, // the rate limiter is handled in the exchange query method
|
||
|
|
Q: func(startTime, endTime time.Time) (interface{}, error) {
|
||
|
|
return e.Exchange.QueryKLines(ctx, symbol, interval, types.KLineQueryOptions{
|
||
|
|
StartTime: &startTime,
|
||
|
|
EndTime: &endTime,
|
||
|
|
})
|
||
|
|
},
|
||
|
|
T: func(obj interface{}) time.Time {
|
||
|
|
return time.Time(obj.(types.KLine).StartTime)
|
||
|
|
},
|
||
|
|
ID: func(obj interface{}) string {
|
||
|
|
kline := obj.(types.KLine)
|
||
|
|
return strconv.FormatInt(kline.StartTime.UnixMilli(), 10)
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
c = make(chan types.KLine, 3000)
|
||
|
|
errC = query.Query(ctx, c, startTime, endTime)
|
||
|
|
return c, errC
|
||
|
|
}
|