diff --git a/db/db.go b/db/db.go index c7459f3..77a2abc 100644 --- a/db/db.go +++ b/db/db.go @@ -46,6 +46,9 @@ func InitDb() error { fmt.Printf("err1 is %v", err) return err } + if err = master.Ping(); err != nil { + return err + } case "mysql": driveSource := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8", v.User, v.Pwd, v.Host, v.Port, v.DbName) @@ -54,6 +57,10 @@ func InitDb() error { fmt.Printf("err2 is %v", err) return err } + if err = master.Ping(); err != nil { + return err + } + } } else { switch v.DriverName { @@ -66,6 +73,9 @@ func InitDb() error { fmt.Printf("err1 is %v", err) return err } + if err = slave.Ping(); err != nil { + return err + } slaves = append(slaves, slave) case "mysql": driveSource := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8", @@ -75,6 +85,9 @@ func InitDb() error { fmt.Printf("err2 is %v", err) return err } + if err = slave.Ping(); err != nil { + return err + } slaves = append(slaves, slave) } }