huanghaosheng/VLoop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Sisyphus
2026-04-25 20:22:03 +08:00
parent 18245c06bf
commit a5452e3d68
2 changed files with 42 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
backend/internal/db/db.go
View File

@@ -3,8 +3,10 @@ package db
import (
"feedsystem_video_go/internal/account"
"feedsystem_video_go/internal/config"
"feedsystem_video_go/internal/message"
"feedsystem_video_go/internal/social"
"feedsystem_video_go/internal/video"
"feedsystem_video_go/internal/worker"
"fmt"
"gorm.io/driver/mysql"
@@ -24,7 +26,11 @@ func NewDB(dbcfg config.DatabaseConfig) (*gorm.DB, error) {
}
func AutoMigrate(db *gorm.DB) error {
return db.AutoMigrate(&account.Account{}, &video.Video{}, &video.Like{}, &video.Comment{}, &social.Social{}, &video.OutboxMsg{})
return db.AutoMigrate(
&account.Account{}, &video.Video{}, &video.Like{}, &video.Comment{},
&social.Social{}, &video.OutboxMsg{}, &video.Tag{}, &video.VideoTag{},
&message.Message{}, &worker.Notification{},
)
}
func CloseDB(db *gorm.DB) error {

1
backend/internal/http/router.go
View File

@@ -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")