From 6eb4e788c7dda251e4cf70f934f692806572fb85 Mon Sep 17 00:00:00 2001
From: suziheng <suziheng@forchange.tech>
Date: Tue, 24 Dec 2024 10:08:38 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E8=BF=81=E7=A7=BB?=
 =?UTF-8?q?=E5=BC=80=E5=85=B3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 model/main.go | 53 ++++++++++++++++++++++++++++-----------------------
 1 file changed, 29 insertions(+), 24 deletions(-)

diff --git a/model/main.go b/model/main.go
index 6d567e17..c71243b6 100644
--- a/model/main.go
+++ b/model/main.go
@@ -116,7 +116,10 @@ func InitDB() {
 		return
 	}
 
+	sqlDB := setDBConns(DB)
 
+	if !config.IsMasterNode {
+		return
 	}
 
 	if common.UsingMySQL {
@@ -132,30 +135,32 @@ func InitDB() {
 }
 
 func migrateDB() error {
-	var err error
-	if err = DB.AutoMigrate(&Channel{}); err != nil {
-		return err
-	}
-	if err = DB.AutoMigrate(&Token{}); err != nil {
-		return err
-	}
-	if err = DB.AutoMigrate(&User{}); err != nil {
-		return err
-	}
-	if err = DB.AutoMigrate(&Option{}); err != nil {
-		return err
-	}
-	if err = DB.AutoMigrate(&Redemption{}); err != nil {
-		return err
-	}
-	if err = DB.AutoMigrate(&Ability{}); err != nil {
-		return err
-	}
-	if err = DB.AutoMigrate(&Log{}); err != nil {
-		return err
-	}
-	if err = DB.AutoMigrate(&Channel{}); err != nil {
-		return err
+	if env.Bool("StartSqlMigration", false) {
+		var err error
+		if err = DB.AutoMigrate(&Channel{}); err != nil {
+			return err
+		}
+		if err = DB.AutoMigrate(&Token{}); err != nil {
+			return err
+		}
+		if err = DB.AutoMigrate(&User{}); err != nil {
+			return err
+		}
+		if err = DB.AutoMigrate(&Option{}); err != nil {
+			return err
+		}
+		if err = DB.AutoMigrate(&Redemption{}); err != nil {
+			return err
+		}
+		if err = DB.AutoMigrate(&Ability{}); err != nil {
+			return err
+		}
+		if err = DB.AutoMigrate(&Log{}); err != nil {
+			return err
+		}
+		if err = DB.AutoMigrate(&Channel{}); err != nil {
+			return err
+		}
 	}
 	return nil
 }