diff --git a/server/http_context.go b/server/http_context.go index 0af46cd..d50c56b 100644 --- a/server/http_context.go +++ b/server/http_context.go @@ -7,19 +7,19 @@ import ( // NewHttpContext 基于 gin.Context 创建一个新的 HttpContext func NewHttpContext(ctx *gin.Context) *HttpContext { hc := &HttpContext{ - ctx: ctx, + Context: ctx, } return hc } // HttpContext 基于 gin.Context 的 http 请求上下文 type HttpContext struct { - ctx *gin.Context + *gin.Context } // Gin 获取 gin.Context func (slf *HttpContext) Gin() *gin.Context { - return slf.ctx + return slf.Context } // ReadTo 读取请求数据到指定结构体,如果失败则返回错误 diff --git a/server/http_router.go b/server/http_router.go index 7a1c1f2..d6efe09 100644 --- a/server/http_router.go +++ b/server/http_router.go @@ -146,3 +146,9 @@ func (slf *HttpRouter[Context]) Group(relativePath string, handlers ...HandlerFu packer: slf.packer, } } + +// Use 将中间件附加到路由组。 +func (slf *HttpRouter[Context]) Use(middleware ...HandlerFunc[Context]) *HttpRouter[Context] { + slf.group.Use(slf.handlesConvert(middleware)...) + return slf +}