options.go

package oracle

import (
	"go.uber.org/zap"

	oraclemetrics "github.com/skip-mev/connect/v2/oracle/metrics"
	"github.com/skip-mev/connect/v2/oracle/types"
	mmclienttypes "github.com/skip-mev/connect/v2/service/clients/marketmap/types"
	mmtypes "github.com/skip-mev/connect/v2/x/marketmap/types"
)

// Option is a functional option for the market map state.
type Option func(impl *OracleImpl)

// WithLogger sets the logger for the oracle.
func WithLogger(logger *zap.Logger) Option {
	return func(m *OracleImpl) {
		if logger == nil {
			panic("logger cannot be nil")
		}

		m.logger = logger.With(zap.String("process", "oracle"))
	}
}

// WithMarketMap sets the market map for the oracle.
func WithMarketMap(marketMap mmtypes.MarketMap) Option {
	return func(m *OracleImpl) {
		if err := marketMap.ValidateBasic(); err != nil {
			panic(err)
		}

		m.marketMap = marketMap
	}
}

// WithLastUpdated sets the last updated for the oracle.
func WithLastUpdated(lastUpdated uint64) Option {
	return func(m *OracleImpl) {
		m.lastUpdated = lastUpdated
	}
}

// WithPriceAPIQueryHandlerFactory sets the Price API query handler factory for the oracle.
// Specifically, this is what is utilized to construct price providers that are API based.
func WithPriceAPIQueryHandlerFactory(factory types.PriceAPIQueryHandlerFactory) Option {
	return func(m *OracleImpl) {
		if factory == nil {
			panic("api query handler factory cannot be nil")
		}

		m.priceAPIFactory = factory
	}
}

// WithPriceWebSocketQueryHandlerFactory sets the websocket query handler factory for the oracle.
// Specifically, this is what is utilized to construct price providers that are websocket based.
func WithPriceWebSocketQueryHandlerFactory(factory types.PriceWebSocketQueryHandlerFactory) Option {
	return func(m *OracleImpl) {
		if factory == nil {
			panic("websocket query handler factory cannot be nil")
		}

		m.priceWSFactory = factory
	}
}

// WithMarketMapperFactory sets the market map factory for the oracle.
// Specifically, this is what is utilized to construct market map providers.
func WithMarketMapperFactory(factory mmclienttypes.MarketMapFactory) Option {
	return func(m *OracleImpl) {
		if factory == nil {
			panic("market map factory cannot be nil")
		}

		m.marketMapperFactory = factory
	}
}

// WithWriteTo sets the file path to which market map updates will be written to. Note that this is optional.
func WithWriteTo(filePath string) Option {
	return func(m *OracleImpl) {
		m.writeTo = filePath
	}
}

// WithPriceProviders allows pre-instantiated price providers to be used in the Oracle's price fetching loop.
// This option is mainly used for testing, but can be useful for programmatically setting customized providers.
func WithPriceProviders(pps ...*types.PriceProvider) Option {
	return func(m *OracleImpl) {
		for _, pp := range pps {
			m.priceProviders[pp.Name()] = ProviderState{Provider: pp}
		}
	}
}

// WithMetrics sets the metrics service the Oracle will use to emit metrics.
func WithMetrics(met oraclemetrics.Metrics) Option {
	return func(m *OracleImpl) {
		m.metrics = met
	}
}