From 8eb99658390b518fa13d66624a954043309a84c0 Mon Sep 17 00:00:00 2001 From: kercylan98 Date: Fri, 5 Jan 2024 09:32:14 +0800 Subject: [PATCH] =?UTF-8?q?other:=20=E4=BC=98=E5=8C=96=20server=20?= =?UTF-8?q?=E5=8C=85=20http=20=E5=8C=85=E8=A3=85=E5=99=A8=E4=BD=BF?= =?UTF-8?q?=E7=94=A8=E4=BD=93=E9=AA=8C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/http_context.go | 6 +++--- server/http_router.go | 6 ++++++ 2 files changed, 9 insertions(+), 3 deletions(-) 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 +}