增加三个数据库自动重连功能

db/checkConnection.go

@@ -0,0 +1,30 @@
+package db
+
+import "time"
+
+// StartCheckConnection 自动重连数据库(5S), dbType = 0 - oracle; 1 - mysql; 2 - mongodb
+func StartCheckConnection(dbType int) {
+	switch dbType {
+	case 0: // oracle
+		for {
+			if err := GetEngine().Ping(); err != nil {
+				InitDbEngine()
+			}
+			time.Sleep(time.Duration(5000) * time.Millisecond)
+		}
+	case 1: // mysql
+		for {
+			if err := GetMySQLEngine().Ping(); err != nil {
+				InitMySQLDbEngine()
+			}
+			time.Sleep(time.Duration(5000) * time.Millisecond)
+		}
+	case 2: // mongodb
+		for {
+			if err := GetMongoDB().Session.Ping(); err != nil {
+				InitMongoDB()
+			}
+			time.Sleep(time.Duration(5000) * time.Millisecond)
+		}
+	}
+}

main.go

@@ -51,6 +51,7 @@ func main() {
 		logger.GetLogger().Errorf("init db engine failed:", err.Error())
 		return
 	}
+	go db.StartCheckConnection(0)
 	defer db.Close()
 
 	// 初始化MongoDB引擎
@@ -59,6 +60,7 @@ func main() {
 		logger.GetLogger().Errorf("init mongodb engine failed:", err.Error())
 		return
 	}
+	go db.StartCheckConnection(2)
 	defer db.CloseMongoDB()
 
 	// 初始化MySQL引擎
@@ -67,6 +69,7 @@ func main() {
 		logger.GetLogger().Errorf("init mysql engine failed:", err.Error())
 		return
 	}
+	go db.StartCheckConnection(1)
 	defer db.CloseMySQL()
 
 	// 初始化Redis客户端