fix: 미들웨어가 의도되지 않은 시점에서 실행되는 부분 해결

unsafe-risk · Jul 5, 2022 · 63a4dab · 63a4dab
1 parent ea05df0
commit 63a4dab
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/router_lambda.go b/router_lambda.go
@@ -31,7 +31,10 @@ func (lr *lambdaRouter) AddRoute(method string, path string, handler HandlerFunc
 
 	method = replaceMethodWildcardToBlank(method)
 	r.params.setParamsInfo(method, findParamKeys(path))
-	r.handlers.setHandler(method, wrapMiddleware(handler, middleware...))
+	r.handlers.setHandler(method, func(c Context) error {
+		wrappedHandler := wrapMiddleware(handler, middleware...)
+		return wrappedHandler(c)
+	})
 }
 
 func (lr *lambdaRouter) DelRoute(method string, path string) {

diff --git a/router_local.go b/router_local.go
@@ -142,7 +142,10 @@ func (lr *localRouter) AddRoute(method string, path string, handler HandlerFunc,
 
 	method = replaceMethodWildcardToBlank(method)
 	(*targetRoute).params.setParamsInfo(method, pathParams)
-	(*targetRoute).handlers.setHandler(method, wrapMiddleware(handler, middleware...))
+	(*targetRoute).handlers.setHandler(method, func(c Context) error {
+		wrappedHandler := wrapMiddleware(handler, middleware...)
+		return wrappedHandler(c)
+	})
 }
 
 func (lr *localRouter) DelRoute(method string, path string) {