Просмотр исходного кода

fix: only master node can migrate database

JustSong 2 лет назад
Родитель
Сommit
567916bd80
2 измененных файлов с 6 добавлено и 0 удалено
  1. 3 0
      common/constants.go
  2. 3 0
      model/main.go

+ 3 - 0
common/constants.go

@@ -1,6 +1,7 @@
 package common
 
 import (
+	"os"
 	"sync"
 	"time"
 
@@ -67,6 +68,8 @@ var PreConsumedQuota = 500
 
 var RootUserEmail = ""
 
+var IsMasterNode = os.Getenv("SYNC_FREQUENCY") == ""
+
 const (
 	RoleGuestUser  = 0
 	RoleCommonUser = 1

+ 3 - 0
model/main.go

@@ -55,6 +55,9 @@ func InitDB() (err error) {
 	}
 	if err == nil {
 		DB = db
+		if !common.IsMasterNode {
+			return nil
+		}
 		err := db.AutoMigrate(&Channel{})
 		if err != nil {
 			return err