Refactor video handling and integrate new services
- Removed legacy video handling code and models to streamline the codebase. - Updated main.go to include new services for removing watermarks and smoke logging. - Enhanced route registration to accommodate new handlers for watermark removal and smoke logging. - Improved database migration to include new models for watermark processing.
This commit is contained in:
+19
-4
@@ -9,8 +9,14 @@ import (
|
||||
"wx_service/internal/database"
|
||||
"wx_service/internal/handler"
|
||||
"wx_service/internal/model"
|
||||
rmhandler "wx_service/internal/remove_watermark/handler"
|
||||
rmmodel "wx_service/internal/remove_watermark/model"
|
||||
rmservice "wx_service/internal/remove_watermark/service"
|
||||
"wx_service/internal/routes"
|
||||
"wx_service/internal/service"
|
||||
smokehandler "wx_service/internal/smoke/handler"
|
||||
smokemodel "wx_service/internal/smoke/model"
|
||||
smokeservice "wx_service/internal/smoke/service"
|
||||
)
|
||||
|
||||
func main() {
|
||||
@@ -22,7 +28,13 @@ func main() {
|
||||
log.Fatalf("init database failed: %v", err)
|
||||
}
|
||||
// 3) 自动建表/迁移(开发阶段很方便;生产环境可改为手动迁移)
|
||||
if err := database.AutoMigrate(&model.MiniProgram{}, &model.User{}, &model.VideoParseLog{}, &model.VideoParseUnlock{}); err != nil {
|
||||
if err := database.AutoMigrate(
|
||||
&model.MiniProgram{},
|
||||
&model.User{},
|
||||
&rmmodel.VideoParseLog{},
|
||||
&rmmodel.VideoParseUnlock{},
|
||||
&smokemodel.SmokeLog{},
|
||||
); err != nil {
|
||||
log.Fatalf("auto migrate failed: %v", err)
|
||||
}
|
||||
|
||||
@@ -34,14 +46,17 @@ func main() {
|
||||
miniProgramService := service.NewMiniProgramService(database.DB)
|
||||
authService := service.NewAuthService(database.DB, miniProgramService)
|
||||
authHandler := handler.NewAuthHandler(authService)
|
||||
videoService, err := service.NewVideoService(database.DB, config.AppConfig.ShortVideo)
|
||||
videoService, err := rmservice.NewVideoService(database.DB, config.AppConfig.ShortVideo)
|
||||
if err != nil {
|
||||
log.Fatalf("init video service failed: %v", err)
|
||||
}
|
||||
videoHandler := handler.NewVideoHandler(videoService)
|
||||
videoHandler := rmhandler.NewVideoHandler(videoService)
|
||||
|
||||
smokeLogService := smokeservice.NewSmokeLogService(database.DB)
|
||||
smokeHandler := smokehandler.NewSmokeHandler(smokeLogService)
|
||||
|
||||
// 6) 注册路由:把 URL 映射到 handler
|
||||
routes.Register(router, database.DB, authHandler, videoHandler)
|
||||
routes.Register(router, database.DB, authHandler, videoHandler, smokeHandler)
|
||||
|
||||
// 7) 启动监听端口
|
||||
addr := ":" + config.AppConfig.Server.Port
|
||||
|
||||
Reference in New Issue
Block a user