mirror of
https://github.com/c9s/bbgo.git
synced 2024-09-20 08:11:08 +00:00
batch: add margin loan/repay/interest batch query
This commit is contained in:
parent
e66eb08db4
commit
c3f2c9eb4a
35
pkg/exchange/batch/margin_interest.go
Normal file
35
pkg/exchange/batch/margin_interest.go
Normal file
|
@ -0,0 +1,35 @@
|
|||
package batch
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"golang.org/x/time/rate"
|
||||
|
||||
"github.com/c9s/bbgo/pkg/types"
|
||||
)
|
||||
|
||||
type MarginInterestBatchQuery struct {
|
||||
types.MarginHistory
|
||||
}
|
||||
|
||||
func (e *MarginInterestBatchQuery) Query(ctx context.Context, asset string, startTime, endTime time.Time) (c chan types.MarginInterest, errC chan error) {
|
||||
query := &AsyncTimeRangedBatchQuery{
|
||||
Type: types.MarginInterest{},
|
||||
Limiter: rate.NewLimiter(rate.Every(5*time.Second), 2),
|
||||
Q: func(startTime, endTime time.Time) (interface{}, error) {
|
||||
return e.QueryInterestHistory(ctx, asset, &startTime, &endTime)
|
||||
},
|
||||
T: func(obj interface{}) time.Time {
|
||||
return time.Time(obj.(types.MarginInterest).Time)
|
||||
},
|
||||
ID: func(obj interface{}) string {
|
||||
interest := obj.(types.MarginInterest)
|
||||
return interest.Time.String()
|
||||
},
|
||||
}
|
||||
|
||||
c = make(chan types.MarginInterest, 100)
|
||||
errC = query.Query(ctx, c, startTime, endTime)
|
||||
return c, errC
|
||||
}
|
36
pkg/exchange/batch/margin_loan.go
Normal file
36
pkg/exchange/batch/margin_loan.go
Normal file
|
@ -0,0 +1,36 @@
|
|||
package batch
|
||||
|
||||
import (
|
||||
"context"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"golang.org/x/time/rate"
|
||||
|
||||
"github.com/c9s/bbgo/pkg/types"
|
||||
)
|
||||
|
||||
type MarginLoanBatchQuery struct {
|
||||
types.MarginHistory
|
||||
}
|
||||
|
||||
func (e *MarginLoanBatchQuery) Query(ctx context.Context, asset string, startTime, endTime time.Time) (c chan types.MarginLoan, errC chan error) {
|
||||
query := &AsyncTimeRangedBatchQuery{
|
||||
Type: types.MarginLoan{},
|
||||
Limiter: rate.NewLimiter(rate.Every(5*time.Second), 2),
|
||||
Q: func(startTime, endTime time.Time) (interface{}, error) {
|
||||
return e.QueryLoanHistory(ctx, asset, &startTime, &endTime)
|
||||
},
|
||||
T: func(obj interface{}) time.Time {
|
||||
return time.Time(obj.(types.MarginRepay).Time)
|
||||
},
|
||||
ID: func(obj interface{}) string {
|
||||
loan := obj.(types.MarginLoan)
|
||||
return strconv.FormatUint(loan.TransactionID, 10)
|
||||
},
|
||||
}
|
||||
|
||||
c = make(chan types.MarginLoan, 100)
|
||||
errC = query.Query(ctx, c, startTime, endTime)
|
||||
return c, errC
|
||||
}
|
36
pkg/exchange/batch/margin_repay.go
Normal file
36
pkg/exchange/batch/margin_repay.go
Normal file
|
@ -0,0 +1,36 @@
|
|||
package batch
|
||||
|
||||
import (
|
||||
"context"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"golang.org/x/time/rate"
|
||||
|
||||
"github.com/c9s/bbgo/pkg/types"
|
||||
)
|
||||
|
||||
type MarginRepayBatchQuery struct {
|
||||
types.MarginHistory
|
||||
}
|
||||
|
||||
func (e *MarginRepayBatchQuery) Query(ctx context.Context, asset string, startTime, endTime time.Time) (c chan types.MarginRepay, errC chan error) {
|
||||
query := &AsyncTimeRangedBatchQuery{
|
||||
Type: types.MarginRepay{},
|
||||
Limiter: rate.NewLimiter(rate.Every(5*time.Second), 2),
|
||||
Q: func(startTime, endTime time.Time) (interface{}, error) {
|
||||
return e.QueryRepayHistory(ctx, asset, &startTime, &endTime)
|
||||
},
|
||||
T: func(obj interface{}) time.Time {
|
||||
return time.Time(obj.(types.MarginRepay).Time)
|
||||
},
|
||||
ID: func(obj interface{}) string {
|
||||
loan := obj.(types.MarginRepay)
|
||||
return strconv.FormatUint(loan.TransactionID, 10)
|
||||
},
|
||||
}
|
||||
|
||||
c = make(chan types.MarginRepay, 100)
|
||||
errC = query.Query(ctx, c, startTime, endTime)
|
||||
return c, errC
|
||||
}
|
Loading…
Reference in New Issue
Block a user