feat: after middlewares

tupatech · Apr 30, 2024 · d766703 · d766703
1 parent be035cd
commit d766703
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 4 deletions.
diff --git a/tupa.go b/tupa.go
@@ -209,6 +209,24 @@ func (a *APIServer) MakeHTTPHandlerFuncHelper(routeInfo RouteInfo) http.HandlerF
 		} else {
 			WriteJSONHelper(w, http.StatusMethodNotAllowed, APIError{Error: "Método HTTP não permitido"})
 		}
+
+		allAfterMiddlewares := MiddlewareChain{}
+		allAfterMiddlewares = append(allAfterMiddlewares, routeInfo.AfterMiddlewares...)
+
+		doneCh = a.executeMiddlewaresAsync(ctx, allAfterMiddlewares)
+		errorsSlice = <-doneCh
+
+		if len(errorsSlice) > 0 {
+			err := errorsSlice[0]
+			if apiErr, ok := err.(APIHandlerErr); ok {
+				slog.Error("API Error", "err:", apiErr, "status:", apiErr.Status)
+				WriteJSONHelper(w, apiErr.Status, APIError{Error: apiErr.Error()})
+			} else {
+				slog.Error("API Error", "err:", apiErr, "status:", apiErr.Status)
+				WriteJSONHelper(w, http.StatusInternalServerError, APIError{Error: err.Error()})
+			}
+			return
+		}
 	}
 }
 

diff --git a/types.go b/types.go
@@ -18,8 +18,9 @@ type HandleFunc func(*TupaContext) error
 type HTTPMethod string
 
 type RouteInfo struct {
-	Path        string
-	Method      HTTPMethod
-	Handler     APIFunc
-	Middlewares []MiddlewareFunc
+	Path             string
+	Method           HTTPMethod
+	Handler          APIFunc
+	Middlewares      []MiddlewareFunc
+	AfterMiddlewares []MiddlewareFunc
 }