added max db connections config (#1068)

* added max db connections config * fix lint
kwilteam · Oct 21, 2024 · aa94e4c · aa94e4c
1 parent e8510e5
commit aa94e4c
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 1 deletion.
diff --git a/cmd/common.go b/cmd/common.go
@@ -79,6 +79,7 @@ var DefaultConfig = func() *commonConfig.KwildConfig {
 			ChallengeExpiry:      commonConfig.Duration(10 * time.Second),
 			ChallengeRateLimit:   10.0, // req/s
 			ReadTxTimeout:        commonConfig.Duration(5 * time.Second),
+			MaxDBConnections:     24,
 			Extensions:           make(map[string]map[string]string),
 			Snapshots: commonConfig.SnapshotConfig{
 				Enable:          false,

diff --git a/cmd/kwild/config/flags.go b/cmd/kwild/config/flags.go
@@ -59,6 +59,7 @@ func AddConfigFlags(flagSet *pflag.FlagSet, cfg *config.KwildConfig) {
 	flagSet.MarkDeprecated("app.rpc-req-limit", "use --app.rpc-max-req-size instead")
 
 	flagSet.Var(&cfg.AppConfig.ReadTxTimeout, "app.db-read-timeout", "timeout for database reads initiated by RPC requests")
+	flagSet.Uint32Var(&cfg.AppConfig.MaxDBConnections, "app.db-max-connections", cfg.AppConfig.MaxDBConnections, "maximum number of database connections")
 
 	// genesis state / migration flags
 	flagSet.StringVar(&cfg.AppConfig.GenesisState, "app.genesis-state", cfg.AppConfig.GenesisState, "Path to the genesis state file")

diff --git a/cmd/kwild/server/build.go b/cmd/kwild/server/build.go
@@ -557,7 +557,7 @@ func buildDB(d *coreDependencies, closer *closeFuncs) *pg.DB {
 	// If yes, restore the database from the snapshot
 	fromSnapshot := restoreDB(d)
 
-	db, err := d.dbOpener(d.ctx, d.cfg.AppConfig.DBName, 24)
+	db, err := d.dbOpener(d.ctx, d.cfg.AppConfig.DBName, d.cfg.AppConfig.MaxDBConnections)
 	if err != nil {
 		failBuild(err, "kwild database open failed")
 	}

diff --git a/common/config/config.go b/common/config/config.go
@@ -72,6 +72,7 @@ type AppConfig struct {
 	ChallengeExpiry    Duration                     `mapstructure:"challenge_expiry"`
 	ChallengeRateLimit float64                      `mapstructure:"challenge_rate_limit"`
 	ReadTxTimeout      Duration                     `mapstructure:"db_read_timeout"`
+	MaxDBConnections   uint32                       `mapstructure:"db_max_connections"`
 	ExtensionEndpoints []string                     `mapstructure:"extension_endpoints"`
 	AdminRPCPass       string                       `mapstructure:"admin_pass"`
 	NoTLS              bool                         `mapstructure:"admin_notls"`