37 lines
681 B
Go
37 lines
681 B
Go
package coinmarketcap
|
|
|
|
import (
|
|
"context"
|
|
|
|
v1 "git.qtrade.icu/lychiyu/bbgo/pkg/datasource/coinmarketcap/v1"
|
|
)
|
|
|
|
type DataSource struct {
|
|
client *v1.RestClient
|
|
}
|
|
|
|
func New(apiKey string) *DataSource {
|
|
client := v1.New()
|
|
client.Auth(apiKey)
|
|
return &DataSource{client: client}
|
|
}
|
|
|
|
func (d *DataSource) QueryMarketCapInUSD(ctx context.Context, limit int) (map[string]float64, error) {
|
|
req := v1.ListingsLatestRequest{
|
|
Client: d.client,
|
|
Limit: &limit,
|
|
}
|
|
|
|
resp, err := req.Do(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
marketcaps := make(map[string]float64)
|
|
for _, data := range resp {
|
|
marketcaps[data.Symbol] = data.Quote["USD"].MarketCap
|
|
}
|
|
|
|
return marketcaps, nil
|
|
}
|