Add option for refresh interval

config.go

@@ -13,6 +13,7 @@ import (
 const (
 	envVarListenAddress       = "NETATMO_EXPORTER_ADDR"
 	envVarLogLevel            = "NETATMO_LOG_LEVEL"
+	envVarRefreshInterval     = "NETATMO_REFRESH_INTERVAL"
 	envVarStaleDuration       = "NETATMO_AGE_STALE"
 	envVarNetatmoClientID     = "NETATMO_CLIENT_ID"
 	envVarNetatmoClientSecret = "NETATMO_CLIENT_SECRET"
@@ -21,12 +22,14 @@ const (
 
 	flagListenAddress       = "addr"
 	flagLogLevel            = "log-level"
+	flagRefreshInterval     = "refresh-interval"
 	flagStaleDuration       = "age-stale"
 	flagNetatmoClientID     = "client-id"
 	flagNetatmoClientSecret = "client-secret"
 	flagNetatmoUsername     = "username"
 	flagNetatmoPassword     = "password"
 
+	defaultRefreshInterval = 8 * time.Minute
 	defaultStaleDuration   = 30 * time.Minute
 )
 
@@ -34,6 +37,7 @@ var (
 	defaultConfig = config{
 		Addr:            ":9210",
 		LogLevel:        logLevel(logrus.InfoLevel),
+		RefreshInterval: defaultRefreshInterval,
 		StaleDuration:   defaultStaleDuration,
 	}
 
@@ -68,6 +72,7 @@ func (l *logLevel) Set(value string) error {
 type config struct {
 	Addr            string
 	LogLevel        logLevel
+	RefreshInterval time.Duration
 	StaleDuration   time.Duration
 	Netatmo         netatmo.Config
 }
@@ -82,6 +87,7 @@ func parseConfig(args []string, getenv func(string) string) (config, error) {
 	flagSet := pflag.NewFlagSet(args[0], pflag.ExitOnError)
 	flagSet.StringVarP(&cfg.Addr, flagListenAddress, "a", cfg.Addr, "Address to listen on.")
 	flagSet.Var(&cfg.LogLevel, flagLogLevel, "Sets the minimum level output through logging.")
+	flagSet.DurationVar(&cfg.RefreshInterval, flagRefreshInterval, cfg.RefreshInterval, "Time interval used for internal caching of NetAtmo sensor data.")
 	flagSet.DurationVar(&cfg.StaleDuration, flagStaleDuration, cfg.StaleDuration, "Data age to consider as stale. Stale data does not create metrics anymore.")
 	flagSet.StringVarP(&cfg.Netatmo.ClientID, flagNetatmoClientID, "i", cfg.Netatmo.ClientID, "Client ID for NetAtmo app.")
 	flagSet.StringVarP(&cfg.Netatmo.ClientSecret, flagNetatmoClientSecret, "s", cfg.Netatmo.ClientSecret, "Client secret for NetAtmo app.")
@@ -113,6 +119,10 @@ func parseConfig(args []string, getenv func(string) string) (config, error) {
 		return config{}, errNoNetatmoPassword
 	}
 
+	if cfg.StaleDuration < cfg.RefreshInterval {
+		return config{}, fmt.Errorf("stale duration smaller than refresh interval: %s < %s", cfg.StaleDuration, cfg.RefreshInterval)
+	}
+
 	return cfg, nil
 }
 
@@ -127,6 +137,15 @@ func applyEnvironment(cfg *config, getenv func(string) string) error {
 		}
 	}
 
+	if envRefreshInterval := getenv(envVarRefreshInterval); envRefreshInterval != "" {
+		duration, err := time.ParseDuration(envRefreshInterval)
+		if err != nil {
+			return err
+		}
+
+		cfg.RefreshInterval = duration
+	}
+
 	if envStaleDuration := getenv(envVarStaleDuration); envStaleDuration != "" {
 		duration, err := time.ParseDuration(envStaleDuration)
 		if err != nil {

config_test.go

@@ -41,6 +41,7 @@ func TestParseConfig(t *testing.T) {
 			wantConfig: config{
 				Addr:            defaultConfig.Addr,
 				LogLevel:        logLevel(logrus.InfoLevel),
+				RefreshInterval: defaultRefreshInterval,
 				StaleDuration:   defaultStaleDuration,
 				Netatmo: netatmo.Config{
 					ClientID:     "id",
@@ -59,6 +60,7 @@ func TestParseConfig(t *testing.T) {
 			env: map[string]string{
 				envVarListenAddress:       ":8080",
 				envVarLogLevel:            "debug",
+				envVarRefreshInterval:     "5m",
 				envVarStaleDuration:       "10m",
 				envVarNetatmoClientID:     "id",
 				envVarNetatmoClientSecret: "secret",
@@ -68,6 +70,7 @@ func TestParseConfig(t *testing.T) {
 			wantConfig: config{
 				Addr:            ":8080",
 				LogLevel:        logLevel(logrus.DebugLevel),
+				RefreshInterval: 5 * time.Minute,
 				StaleDuration:   10 * time.Minute,
 				Netatmo: netatmo.Config{
 					ClientID:     "id",