Add MaxConns and Idle configurations to dbo

Author: christianezeani

dbo/connect.go

@@ -1,6 +1,8 @@
 package dbo
 
 import (
+	"time"
+
 	"gorm.io/gorm"
 )
 
@@ -18,6 +20,30 @@ func Connect(opts *Options) (i Instance, err error) {
 		return
 	}
 
+	sqldb, err := db.DB()
+
+	if err != nil {
+		return
+	}
+
+	if opts.MaxOpenConns > 0 {
+		sqldb.SetMaxOpenConns(opts.MaxOpenConns)
+	}
+
+	if opts.MaxIdleConns > 0 {
+		sqldb.SetMaxIdleConns(opts.MaxIdleConns)
+	}
+
+	if opts.ConnMaxLifetime > 0 {
+		duration := time.Millisecond * time.Duration(opts.ConnMaxLifetime)
+		sqldb.SetConnMaxLifetime(duration)
+	}
+
+	if opts.ConnMaxIdleTime > 0 {
+		duration := time.Millisecond * time.Duration(opts.ConnMaxIdleTime)
+		sqldb.SetConnMaxIdleTime(duration)
+	}
+
 	db.Use(NewPlugin())
 
 	switch dbDriver(opts) {

dbo/options.go

@@ -20,17 +20,21 @@ const (
 )
 
 type Options struct {
-	Driver    DRIVER
-	Host      string
-	Port      string
-	Username  string
-	Password  string
-	DBName    string
-	Charset   string
-	Collation string
-	DSN       string
-	Engine    ENGINE
-	Config    *gorm.Config
+	Driver          DRIVER
+	Host            string
+	Port            string
+	Username        string
+	Password        string
+	DBName          string
+	Charset         string
+	Collation       string
+	DSN             string
+	Engine          ENGINE
+	Config          *gorm.Config
+	MaxOpenConns    int
+	MaxIdleConns    int
+	ConnMaxLifetime int // Maximum lifetime for a connection (in milliseconds)
+	ConnMaxIdleTime int // Maximum idle time for a connection (in milliseconds)
 }
 
 func (c *Options) getConfig() *gorm.Config {