diff --git a/cmd/worker/main.go b/cmd/worker/main.go index 681c4b0..baf0402 100644 --- a/cmd/worker/main.go +++ b/cmd/worker/main.go @@ -26,7 +26,7 @@ func main() { fmt.Println("fail to close storage db: ", err) } }() - priceResolver, err := services.NewPriceResolver() + priceResolver, err := services.NewPriceResolver(cfg) if err != nil { panic(err) } diff --git a/internal/services/point_service_test.go b/internal/services/point_service_test.go index cb1509c..1585241 100644 --- a/internal/services/point_service_test.go +++ b/internal/services/point_service_test.go @@ -5,13 +5,14 @@ import ( "github.com/sirupsen/logrus" + "github.com/vultisig/airdrop-registry/config" "github.com/vultisig/airdrop-registry/internal/common" "github.com/vultisig/airdrop-registry/internal/liquidity" "github.com/vultisig/airdrop-registry/internal/models" ) func TestPointService(t *testing.T) { - priceResolver, err := NewPriceResolver() + priceResolver, err := NewPriceResolver(&config.Config{}) if err != nil { t.Errorf("Failed to create price resolver: %v", err) t.FailNow() diff --git a/internal/services/price_service.go b/internal/services/price_service.go index b93684a..ee97c30 100644 --- a/internal/services/price_service.go +++ b/internal/services/price_service.go @@ -12,6 +12,7 @@ import ( "github.com/patrickmn/go-cache" "github.com/sirupsen/logrus" + "github.com/vultisig/airdrop-registry/config" "github.com/vultisig/airdrop-registry/internal/models" ) @@ -26,12 +27,13 @@ type PriceResolver struct { OpenSeaAPIKey string } -func NewPriceResolver() (*PriceResolver, error) { +func NewPriceResolver(cfg *config.Config) (*PriceResolver, error) { pr := &PriceResolver{ logger: logrus.WithField("module", "price_resolver").Logger, lifiBaseAddress: "https://li.quest", coingeckoBaseAddress: "https://api.vultisig.com/coingeicko/api/v3/simple/price", priceCache: *cache.New(4*time.Hour, 5*time.Hour), + OpenSeaAPIKey: cfg.OpenSea.APIKey, } result, err := pr.getCMCMap() if err != nil { diff --git a/internal/services/price_service_test.go b/internal/services/price_service_test.go index e09364a..ae2df8b 100644 --- a/internal/services/price_service_test.go +++ b/internal/services/price_service_test.go @@ -10,10 +10,11 @@ import ( "github.com/patrickmn/go-cache" "github.com/sirupsen/logrus" "github.com/stretchr/testify/assert" + "github.com/vultisig/airdrop-registry/config" ) func TestGetCMCMap(t *testing.T) { - pr, err := NewPriceResolver() + pr, err := NewPriceResolver(&config.Config{}) assert.Nil(t, err) assert.NotNil(t, pr) }