From b697409ec04adb71e4b029cb93297bc3fb044190 Mon Sep 17 00:00:00 2001
From: noreply <noreply@yourdomain.com>
Date: Mon, 14 Oct 2024 20:12:34 +0800
Subject: [PATCH 1/2] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E5=BA=93=E8=BF=9E=E6=8E=A5=E5=A4=B1=E8=B4=A5=E6=8C=82=E8=B5=B7?=
 =?UTF-8?q?=E7=9A=84=E9=80=BB=E8=BE=91=E4=B8=BA10s=E5=90=8E=E9=87=8D?=
 =?UTF-8?q?=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/model/dao/mdb_postgres.go | 17 +++++++++++++----
 src/model/dao/rdb.go          |  8 ++++++--
 2 files changed, 19 insertions(+), 6 deletions(-)

diff --git a/src/model/dao/mdb_postgres.go b/src/model/dao/mdb_postgres.go
index 334b5a8..fbac9ee 100644
--- a/src/model/dao/mdb_postgres.go
+++ b/src/model/dao/mdb_postgres.go
@@ -8,7 +8,6 @@ import (
 	"github.com/assimon/luuu/util/log"
 	"github.com/gookit/color"
 	"github.com/spf13/viper"
-
 	"gorm.io/driver/postgres"
 	"gorm.io/gorm"
 	"gorm.io/gorm/logger"
@@ -36,7 +35,11 @@ func PostgreSQLInit() {
 		},
 	)
 	if err != nil {
-		panic(err)
+		color.Red.Printf("[store_db] postgres open DB,err=%s\n", err)
+		// panic(err)
+		time.Sleep(10 * time.Second)
+		PostgreSQLInit()
+		return
 	}
 
 	if config.AppDebug {
@@ -45,7 +48,10 @@ func PostgreSQLInit() {
 	sqlDB, err := Mdb.DB()
 	if err != nil {
 		color.Red.Printf("[store_db] postgres get DB,err=%s\n", err)
-		panic(err)
+		// panic(err)
+		time.Sleep(10 * time.Second)
+		PostgreSQLInit()
+		return
 	}
 	sqlDB.SetMaxIdleConns(viper.GetInt("postgres_max_idle_conns"))
 	sqlDB.SetMaxOpenConns(viper.GetInt("postgres_max_open_conns"))
@@ -53,7 +59,10 @@ func PostgreSQLInit() {
 	err = sqlDB.Ping()
 	if err != nil {
 		color.Red.Printf("[store_db] postgres connDB err:%s", err.Error())
-		panic(err)
+		// panic(err)
+		time.Sleep(10 * time.Second)
+		PostgreSQLInit()
+		return
 	}
 	log.Sugar.Debug("[store_db] postgres connDB success")
 }
diff --git a/src/model/dao/rdb.go b/src/model/dao/rdb.go
index f979286..a1e990f 100644
--- a/src/model/dao/rdb.go
+++ b/src/model/dao/rdb.go
@@ -3,11 +3,12 @@ package dao
 import (
 	"context"
 	"fmt"
+	"time"
+
 	"github.com/assimon/luuu/util/log"
 	"github.com/go-redis/redis/v8"
 	"github.com/gookit/color"
 	"github.com/spf13/viper"
-	"time"
 )
 
 var Rdb *redis.Client
@@ -34,7 +35,10 @@ func RedisInit() {
 		log.Sugar.Debug("[store_redis] Nil reply returned by Rdb when key does not exist.")
 	} else if err != nil {
 		color.Red.Printf("[store_redis] redis connRdb err,err=%s", err)
-		panic(err)
+		// panic(err)
+		time.Sleep(10 * time.Second)
+		RedisInit()
+		return
 	} else {
 		log.Sugar.Debug("[store_redis] redis connRdb success,suc=%s", pong)
 	}

From 392c922130bc3d64dfb3e29d5abe083b7cb389cf Mon Sep 17 00:00:00 2001
From: noreply <noreply@yourdomain.com>
Date: Mon, 14 Oct 2024 20:14:10 +0800
Subject: [PATCH 2/2] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E5=BA=93=E8=BF=9E=E6=8E=A5=E5=A4=B1=E8=B4=A5=E6=8C=82=E8=B5=B7?=
 =?UTF-8?q?=E7=9A=84=E9=80=BB=E8=BE=91=E4=B8=BA10s=E5=90=8E=E9=87=8D?=
 =?UTF-8?q?=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/model/dao/mdb_mysql.go | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/src/model/dao/mdb_mysql.go b/src/model/dao/mdb_mysql.go
index a8f8f98..bfcf63f 100644
--- a/src/model/dao/mdb_mysql.go
+++ b/src/model/dao/mdb_mysql.go
@@ -25,7 +25,11 @@ func MysqlInit() {
 		Logger: logger.Default.LogMode(logger.Error),
 	})
 	if err != nil {
-		panic(err)
+		color.Red.Printf("[store_db] mysql open DB,err=%s\n", err)
+		// panic(err)
+		time.Sleep(10 * time.Second)
+		MysqlInit()
+		return
 	}
 	if config.AppDebug {
 		Mdb = Mdb.Debug()
@@ -33,7 +37,10 @@ func MysqlInit() {
 	sqlDB, err := Mdb.DB()
 	if err != nil {
 		color.Red.Printf("[store_db] mysql get DB,err=%s\n", err)
-		panic(err)
+		// panic(err)
+		time.Sleep(10 * time.Second)
+		MysqlInit()
+		return
 	}
 	sqlDB.SetMaxIdleConns(viper.GetInt("mysql_max_idle_conns"))
 	sqlDB.SetMaxOpenConns(viper.GetInt("mysql_max_open_conns"))
@@ -41,7 +48,10 @@ func MysqlInit() {
 	err = sqlDB.Ping()
 	if err != nil {
 		color.Red.Printf("[store_db] mysql connDB err:%s", err.Error())
-		panic(err)
+		// panic(err)
+		time.Sleep(10 * time.Second)
+		MysqlInit()
+		return
 	}
 	log.Sugar.Debug("[store_db] mysql connDB success")
 }