Fix bug of the http module

dobyte · Feb 19, 2025 · d384dc1 · d384dc1
1 parent fc198d6
commit d384dc1
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 11 deletions.
diff --git a/component/http/proxy.go b/component/http/proxy.go
@@ -15,7 +15,7 @@ func newProxy(s *Server) *Proxy {
 
 // Router 获取路由器
 func (p *Proxy) Router() Router {
-	return &router{app: p.server.app}
+	return &router{app: p.server.app, proxy: p}
 }
 
 // NewMeshClient 新建微服务客户端

diff --git a/component/http/router.go b/component/http/router.go
@@ -32,7 +32,8 @@ type Router interface {
 }
 
 type router struct {
-	app *fiber.App
+	app   *fiber.App
+	proxy *Proxy
 }
 
 // Get 添加GET请求处理器
@@ -96,7 +97,7 @@ func (r *router) Add(methods []string, path string, handler any, middlewares ...
 			handlers = append(handlers, h)
 		case Handler:
 			handlers = append(handlers, func(ctx fiber.Ctx) error {
-				return h(&context{Ctx: ctx})
+				return h(&context{Ctx: ctx, proxy: r.proxy})
 			})
 		}
 	}
@@ -106,7 +107,7 @@ func (r *router) Add(methods []string, path string, handler any, middlewares ...
 		r.app.Add(methods, path, h, handlers...)
 	case Handler:
 		r.app.Add(methods, path, func(ctx fiber.Ctx) error {
-			return h(&context{Ctx: ctx})
+			return h(&context{Ctx: ctx, proxy: r.proxy})
 		}, handlers...)
 	}
 
@@ -124,15 +125,16 @@ func (r *router) Group(prefix string, middlewares ...any) Router {
 			handlers = append(handlers, h)
 		case Handler:
 			handlers = append(handlers, func(ctx fiber.Ctx) error {
-				return h(&context{Ctx: ctx})
+				return h(&context{Ctx: ctx, proxy: r.proxy})
 			})
 		}
 	}
 
-	return &routeGroup{router: r.app.Group(prefix, handlers...)}
+	return &routeGroup{proxy: r.proxy, router: r.app.Group(prefix, handlers...)}
 }
 
 type routeGroup struct {
+	proxy  *Proxy
 	router fiber.Router
 }
 
@@ -197,7 +199,7 @@ func (r *routeGroup) Add(methods []string, path string, handler any, middlewares
 			handlers = append(handlers, h)
 		case Handler:
 			handlers = append(handlers, func(ctx fiber.Ctx) error {
-				return h(&context{Ctx: ctx})
+				return h(&context{Ctx: ctx, proxy: r.proxy})
 			})
 		}
 	}
@@ -207,7 +209,7 @@ func (r *routeGroup) Add(methods []string, path string, handler any, middlewares
 		r.router.Add(methods, path, h, handlers...)
 	case Handler:
 		r.router.Add(methods, path, func(ctx fiber.Ctx) error {
-			return h(&context{Ctx: ctx})
+			return h(&context{Ctx: ctx, proxy: r.proxy})
 		}, handlers...)
 	}
 
@@ -225,10 +227,10 @@ func (r *routeGroup) Group(prefix string, middlewares ...any) Router {
 			handlers = append(handlers, h)
 		case Handler:
 			handlers = append(handlers, func(ctx fiber.Ctx) error {
-				return h(&context{Ctx: ctx})
+				return h(&context{Ctx: ctx, proxy: r.proxy})
 			})
 		}
 	}
 
-	return &routeGroup{router: r.router.Group(prefix, handlers...)}
+	return &routeGroup{router: r.router.Group(prefix, handlers...), proxy: r.proxy}
 }
diff --git a/component/http/server.go b/component/http/server.go
@@ -61,7 +61,7 @@ func NewServer(opts ...Option) *Server {
 		switch handler := o.middlewares[i].(type) {
 		case Handler:
 			s.app.Use(func(ctx fiber.Ctx) error {
-				return handler(&context{Ctx: ctx})
+				return handler(&context{Ctx: ctx, proxy: s.proxy})
 			})
 		case fiber.Handler:
 			s.app.Use(handler)