Add enable/disable power price collecting

power_prices/power_prices.go

@@ -11,10 +11,12 @@ import (
 )
 
 type PowerPriceSourceConfigs struct {
+	AllInPower AllInPowerConfig `mapstructure:"all-in-power"`
 	Entsoe EntsoeConfig `mapstructure:"entsoe"`
 }
 
 type Config struct {
+	Enabled bool `mapstructure:"enabled"`
 	Sources PowerPriceSourceConfigs `mapstructure:"sources"`
 }
 
@@ -45,8 +47,13 @@ func New(
 	logger *logrus.Logger,
 	victoriaMetrics *victoria_metrics.VictoriaMetrics,
 ) *PowerPrices {
-	sources = append(sources, newAllInPower())
-	sources = append(sources, newEntsoe(config.Sources.Entsoe))
+	if !config.Sources.AllInPower.Enable {
+		sources = append(sources, newAllInPower(config.Sources.AllInPower))
+	}
+
+	if !config.Sources.Entsoe.Enable {
+		sources = append(sources, newEntsoe(config.Sources.Entsoe))
+	}
 
 	return &PowerPrices{
 		Config: config,
@@ -97,6 +104,11 @@ func (pp *PowerPrices) updateMetrics() error {
 }
 
 func (pp *PowerPrices) Start() {
+	if !pp.Config.Enabled {
+		pp.logger.Warn("Power price collector is disabled")
+		return
+	}
+
 	pp.logger.Info("Starting power price collector")
 
 	err := pp.updateMetrics()

power_prices/source_allinpower.go

@@ -8,15 +8,21 @@ import (
 	"encoding/json"
 )
 
+type AllInPowerConfig struct {
+	Enable bool `mapstructure:"enable"`
+}
+
 type AllInPower struct {
 	name string
 	Client *http.Client
+	Config AllInPowerConfig
 }
 
-func newAllInPower() *AllInPower {
+func newAllInPower(config AllInPowerConfig) *AllInPower {
 	return &AllInPower{
 		name: "all-in-power",
 		Client: &http.Client{},
+		Config: config,
 	}
 }
 

power_prices/source_entsoe.go

@@ -12,6 +12,7 @@ import (
 const entsoeTimestampFormat = "200601021504"
 
 type EntsoeConfig struct {
+	Enable bool `mapstructure:"enable"`
 	SecurityToken string `mapstructure:"security-token"`
 	Domain string `mapstructure:"domain"`
 }