feat: database pool configuration

app/app.go

@@ -156,6 +156,10 @@ func (app *Application) NodeID() string {
 	return config.NODE
 }
 
+func (app *Application) Config() *config.Config {
+	return app.cfg
+}
+
 // Start starts application
 func (app *Application) Start() error {
 	app.mux.Lock()

config.yml

@@ -13,6 +13,8 @@ database:
   username: webhookx
   password:
   database: webhookx
+  max_pool_size: 40                       # The maximum number of connections
+  max_lifetime: 1800                      # The maximum lifetime (in seconds) of a connection
 
 redis:
   host: localhost

config/database.go

@@ -5,11 +5,13 @@ import (
 )
 
 type DatabaseConfig struct {
-	Host     string `yaml:"host" default:"localhost"`
-	Port     uint32 `yaml:"port" default:"5432"`
-	Username string `yaml:"username" default:"webhookx"`
-	Password string `yaml:"password" default:""`
-	Database string `yaml:"database" default:"webhookx"`
+	Host        string `yaml:"host" default:"localhost"`
+	Port        uint32 `yaml:"port" default:"5432"`
+	Username    string `yaml:"username" default:"webhookx"`
+	Password    string `yaml:"password" default:""`
+	Database    string `yaml:"database" default:"webhookx"`
+	MaxPoolSize uint32 `yaml:"max_pool_size" default:"40" envconfig:"MAX_POOL_SIZE"`
+	MaxLifeTime uint   `yaml:"max_life_time" default:"1800" envconfig:"MAX_LIFETIME"`
 }
 
 func (cfg DatabaseConfig) GetDSN() string {

db/db.go

@@ -10,6 +10,7 @@ import (
 	"github.com/webhookx-io/webhookx/db/dao"
 	"github.com/webhookx-io/webhookx/db/transaction"
 	"go.uber.org/zap"
+	"time"
 )
 
 type DB struct {
@@ -33,10 +34,9 @@ type DB struct {
 
 func initSqlxDB(cfg *config.DatabaseConfig) (*sqlx.DB, error) {
 	db, err := sql.Open("postgres", cfg.GetDSN())
-	// db.SetMaxOpenConns(100)
-	// db.SetMaxIdleConns(100)
-	// db.SetConnMaxLifetime(time.Hour)
-	// db.SetConnMaxIdleTime(time.Hour)
+	db.SetMaxOpenConns(int(cfg.MaxPoolSize))
+	db.SetMaxIdleConns(int(cfg.MaxPoolSize))
+	db.SetConnMaxLifetime(time.Second * time.Duration(cfg.MaxLifeTime))
 	if err != nil {
 		return nil, err
 	}

test/cfg/cfg_test.go

@@ -0,0 +1,38 @@
+package cfg
+
+import (
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+	"github.com/stretchr/testify/assert"
+	"github.com/webhookx-io/webhookx/app"
+	"github.com/webhookx-io/webhookx/test/helper"
+	"github.com/webhookx-io/webhookx/utils"
+	"testing"
+)
+
+var _ = Describe("Configuration", Ordered, func() {
+
+	var app *app.Application
+
+	BeforeAll(func() {
+		app = utils.Must(helper.Start(map[string]string{
+			"WEBHOOKX_DATABASE_MAX_POOL_SIZE": "0",
+			"WEBHOOKX_DATABASE_MAX_LIFETIME":  "3600",
+		}))
+	})
+
+	AfterAll(func() {
+		app.Stop()
+	})
+
+	It("database configuration", func() {
+		assert.EqualValues(GinkgoT(), 0, app.Config().Database.MaxPoolSize)
+		assert.EqualValues(GinkgoT(), 3600, app.Config().Database.MaxLifeTime)
+	})
+
+})
+
+func TestConfiguration(t *testing.T) {
+	RegisterFailHandler(Fail)
+	RunSpecs(t, "Configuration Suite")
+}