fix: AutoMigrate 注册 Tag/VideoTag/Message/Notification 新表

backend/internal/db/db.go

@@ -1,36 +1,42 @@
-package db
+package db
-
+
-import (
+import (
-	"feedsystem_video_go/internal/account"
+	"feedsystem_video_go/internal/account"
-	"feedsystem_video_go/internal/config"
+	"feedsystem_video_go/internal/config"
-	"feedsystem_video_go/internal/social"
+	"feedsystem_video_go/internal/message"
-	"feedsystem_video_go/internal/video"
+	"feedsystem_video_go/internal/social"
-	"fmt"
+	"feedsystem_video_go/internal/video"
-
+	"feedsystem_video_go/internal/worker"
-	"gorm.io/driver/mysql"
+	"fmt"
-	"gorm.io/gorm"
+
-)
+	"gorm.io/driver/mysql"
-
+	"gorm.io/gorm"
-func NewDB(dbcfg config.DatabaseConfig) (*gorm.DB, error) {
+)
-	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+
-		dbcfg.User, dbcfg.Password, dbcfg.Host, dbcfg.Port, dbcfg.DBName)
+func NewDB(dbcfg config.DatabaseConfig) (*gorm.DB, error) {
-
+	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
-	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
+		dbcfg.User, dbcfg.Password, dbcfg.Host, dbcfg.Port, dbcfg.DBName)
-	if err != nil {
+
-		return nil, err
+	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
-	}
+	if err != nil {
-
+		return nil, err
-	return db, nil
+	}
-}
+
-
+	return db, nil
-func AutoMigrate(db *gorm.DB) error {
+}
-	return db.AutoMigrate(&account.Account{}, &video.Video{}, &video.Like{}, &video.Comment{}, &social.Social{}, &video.OutboxMsg{})
+
-}
+func AutoMigrate(db *gorm.DB) error {
-
+	return db.AutoMigrate(
-func CloseDB(db *gorm.DB) error {
+		&account.Account{}, &video.Video{}, &video.Like{}, &video.Comment{},
-	sqlDB, err := db.DB()
+		&social.Social{}, &video.OutboxMsg{}, &video.Tag{}, &video.VideoTag{},
-	if err != nil {
+		&message.Message{}, &worker.Notification{},
-		return err
+	)
-	}
+}
-	return sqlDB.Close()
+
-}
+func CloseDB(db *gorm.DB) error {
+	sqlDB, err := db.DB()
+	if err != nil {
+		return err
+	}
+	return sqlDB.Close()
+}

backend/internal/http/router.go

@@ -176,7 +176,6 @@ func SetRouter(db *gorm.DB, cache *rediscache.Client, rmq *rabbitmq.RabbitMQ) *g
 	}
 	// message
 	messageRepo := message.NewRepository(db)
-	_ = messageRepo.AutoMigrate(context.Background())
 	messageService := message.NewService(messageRepo)
 	messageHandler := message.NewHandler(messageService)
 	messageGroup := r.Group("/message")