From 8a72e9025710740d546ca53a283a49c672e0d21d Mon Sep 17 00:00:00 2001
From: xl <as2252258@163.com>
Date: Sat, 15 Jul 2023 01:56:39 +0800
Subject: [PATCH] qqq

---
 Handler/OnRequest.php | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/Handler/OnRequest.php b/Handler/OnRequest.php
index be84aaa..ae697fd 100644
--- a/Handler/OnRequest.php
+++ b/Handler/OnRequest.php
@@ -61,6 +61,11 @@ class OnRequest implements OnRequestInterface
     public ResponseInterface $response;
 
 
+    /**
+     * @var MiddlewareManager
+     */
+    public MiddlewareManager $middlewareManager;
+
     /**
      * @throws Exception
      */
@@ -74,6 +79,8 @@ class OnRequest implements OnRequestInterface
         $this->exception = $container->get($exception);
         $this->router = $container->get(DataGrip::class)->get(ROUTER_TYPE_HTTP);
         $this->emitter = $this->response->emmit;
+
+        $this->middlewareManager = \Kiri::getDi()->get(MiddlewareManager::class);
     }
 
 
@@ -89,11 +96,7 @@ class OnRequest implements OnRequestInterface
             $PsrRequest = $this->initPsr7RequestAndPsr7Response($request);
             $dispatcher = $this->router->query($request->server['path_info'], $request->getMethod());
 
-            $middleware = [];
-            if ($dispatcher->getClass() !== Kiri\Router\Base\NotFoundController::class) {
-                $middlewareManager = \Kiri::getDi()->get(MiddlewareManager::class);
-                $middleware = $middlewareManager->get($dispatcher->getClass(), $dispatcher->getMethod());
-            }
+            $middleware = $this->middlewareManager->get($dispatcher->getClass(), $dispatcher->getMethod());
 
             $PsrResponse = (new HttpRequestHandler($middleware, $dispatcher))->handle($PsrRequest);
         } catch (\Throwable $throwable) {