bbgo_origin/pkg/datasource/coinmarketcap/datasource.go
2022-09-14 02:44:57 +08:00

37 lines
673 B
Go

package coinmarketcap
import (
"context"
v1 "github.com/c9s/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
}