From 301330ff981c739a7dd7834fdcda8de7a5f288d9 Mon Sep 17 00:00:00 2001
From: zhangxin <Xin.zhang@le-it.com.cn>
Date: Fri, 26 Mar 2021 11:04:28 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=9B=E5=BB=BA=E6=95=B0=E6=8D=AE=E5=BA=93en?=
 =?UTF-8?q?gine=E5=90=8E=EF=BC=8C=E8=BF=9B=E8=A1=8C=E8=BF=9E=E6=8E=A5?=
 =?UTF-8?q?=E6=B5=8B=E8=AF=95=EF=BC=8C=E6=B5=8B=E8=AF=95=E4=B8=8D=E9=80=9A?=
 =?UTF-8?q?=E8=BF=87=E8=BF=94=E5=9B=9E=E9=94=99=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 db/db.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

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)
 			}
 		}