Fix root node allocation, add early return for root node matches

This commit is contained in:
Sky Johnson 2024-09-11 11:40:13 -05:00
parent 327981e72e
commit 1bff49e3ba
3 changed files with 72 additions and 43 deletions

View File

@ -11,9 +11,12 @@ class SegmentRouter implements Router
return str_starts_with($segment, ':') ? ':x' : $segment; return str_starts_with($segment, ':') ? ':x' : $segment;
}, explode('/', trim($route, '/'))); }, explode('/', trim($route, '/')));
// Push each segment into the routes array as a node // Push each segment into the routes array as a node, except if this is the root node
$node = &$this->routes; $node = &$this->routes;
foreach ($segments as $segment) $node = &$node[$segment]; foreach ($segments as $segment) {
if ($segment === '') continue;
$node = &$node[$segment];
}
// Add the handler to the last node // Add the handler to the last node
$node[$method] = $handler; $node[$method] = $handler;
@ -23,44 +26,36 @@ class SegmentRouter implements Router
public function lookup(string $method, string $uri): int|array public function lookup(string $method, string $uri): int|array
{ {
// Expand the URI into segments
$uriSegments = explode('/', trim($uri, '/'));
$node = $this->routes; $node = $this->routes;
if (isset($node[$uri])) {
if (isset($node[$uri][$method])) return [$node[$uri][$method], []];
return 405;
}
$uriSegments = explode('/', trim($uri, '/'));
$params = []; $params = [];
// Traverse the routes array to find the handler
foreach ($uriSegments as $segment) { foreach ($uriSegments as $segment) {
// Check if the segment exists in the node
if (isset($node[$segment])) { if (isset($node[$segment])) {
$node = $node[$segment]; $node = $node[$segment];
} else { continue;
// Handle dynamic segments (starting with ":")
$dynamicSegment = null;
// Loop through the node and find the first dynamic segment
foreach ($node as $k => $v) {
if (str_starts_with($k, ':')) {
$dynamicSegment = $k;
break; // Break early as we only need one match
}
}
// If no dynamic segment was found, return 404
if ($dynamicSegment === null) return 404;
// Otherwise, store the parameter and move to the dynamic node
$params[] = $segment;
$node = $node[$dynamicSegment];
} }
if (isset($node[':x'])) {
$params[] = $segment;
$node = $node[':x'];
continue;
}
return 404;
} }
// Check if the HTTP method is supported
if (!isset($node[$method])) return 405; if (!isset($node[$method])) return 405;
// Return the handler and parameters return [$node[$method], $params];
return [$node[$method] , $params];
} }
public function clear(): Router public function clear(): Router
{ {

View File

@ -1,6 +1,5 @@
/ /
├── / ├── GET
│ └── GET
├── :x ├── :x
│ └── GET │ └── GET
├── tags ├── tags

View File

@ -107,32 +107,67 @@ function readAndAddRoutes(string $file, Router &$r): array
} }
function runIterations(int $iterations, $r, array $routes) { function runIterations(int $iterations, $r, array $routes) {
echo "Running $iterations iterations\n"; echo "\n🚀 Running $iterations iterations...\n";
$start = microtime(true);
$interval = $iterations / 10; $start = microtime(true); // start the timer
$reqs = 0; // track the timing of lookups
$longest = 0; // track the longest lookup time
$shortest = 0; // track the shortest lookup time
$longestRoute = '';
$shortestRoute = '';
for ($i = 0; $i < $iterations; $i++) { for ($i = 0; $i < $iterations; $i++) {
// pick a random route from the array // pick a random route from the array
[$method, $path] = $routes[array_rand($routes)]; [$method, $path] = $routes[array_rand($routes)];
/*
// replace all :params/ with random values // replace all :params/ with random values
$uri = preg_replace_callback('/:(\w+)/', function($matches) { $uri = preg_replace_callback('/:(\w+)/', function($matches) {
return rand(1, 100); return rand(1, 100);
}, $path); }, $path);
*/
$uri = $path; $start2 = microtime(true); // start the timer for the lookup
$start2 = microtime(true);
$res = $r->lookup($method, $uri); $res = $r->lookup($method, $uri); // run the lookup
$reqs += microtime(true) - $start2; // add this lookup time
if ($shortest == 0 || microtime(true) - $start2 < $shortest) {
$shortest = microtime(true) - $start2; // track the shortest lookup time
$shortestRoute = "$method $uri";
}
if (microtime(true) - $start2 > $longest) {
$longest = microtime(true) - $start2; // track the longest lookup time
$longestRoute = "$method $uri";
}
// if any error was encountered, print it and exit
if ($res === 404 || $res === 405) { if ($res === 404 || $res === 405) {
echo Color::red("Failed to handle request.\n$method $res\n"."├─ URI: $uri\n└─ Path: $path\n"); echo Color::red("Failed to handle request.\n$method $res\n"."├─ URI: $uri\n└─ Path: $path\n");
echo Color::yellow("Completed $i iterations before failure.\n"); echo Color::yellow("Completed $i iterations before failure.\n");
exit(1); exit(1);
} }
if ($i !== 0 && $i % ($interval) === 0) echoMemoryAndTime($i, $start2);
} }
echo Color::blue("✔️ Done!")."\n";
// echo peak memory usage
echo "Peak memory: " . Color::magenta(round(memory_get_peak_usage() / 1024, 1) . " kb\n");
// total time used for this run
echo "Time: " . Color::cyan(number_format(microtime(true) - $start, 10) . " s\n"); echo "Time: " . Color::cyan(number_format(microtime(true) - $start, 10) . " s\n");
// echo the average time per request // echo the average time per request
echo "Avg/lookup: " . Color::yellow(number_format((microtime(true) - $start) / $iterations, 10) . " s\n"); echo "Avg/lookup: " . Color::yellow(number_format($reqs / $iterations, 10) . " s\n");
echo "\n";
// echo the shortest lookup time
echo "Shortest lookup: " . Color::green(number_format($shortest, 10) . " s\n");
// echo the longest lookup time
echo "Longest lookup: " . Color::red(number_format($longest, 10) . " s\n");
echo Color::black("==============================================") . "\n";
// echo the longest and shortest routes
echo "Shortest route: " . Color::green($shortestRoute) . "\n";
echo "Longest route: " . Color::red($longestRoute) . "\n";
} }
// take a route tree (array) and store it in a file to be read // take a route tree (array) and store it in a file to be read