Update Router.php

isu73 · Oct 20, 2020 · 9a58e68 · 9a58e68
1 parent 38042be
commit 9a58e68
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/src/Bramus/Router/Router.php b/src/Bramus/Router/Router.php
@@ -289,11 +289,7 @@ public function run($callback = null)
 
         // If no route was handled, trigger the 404 (if any)
         if ($numHandled === 0) {
-            if ($this->notFoundCallback) {
-                $this->invoke($this->notFoundCallback);
-            } else {
-                header($_SERVER['SERVER_PROTOCOL'] . ' 404 Not Found');
-            }
+            $this->return404();
         } // If a route was handled, perform the finish callback (if any)
         else {
             if ($callback && is_callable($callback)) {
@@ -320,6 +316,17 @@ public function set404($fn)
         $this->notFoundCallback = $fn;
     }
 
+    /**
+     * Triggers 404 response
+     */
+    public function return404(){
+        if ($this->notFoundCallback) {
+            $this->invoke($this->notFoundCallback);
+        } else {
+            header($_SERVER['SERVER_PROTOCOL'] . ' 404 Not Found');
+        }
+    }
+
     /**
      * Handle a a set of routes: if a match is found, execute the relating handling function.
      *