Make DB connection max lifetime configurable.

cmd/keymasterd/config.go

@@ -144,6 +144,7 @@ type OpenIDConnectIDPConfig struct {
 
 type ProfileStorageConfig struct {
 	AwsSecretId         string        `yaml:"aws_secret_id"`
+	ConnectionLifetime  time.Duration `yaml:"connection_lifetime"`
 	StorageUrl          string        `yaml:"storage_url"`
 	SyncDelay           time.Duration `yaml:"sync_delay"`
 	SyncInterval        time.Duration `yaml:"sync_interval"`

cmd/keymasterd/storage.go

@@ -23,6 +23,9 @@ const (
 	profileDBFilename = "userProfiles.sqlite3"
 	cachedDBFilename  = "cachedDB.sqlite3"
 
+	dbConnectionLifetimeDefault = time.Minute * 15
+	dbConnectionLifetimeMaximum = time.Hour
+
 	dbSyncDelayDefault = time.Second * 3
 	dbSyncDelayMinimum = time.Second
 	dbSyncDelayMaximum = time.Minute
@@ -47,6 +50,14 @@ func (config *ProfileStorageConfig) setSyncLimits() {
 	} else if config.SyncInterval > dbSyncIntervalMaximum {
 		config.SyncInterval = dbSyncIntervalMaximum
 	}
+	if config.ConnectionLifetime < 1 {
+		config.ConnectionLifetime = dbConnectionLifetimeDefault
+	}
+	if config.ConnectionLifetime < config.SyncInterval {
+		config.ConnectionLifetime = config.SyncInterval
+	} else if config.ConnectionLifetime > dbConnectionLifetimeMaximum {
+		config.ConnectionLifetime = dbConnectionLifetimeMaximum
+	}
 }
 
 func (state *RuntimeState) expandStorageUrl() error {
@@ -130,7 +141,7 @@ func initDBPostgres(state *RuntimeState) (err error) {
 		}
 	}
 	// Ensure that broken connections are replaced.
-	state.db.SetConnMaxLifetime(state.Config.ProfileStorage.SyncInterval >> 1)
+	state.db.SetConnMaxLifetime(state.Config.ProfileStorage.ConnectionLifetime)
 	return nil
 }