feat: make keep alive interval configurable

internal/config/config.go

@@ -14,12 +14,16 @@ import (
 	"time"
 )
 
-const (
+var (
 	KeepAliveInterval = 1 * time.Minute
 )
 
 func LoadConfig(path string) (*Config, error) {
-	config := defaultConfig()
+	config, err := defaultConfig()
+
+	if err != nil {
+		return nil, err
+	}
 
 	if len(path) != 0 {
 		expandedPath, err := homedir.Expand(path)
@@ -42,6 +46,8 @@ func LoadConfig(path string) (*Config, error) {
 		}
 	}
 
+	KeepAliveInterval = config.PollNet.KeepAliveInterval
+
 	return config, nil
 }
 
@@ -61,6 +67,7 @@ const (
 	tlsAcmeCAKey                = "IONSCALE_TLS_ACME_CA"
 	tlsAcmeEmailKey             = "IONSCALE_TLS_ACME_EMAIL"
 	tlsAcmePath                 = "IONSCALE_TLS_ACME_PATH"
+	pollNetKeepAliveInterval    = "IONSCALE_POLL_NET_KEEP_ALIVE_INTERVAL"
 	metricsListenAddrKey        = "IONSCALE_METRICS_LISTEN_ADDR"
 	loggingLevelKey             = "IONSCALE_LOGGING_LEVEL"
 	loggingFormatKey            = "IONSCALE_LOGGING_FORMAT"
@@ -71,8 +78,13 @@ const (
 	authProviderScopesKey       = "IONSCALE_AUTH_PROVIDER_SCOPES"
 )
 
-func defaultConfig() *Config {
-	return &Config{
+func defaultConfig() (*Config, error) {
+	defaultKeepAliveInterval, err := GetDuration(pollNetKeepAliveInterval, 1*time.Minute)
+	if err != nil {
+		return nil, err
+	}
+
+	c := &Config{
 		HttpListenAddr:    GetString(httpListenAddrKey, ":8080"),
 		HttpsListenAddr:   GetString(httpsListenAddrKey, ":8443"),
 		MetricsListenAddr: GetString(metricsListenAddrKey, ":9091"),
@@ -95,6 +107,9 @@ func defaultConfig() *Config {
 			AcmeEmail:   GetString(tlsAcmeEmailKey, ""),
 			AcmePath:    GetString(tlsAcmePath, ""),
 		},
+		PollNet: PollNet{
+			KeepAliveInterval: defaultKeepAliveInterval,
+		},
 		AuthProvider: AuthProvider{
 			Issuer:       GetString(authProviderIssuerKey, ""),
 			ClientID:     GetString(authProviderClientIdKey, ""),
@@ -107,6 +122,7 @@ func defaultConfig() *Config {
 			File:   GetString(loggingFileKey, ""),
 		},
 	}
+	return c, nil
 }
 
 type ServerKeys struct {
@@ -121,6 +137,7 @@ type Config struct {
 	MetricsListenAddr string       `yaml:"metrics_listen_addr,omitempty"`
 	ServerUrl         string       `yaml:"server_url,omitempty"`
 	Tls               Tls          `yaml:"tls,omitempty"`
+	PollNet           PollNet      `yaml:"poll_net,omitempty"`
 	Keys              Keys         `yaml:"keys,omitempty"`
 	Database          Database     `yaml:"database,omitempty"`
 	AuthProvider      AuthProvider `yaml:"auth_provider,omitempty"`
@@ -138,6 +155,10 @@ type Tls struct {
 	AcmePath    string `yaml:"acme_path,omitempty"`
 }
 
+type PollNet struct {
+	KeepAliveInterval time.Duration `yaml:"keep_alive_interval"`
+}
+
 type Logging struct {
 	Level  string `yaml:"level,omitempty"`
 	Format string `yaml:"format,omitempty"`

internal/config/funcs.go

@@ -3,6 +3,7 @@ package config
 import (
 	"os"
 	"strings"
+	"time"
 )
 
 func GetBool(key string, defaultValue bool) bool {
@@ -28,3 +29,15 @@ func GetStrings(key string, defaultValue []string) []string {
 	}
 	return defaultValue
 }
+
+func GetDuration(key string, defaultValue time.Duration) (time.Duration, error) {
+	v := os.Getenv(key)
+	if len(v) > 0 {
+		duration, err := time.ParseDuration(v)
+		if err != nil {
+			return defaultValue, err
+		}
+		return duration, nil
+	}
+	return defaultValue, nil
+}