Commit 28d93b57 by huydx Committed by bergquist

(feat) support max connection setting for database configuration

parent 3c7cf3f7
...@@ -73,6 +73,11 @@ password = ...@@ -73,6 +73,11 @@ password =
# Example: mysql://user:secret@host:port/database # Example: mysql://user:secret@host:port/database
url = url =
# Max conn setting default is 0 (mean not set)
max_conn =
max_idle_conn =
max_open_conn =
# For "postgres", use either "disable", "require" or "verify-full" # For "postgres", use either "disable", "require" or "verify-full"
# For "mysql", use either "true", "false", or "skip-verify". # For "mysql", use either "true", "false", or "skip-verify".
ssl_mode = disable ssl_mode = disable
......
...@@ -29,6 +29,9 @@ type DatabaseConfig struct { ...@@ -29,6 +29,9 @@ type DatabaseConfig struct {
ClientKeyPath string ClientKeyPath string
ClientCertPath string ClientCertPath string
ServerCertName string ServerCertName string
MaxConn int
MaxOpenConn int
MaxIdleConn int
} }
var ( var (
...@@ -150,7 +153,15 @@ func getEngine() (*xorm.Engine, error) { ...@@ -150,7 +153,15 @@ func getEngine() (*xorm.Engine, error) {
} }
sqlog.Info("Initializing DB", "dbtype", DbCfg.Type) sqlog.Info("Initializing DB", "dbtype", DbCfg.Type)
return xorm.NewEngine(DbCfg.Type, cnnstr) engine, err := xorm.NewEngine(DbCfg.Type, cnnstr)
if err != nil {
return nil, err
} else {
engine.SetMaxConns(DbCfg.MaxConn)
engine.SetMaxOpenConns(DbCfg.MaxOpenConn)
engine.SetMaxIdleConns(DbCfg.MaxIdleConn)
}
return engine, nil
} }
func LoadConfig() { func LoadConfig() {
...@@ -177,6 +188,9 @@ func LoadConfig() { ...@@ -177,6 +188,9 @@ func LoadConfig() {
DbCfg.Host = sec.Key("host").String() DbCfg.Host = sec.Key("host").String()
DbCfg.Name = sec.Key("name").String() DbCfg.Name = sec.Key("name").String()
DbCfg.User = sec.Key("user").String() DbCfg.User = sec.Key("user").String()
DbCfg.MaxConn = sec.Key("max_conn").MustInt(0)
DbCfg.MaxOpenConn = sec.Key("max_open_conn").MustInt(0)
DbCfg.MaxIdleConn = sec.Key("max_idle_conn").MustInt(0)
if len(DbCfg.Pwd) == 0 { if len(DbCfg.Pwd) == 0 {
DbCfg.Pwd = sec.Key("password").String() DbCfg.Pwd = sec.Key("password").String()
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment