nepiedg
bbc2f5f1d5
- 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.
44 lines
1.2 KiB
Go
44 lines
1.2 KiB
Go
package routes
|
||
|
||
import (
|
||
"net/http"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"gorm.io/gorm"
|
||
|
||
"wx_service/internal/handler"
|
||
"wx_service/internal/middleware"
|
||
rmhandler "wx_service/internal/remove_watermark/handler"
|
||
smokehandler "wx_service/internal/smoke/handler"
|
||
)
|
||
|
||
func Register(
|
||
router *gin.Engine,
|
||
db *gorm.DB,
|
||
authHandler *handler.AuthHandler,
|
||
videoHandler *rmhandler.VideoHandler,
|
||
smokeHandler *smokehandler.SmokeHandler,
|
||
) {
|
||
// Register 用来集中注册所有 HTTP 路由，便于工程结构更清晰：
|
||
// - main 只负责初始化（配置/DB/依赖注入）
|
||
// - routes 只负责把 URL 映射到 handler
|
||
api := router.Group("/api/v1")
|
||
{
|
||
// 登录接口：用微信 code 换取/创建用户并返回 session_key（作为后续 Bearer Token）
|
||
api.POST("/auth/login", authHandler.LoginWithWeChat)
|
||
|
||
// 需要登录的接口组：统一挂载鉴权中间件
|
||
protected := api.Group("")
|
||
protected.Use(middleware.AuthMiddleware(db))
|
||
{
|
||
registerRemoveWatermarkRoutes(protected, videoHandler)
|
||
registerSmokeRoutes(protected, smokeHandler)
|
||
}
|
||
}
|
||
|
||
// 健康检查：用于容器/负载均衡探活
|
||
router.GET("/healthz", func(c *gin.Context) {
|
||
c.JSON(http.StatusOK, gin.H{"status": "ok"})
|
||
})
|
||
}
|