added max db conns

kwilteam · Oct 9, 2024 · c382055 · c382055
1 parent cb0f6f6
commit c382055
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/cmd/kwild/config/config.go b/cmd/kwild/config/config.go
@@ -79,6 +79,7 @@ type AppConfig struct {
 	RPCTimeout         Duration                     `mapstructure:"rpc_timeout"`
 	RPCMaxReqSize      int                          `mapstructure:"rpc_req_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"`
@@ -566,6 +567,7 @@ func DefaultConfig() *KwildConfig {
 			RPCTimeout:           Duration(45 * time.Second),
 			RPCMaxReqSize:        4_200_000,
 			ReadTxTimeout:        Duration(5 * time.Second),
+			MaxDBConnections:     24,
 			Extensions:           make(map[string]map[string]string),
 			Snapshots: SnapshotConfig{
 				Enabled:         false,

diff --git a/cmd/kwild/config/flags.go b/cmd/kwild/config/flags.go
@@ -38,6 +38,7 @@ func AddConfigFlags(flagSet *pflag.FlagSet, cfg *KwildConfig) {
 	flagSet.Var(&cfg.AppCfg.RPCTimeout, "app.rpc-timeout", "timeout for RPC requests (through reading the request, handling the request, and sending the response)")
 	flagSet.IntVar(&cfg.AppCfg.RPCMaxReqSize, "app.rpc-req-limit", cfg.AppCfg.RPCMaxReqSize, "RPC request size limit")
 	flagSet.Var(&cfg.AppCfg.ReadTxTimeout, "app.db-read-timeout", "timeout for database reads initiated by RPC requests")
+	flagSet.Uint32Var(&cfg.AppCfg.MaxDBConnections, "app.db-max-connections", cfg.AppCfg.MaxDBConnections, "maximum number of database connections")
 
 	// Extension endpoints flags
 	flagSet.StringSliceVar(&cfg.AppCfg.ExtensionEndpoints, "app.extension-endpoints", cfg.AppCfg.ExtensionEndpoints, "kwild extension endpoints")

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