wonder
8904330632
实现任务日志 SSE 实时推送功能，解决前端无法流式输出日志的问题 变更内容： - 后端：新增 /task-logs/stream SSE 端点，支持实时日志推送 - 后端：实现 2 秒轮询间隔检测新日志，支持心跳机制 - 后端：支持通过 query 参数 access_token 传递 JWT 认证 - 前端：新增 createTaskLogStream 函数封装 EventSource - 前端：修改 TaskCenter.vue 集成 SSE，支持实时更新和自动滚动 - 前端：添加流式连接状态指示器 关联 commit： - 实现后端 SSE Stream 方法 - 注册 SSE 路由 - 实现前端 SSE 客户端 - 集成 SSE 到任务中心页面
160 lines
7.8 KiB
Go
160 lines
7.8 KiB
Go
package router
|
||
|
||
import (
|
||
"context"
|
||
"github.com/1024XEngineer/xinfra/server/internal/config"
|
||
"github.com/1024XEngineer/xinfra/server/internal/handler"
|
||
"github.com/1024XEngineer/xinfra/server/internal/service"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
swaggerFiles "github.com/swaggo/files"
|
||
ginSwagger "github.com/swaggo/gin-swagger"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
type Dependencies struct {
|
||
Config config.Config
|
||
DB *gorm.DB
|
||
}
|
||
|
||
// New 初始化路由。
|
||
// @title xinfra API
|
||
// @version 1.0
|
||
// @description xinfra 平台后端 API 文档
|
||
// @termsOfService http://swagger.io/terms/
|
||
// @contact.name API Support
|
||
// @contact.url http://www.swagger.io/support
|
||
// @contact.email support@swagger.io
|
||
// @license.name Apache 2.0
|
||
// @license.url http://www.apache.org/licenses/LICENSE-2.0.html
|
||
// @BasePath /api/v1
|
||
// @securityDefinitions.apikey BearerAuth
|
||
// @in header
|
||
// @name Authorization
|
||
// @description 请输入 Bearer Token（例如：Bearer xxx）
|
||
func New(deps Dependencies) *gin.Engine {
|
||
if deps.Config.AppEnv == "prod" {
|
||
gin.SetMode(gin.ReleaseMode)
|
||
}
|
||
|
||
r := gin.New()
|
||
r.Use(gin.Logger(), gin.Recovery())
|
||
|
||
registerSwaggerRoutes(r)
|
||
registerTestRoutes(r)
|
||
registerAuthServerRoutes(r, deps)
|
||
|
||
return r
|
||
}
|
||
|
||
func registerSwaggerRoutes(r *gin.Engine) {
|
||
r.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler, ginSwagger.URL("/swagger/doc.json")))
|
||
}
|
||
|
||
func registerTestRoutes(r *gin.Engine) {
|
||
r.GET("/api/v1/ping", handler.Ping)
|
||
|
||
testGroup := r.Group("/api/v1/test")
|
||
{
|
||
testGroup.GET("/success", handler.TestSuccess)
|
||
testGroup.GET("/error/:code", handler.TestError)
|
||
testGroup.GET("/500", handler.Test500)
|
||
testGroup.GET("/401", handler.Test401)
|
||
testGroup.GET("/403", handler.Test403)
|
||
testGroup.GET("/timeout", handler.TestTimeout)
|
||
testGroup.GET("/paginated", handler.TestPaginated)
|
||
}
|
||
}
|
||
|
||
func registerAuthServerRoutes(r *gin.Engine, deps Dependencies) {
|
||
auditService := service.NewAuditService(deps.DB)
|
||
authService := service.NewAuthService(deps.Config, deps.DB, auditService)
|
||
wayenService := service.NewWayenService(deps.Config, deps.DB)
|
||
wayneRoleBindingService := service.NewWayneRoleBindingService(deps.Config, deps.DB)
|
||
deliveryService := service.NewDeliveryService(deps.Config, deps.DB, auditService)
|
||
if deps.Config.DeliverySchedulerEnabled {
|
||
go deliveryService.Run(context.Background())
|
||
}
|
||
|
||
healthHandler := handler.NewHealthHandler(deps.DB)
|
||
authHandler := handler.NewAuthHandler(deps.Config, authService)
|
||
userHandler := handler.NewUserHandler(deps.DB)
|
||
businessLineHandler := handler.NewBusinessLineHandler(deps.DB, wayneRoleBindingService)
|
||
wayenHandler := handler.NewWayenHandler(deps.DB, wayenService, auditService)
|
||
wayneRoleBindingHandler := handler.NewWayneRoleBindingHandler(wayneRoleBindingService, auditService)
|
||
subsystemAuthHandler := handler.NewSubsystemAuthHandler(deps.DB, wayneRoleBindingService, auditService)
|
||
clouddmHandler := handler.NewCloudDMHandler(deps.Config, auditService)
|
||
samlHandler := handler.NewSAMLHandler(deps.Config, authService)
|
||
oauthHandler := handler.NewOAuthHandler(deps.Config, deps.DB, auditService)
|
||
deliveryHandler := handler.NewDeliveryHandler(deliveryService)
|
||
executionHandler := handler.NewExecutionHandler(deliveryService, deps.Config.DeliveryServiceToken)
|
||
containerServiceHandler := handler.NewContainerServiceHandler(deps.DB, wayneRoleBindingService)
|
||
taskLogHandler := handler.NewTaskLogHandler(deps.DB, deliveryService, wayneRoleBindingService)
|
||
|
||
r.GET("/healthz", healthHandler.Healthz)
|
||
r.GET("/readyz", healthHandler.Readyz)
|
||
r.POST("/api/v1/executions", executionHandler.Create)
|
||
r.GET("/auth/.well-known/openid-configuration", oauthHandler.Discovery)
|
||
r.GET("/auth/oauth/authorize", oauthHandler.Authorize)
|
||
r.POST("/auth/oauth/token", oauthHandler.Token)
|
||
r.GET("/auth/oauth/jwks", oauthHandler.JWKS)
|
||
r.GET("/auth/oauth/userinfo", oauthHandler.UserInfo)
|
||
|
||
v1 := r.Group("/auth/api/v1")
|
||
{
|
||
v1.GET("/config", authHandler.Config)
|
||
v1.POST("/login", authHandler.LocalLogin)
|
||
v1.GET("/login/internal-sso", samlHandler.Login)
|
||
v1.POST("/logout", samlHandler.Logout)
|
||
v1.GET("/saml/metadata", samlHandler.Metadata)
|
||
v1.POST("/saml/acs", samlHandler.ACS)
|
||
|
||
protected := v1.Group("")
|
||
protected.Use(handler.AuthMiddleware(deps.Config))
|
||
protected.GET("/users/me", userHandler.Me)
|
||
protected.GET("/users", userHandler.List)
|
||
protected.GET("/business-lines", businessLineHandler.ListCurrentUserBusinessLines)
|
||
protected.GET("/business-lines/all", businessLineHandler.ListAll)
|
||
protected.POST("/business-lines", businessLineHandler.Create)
|
||
protected.PUT("/business-lines/:id", businessLineHandler.Update)
|
||
protected.DELETE("/business-lines/:id", businessLineHandler.Delete)
|
||
protected.POST("/business-lines/authorizations", businessLineHandler.GrantPermission)
|
||
protected.GET("/business-lines/:id/wayne-namespaces", businessLineHandler.ListWayneNamespaces)
|
||
protected.PUT("/business-lines/:id/wayne-namespaces", businessLineHandler.ReplaceWayneNamespaces)
|
||
protected.GET("/wayen/login", wayenHandler.Login)
|
||
protected.GET("/wayen/credential", wayenHandler.GetCredential)
|
||
protected.PUT("/wayen/credential", wayenHandler.SaveCredential)
|
||
protected.GET("/wayne/namespaces", wayneRoleBindingHandler.ListNamespaces)
|
||
protected.GET("/wayne/groups", wayneRoleBindingHandler.ListGroups)
|
||
protected.GET("/wayne/users/:username/roles", wayneRoleBindingHandler.GetCurrentUserRoles)
|
||
protected.GET("/wayne/namespaces/:namespaceid/operator-permissions", wayneRoleBindingHandler.NamespaceOperatorPermissions)
|
||
protected.GET("/wayne/apps/:appid/operator-permissions", wayneRoleBindingHandler.AppOperatorPermissions)
|
||
protected.PUT("/wayne/namespaces/:namespaceid/roles", wayneRoleBindingHandler.BindNamespace)
|
||
protected.DELETE("/wayne/namespaces/:namespaceid/roles", wayneRoleBindingHandler.UnbindNamespace)
|
||
protected.PUT("/wayne/apps/:appid/roles", wayneRoleBindingHandler.BindApp)
|
||
protected.DELETE("/wayne/apps/:appid/roles", wayneRoleBindingHandler.UnbindApp)
|
||
protected.GET("/subsystem-auth/systems", subsystemAuthHandler.ListSystems)
|
||
protected.GET("/subsystem-auth/wayne/roles", subsystemAuthHandler.ListWayneNamespaceRoles)
|
||
protected.GET("/subsystem-auth/wayne/business-lines/:id/namespaces", subsystemAuthHandler.ListWayneBusinessLineNamespaces)
|
||
protected.GET("/subsystem-auth/wayne/users/:username/roles", subsystemAuthHandler.GetWayneUserRoles)
|
||
protected.PUT("/subsystem-auth/wayne/business-lines/:id/namespaces/:namespaceid/users/:username/roles", subsystemAuthHandler.BindWayneNamespaceRoles)
|
||
protected.DELETE("/subsystem-auth/wayne/business-lines/:id/namespaces/:namespaceid/users/:username/roles", subsystemAuthHandler.UnbindWayneNamespaceRoles)
|
||
protected.POST("/subsystem-auth/wayne/business-lines/:id/users/:userid/init", subsystemAuthHandler.InitWayneBusinessLineUser)
|
||
protected.GET("/container-services/business-lines/:id/summary", containerServiceHandler.Summary)
|
||
protected.GET("/container-services/business-lines/:id/workloads", containerServiceHandler.Workloads)
|
||
protected.GET("/clouddm/login", clouddmHandler.Login)
|
||
protected.GET("/delivery/targets", deliveryHandler.Targets)
|
||
protected.PUT("/delivery/quotas", deliveryHandler.UpsertQuota)
|
||
protected.POST("/delivery/mysql", deliveryHandler.CreateMySQL)
|
||
protected.GET("/delivery/tasks", deliveryHandler.List)
|
||
protected.GET("/delivery/tasks/:id", deliveryHandler.Get)
|
||
protected.POST("/delivery/tasks/:id/cancel", deliveryHandler.Cancel)
|
||
protected.POST("/delivery/tasks/:id/rollback/retry", deliveryHandler.RetryRollback)
|
||
protected.POST("/delivery/tasks/:id/rollback/release", deliveryHandler.AcknowledgeRollbackRelease)
|
||
protected.POST("/delivery/tasks/:id/clouddm/retry", deliveryHandler.RetryCloudDMRegistration)
|
||
protected.GET("/task-logs", taskLogHandler.List)
|
||
protected.GET("/task-logs/:id", taskLogHandler.Get)
|
||
protected.GET("/task-logs/stream", taskLogHandler.Stream)
|
||
}
|
||
}
|