Browse Source

add reload funtion when execute command was changed

tags/v1.0.8
liangzy 3 years ago
parent
commit
89d91c6256
1 changed files with 13 additions and 4 deletions
  1. +13
    -4
      db/db.go

+ 13
- 4
db/db.go View File

@@ -6,6 +6,7 @@ import (
_ "github.com/lib/pq" _ "github.com/lib/pq"
"golib.gaore.com/GaoreGo/grconfig" "golib.gaore.com/GaoreGo/grconfig"
"golib.gaore.com/GaoreGo/grlogs" "golib.gaore.com/GaoreGo/grlogs"
"path"
) )


const DEFAULT = "default" const DEFAULT = "default"
@@ -33,9 +34,12 @@ func (d *DbItem) Dsn() string {
} }


func init() { func init() {
Reload()
}


func Reload(dir ...string) {
var err error var err error
var configer *grconfig.Configer
logger := grlogs.GetAli("sql") logger := grlogs.GetAli("sql")


defer func() { defer func() {
@@ -44,16 +48,21 @@ func init() {
} }
}() }()


dsns, err := GetDsn()
if len(dir) == 0 {
configer = grconfig.New("conf/")
} else {
dir = append(dir, "conf/")
configer = grconfig.New(path.Join(dir...))
}
dsns, err := GetDsn(configer)
if err = LoadXorm(dsns); err != nil { if err = LoadXorm(dsns); err != nil {
logger.Critical(err.Error()) logger.Critical(err.Error())
} }
} }


func GetDsn() (DsnList map[string]dsnConfig, err error) {
func GetDsn(configer *grconfig.Configer) (DsnList map[string]dsnConfig, err error) {
var items map[string]*DbItem var items map[string]*DbItem
DsnList = make(map[string]dsnConfig) DsnList = make(map[string]dsnConfig)
configer := grconfig.New("conf/")
err = configer.Item("db", &items) err = configer.Item("db", &items)
if err != nil { if err != nil {
return return


Loading…
Cancel
Save