481 lines
13 KiB
PHP
481 lines
13 KiB
PHP
<?php
|
||
declare(strict_types=1);
|
||
|
||
namespace Kiri\Router;
|
||
|
||
use Closure;
|
||
use Kiri\Di\HotReloadState;
|
||
use Kiri\Server\Events\OnWorkerStart;
|
||
use Kiri;
|
||
use Kiri\Abstracts\CoordinatorManager;
|
||
use Kiri\Coordinator;
|
||
use Kiri\Router\RouteArtifactState;
|
||
use Kiri\Router\Validator\ValidatorMiddleware;
|
||
use Kiri\Router\Base\Middleware as MiddlewareManager;
|
||
use Kiri\Router\Constrict\RequestMethod;
|
||
use Psr\Container\ContainerInterface;
|
||
use Psr\Http\Message\ResponseInterface;
|
||
|
||
/**
|
||
*
|
||
*
|
||
* $component->set([
|
||
* 'request' => [
|
||
* 'middlewares' => []
|
||
* ]
|
||
* ])
|
||
*/
|
||
class Router
|
||
{
|
||
|
||
|
||
const array METHODS = [RequestMethod::REQUEST_POST, RequestMethod::REQUEST_GET, RequestMethod::REQUEST_OPTIONS, RequestMethod::REQUEST_DELETE, RequestMethod::REQUEST_PUT, RequestMethod::REQUEST_HEAD];
|
||
|
||
|
||
/**
|
||
* @var string
|
||
*/
|
||
private static string $type = ROUTER_TYPE_HTTP;
|
||
|
||
private static ?string $currentSourceFile = null;
|
||
|
||
/**
|
||
* 标记首次完整扫描是否已完成
|
||
* Master 进程中完成扫描后设为 true，Worker 通过 fork 继承此标记
|
||
* Worker 启动时检查此标记，避免重复执行全量 app 目录扫描导致 OOM
|
||
* @var bool
|
||
*/
|
||
private static bool $initialScanDone = false;
|
||
|
||
|
||
/**
|
||
* @param string $name
|
||
* @param Closure $closure
|
||
*/
|
||
public static function addServer(string $name, Closure $closure): void
|
||
{
|
||
static::$type = $name;
|
||
$closure();
|
||
static::$type = ROUTER_TYPE_HTTP;
|
||
}
|
||
|
||
|
||
/**
|
||
* @param Closure $handler
|
||
*/
|
||
public static function jsonp(Closure $handler): void
|
||
{
|
||
static::$type = 'json-rpc';
|
||
$handler();
|
||
static::$type = ROUTER_TYPE_HTTP;
|
||
}
|
||
|
||
|
||
/**
|
||
* @param string $route
|
||
* @param string $handler
|
||
*
|
||
* @throws
|
||
*/
|
||
public static function post(string $route, string $handler): void
|
||
{
|
||
$router = Kiri::getDi()->get(DataGrip::class)->get(static::$type);
|
||
$router->addRoute([RequestMethod::REQUEST_POST], $route, $handler);
|
||
}
|
||
|
||
/**
|
||
* @param string $route
|
||
* @param string $handler
|
||
*
|
||
* @throws
|
||
*/
|
||
public static function get(string $route, string $handler): void
|
||
{
|
||
$router = Kiri::getDi()->get(DataGrip::class)->get(static::$type);
|
||
$router->addRoute([RequestMethod::REQUEST_GET], $route, $handler);
|
||
}
|
||
|
||
|
||
/**
|
||
* @param string $route
|
||
* @param string $handler
|
||
*
|
||
* @throws
|
||
*/
|
||
public static function options(string $route, string $handler): void
|
||
{
|
||
$router = Kiri::getDi()->get(DataGrip::class)->get(static::$type);
|
||
$router->addRoute([RequestMethod::REQUEST_OPTIONS], $route, $handler);
|
||
}
|
||
|
||
|
||
/**
|
||
* @param string $route
|
||
* @param string $handler
|
||
*
|
||
* @throws
|
||
*/
|
||
public static function any(string $route, string $handler): void
|
||
{
|
||
$router = Kiri::getDi()->get(DataGrip::class)->get(static::$type);
|
||
$router->addRoute(self::METHODS, $route, $handler);
|
||
}
|
||
|
||
/**
|
||
* @param string $route
|
||
* @param string $handler
|
||
*
|
||
* @throws
|
||
*/
|
||
public static function delete(string $route, string $handler): void
|
||
{
|
||
$router = Kiri::getDi()->get(DataGrip::class)->get(static::$type);
|
||
$router->addRoute([RequestMethod::REQUEST_DELETE], $route, $handler);
|
||
}
|
||
|
||
|
||
/**
|
||
* @param string $route
|
||
* @param string $handler
|
||
*
|
||
* @throws
|
||
*/
|
||
public static function head(string $route, string $handler): void
|
||
{
|
||
$router = Kiri::getDi()->get(DataGrip::class)->get(static::$type);
|
||
$router->addRoute([RequestMethod::REQUEST_HEAD], $route, $handler);
|
||
}
|
||
|
||
|
||
/**
|
||
* @param string $route
|
||
* @param string $handler
|
||
*
|
||
* @throws
|
||
*/
|
||
public static function put(string $route, string $handler): void
|
||
{
|
||
$router = Kiri::getDi()->get(DataGrip::class)->get(static::$type);
|
||
$router->addRoute([RequestMethod::REQUEST_PUT], $route, $handler);
|
||
}
|
||
|
||
|
||
/**
|
||
* @param array|RequestMethod $methods
|
||
* @param string $route
|
||
* @param string|array $handler
|
||
*/
|
||
public static function addRoute(array|RequestMethod $methods, string $route, string|array $handler): void
|
||
{
|
||
$router = Kiri::getDi()->get(DataGrip::class)->get(static::$type);
|
||
if ($methods instanceof RequestMethod) {
|
||
$methods = [$methods];
|
||
}
|
||
$router->addRoute($methods, $route, $handler);
|
||
}
|
||
|
||
|
||
/**
|
||
* @param array $config
|
||
* @param Closure $closure
|
||
*
|
||
* @throws
|
||
*/
|
||
public static function group(array $config, Closure $closure): void
|
||
{
|
||
$router = Kiri::getDi()->get(DataGrip::class)->get(static::$type);
|
||
|
||
$router->groupTack[] = $config;
|
||
|
||
call_user_func($closure);
|
||
|
||
array_pop($router->groupTack);
|
||
}
|
||
|
||
|
||
/**
|
||
* 扫描并构建路由表
|
||
*
|
||
* Master 进程：执行完整扫描（路由文件加载 + app 目录扫描 + DeferRegistry 注入）
|
||
* Worker 进程（首次启动）：仅加载路由文件注册路由表，跳过全量 app 扫描
|
||
* Worker 进程（热重载）：检测到文件变更时执行完整扫描流程
|
||
*
|
||
* 设计原因：
|
||
* - Master 已完成类加载和字节码编译，Worker 通过 fork 继承全部内存
|
||
* - Worker 重复执行 opcache_compile_file + invalidateClasses 不产生新信息
|
||
* - 在应用文件较多时（500+），每个 Worker 的全量扫描会消耗数百 MB 内存导致 OOM
|
||
*
|
||
* @throws
|
||
*/
|
||
public function scan_build_route(): void
|
||
{
|
||
$coordinator = CoordinatorManager::utility(Coordinator::WORKER_START);
|
||
$container = Kiri::getDi();
|
||
|
||
$changedFiles = $container->get(HotReloadState::class)->consume();
|
||
|
||
// Worker 首次启动（无变更文件 + Master 已完成扫描）：
|
||
// 重新 include 路由文件（Router::get/post 显式注册） + 基于 Master 扫描清单重建注解路由
|
||
// 避免 opcache_compile_file，仅用 Reflection 重建路由，内存开销极小
|
||
if (empty($changedFiles) && self::$initialScanDone) {
|
||
$container->get(DataGrip::class)->reset(static::$type);
|
||
$this->read_dir_file(APP_PATH . 'routes');
|
||
$this->rebuildAnnotationRoutes($container);
|
||
$this->reset($container);
|
||
$coordinator->done();
|
||
return;
|
||
}
|
||
|
||
// 标记首次扫描完成（Master 首次启动或 Worker 热重载时执行到此）
|
||
self::$initialScanDone = true;
|
||
|
||
$container->get(DataGrip::class)->reset(static::$type);
|
||
|
||
$scanner = $container->get(Kiri\Di\Scanner::class);
|
||
$artifactState = $container->get(RouteArtifactState::class);
|
||
$scanConfig = array_merge(
|
||
config('servers.reload.scan', []),
|
||
config('site.scanner', [])
|
||
);
|
||
$scanner->setConfig($scanConfig);
|
||
|
||
$normalizedAppPath = str_replace('\\', '/', APP_PATH . 'app');
|
||
$normalizedRoutePath = str_replace('\\', '/', APP_PATH . 'routes');
|
||
$routeChanged = false;
|
||
$appChangedFiles = [];
|
||
|
||
foreach ($changedFiles as $changedFile) {
|
||
if (str_starts_with($changedFile, $normalizedRoutePath . '/')) {
|
||
$routeChanged = true;
|
||
continue;
|
||
}
|
||
|
||
if (str_starts_with($changedFile, $normalizedAppPath . '/')) {
|
||
$appChangedFiles[] = $changedFile;
|
||
}
|
||
}
|
||
|
||
$usedArtifact = false;
|
||
if (($scanConfig['cache_enabled'] ?? false) && !$routeChanged && $artifactState->has(static::$type)) {
|
||
$artifact = $artifactState->load(static::$type);
|
||
$router = $container->get(DataGrip::class)->get(static::$type);
|
||
$usedArtifact = $router->importArtifact($artifact, $appChangedFiles);
|
||
}
|
||
|
||
if (!$usedArtifact) {
|
||
$this->read_dir_file(APP_PATH . 'routes');
|
||
}
|
||
|
||
if (!$routeChanged && !empty($appChangedFiles) && ($scanConfig['cache_enabled'] ?? false)) {
|
||
$scanner->scanFiles($appChangedFiles, APP_PATH . 'app/', null, !$usedArtifact);
|
||
} elseif (!$usedArtifact) {
|
||
$scanner->scan(APP_PATH . 'app/');
|
||
} else {
|
||
$scanner->scanFiles([], APP_PATH . 'app/', null, false);
|
||
}
|
||
$this->reset($container);
|
||
$artifactState->store(static::$type, $container->get(DataGrip::class)->get(static::$type)->exportArtifact());
|
||
|
||
$coordinator->done();
|
||
}
|
||
|
||
|
||
/**
|
||
* @param ContainerInterface $container
|
||
*
|
||
* @return void
|
||
* @throws
|
||
*/
|
||
public function reset(ContainerInterface $container): void
|
||
{
|
||
$router = $container->get(DataGrip::class)->get(static::$type);
|
||
if ((bool)config('servers.reload.scan.prebuild_http_handlers', false)) {
|
||
$router->warmHttpHandlers();
|
||
}
|
||
}
|
||
|
||
|
||
/**
|
||
* 基于 Master 扫描清单重建注解路由（轻量级，无文件 I/O）
|
||
* 遍历 Scanner manifest 中的所有类，用 Reflection 重新发现 #[Route]/#[Get] 等注解
|
||
* 避免 Worker 重复执行 opcache_compile_file，但确保注解路由不丢失
|
||
*
|
||
* @param ContainerInterface $container
|
||
* @return void
|
||
*/
|
||
private function rebuildAnnotationRoutes(ContainerInterface $container): void
|
||
{
|
||
$scanner = $container->get(Kiri\Di\Scanner::class);
|
||
$scanConfig = array_merge(
|
||
config('servers.reload.scan', []),
|
||
config('site.scanner', [])
|
||
);
|
||
$scanner->setConfig($scanConfig);
|
||
|
||
// 从 manifest 获取 Master 扫描过的类
|
||
$manifestEntries = $scanner->getManifestClasses();
|
||
$manifestClasses = [];
|
||
foreach ($manifestEntries as $entry) {
|
||
if (is_array($entry) && isset($entry['classes'])) {
|
||
foreach ($entry['classes'] as $c) {
|
||
$manifestClasses[$c] = true;
|
||
}
|
||
}
|
||
}
|
||
|
||
// 关键：manifest 只包含 Scanner 通过 require_once 新发现的类
|
||
// 但路由文件加载时 $container->get(Controller) 会触发 autoload 提前加载类
|
||
// 导致 Scanner 的 require_once 变成 no-op，该类及其注解永久丢失
|
||
// 因此必须合并 get_declared_classes() 补扫所有已声明的用户空间类
|
||
$allDeclared = get_declared_classes();
|
||
foreach ($allDeclared as $class) {
|
||
$manifestClasses[$class] = true;
|
||
}
|
||
|
||
if (empty($manifestClasses)) {
|
||
\Kiri::getLogger()->warning('Annotation route rebuild: no classes to process');
|
||
return;
|
||
}
|
||
|
||
$routeCount = 0;
|
||
$classCount = 0;
|
||
$errorCount = 0;
|
||
$dispatchCount = 0;
|
||
|
||
// 只处理用户命名空间下的类，排除框架和 PHP 内置类
|
||
$userNamespaces = $scanConfig['user_namespaces'] ?? ['App\\'];
|
||
|
||
foreach (array_keys($manifestClasses) as $class) {
|
||
$isUserClass = false;
|
||
foreach ($userNamespaces as $ns) {
|
||
if (str_starts_with($class, $ns)) {
|
||
$isUserClass = true;
|
||
break;
|
||
}
|
||
}
|
||
if (!$isUserClass) {
|
||
continue;
|
||
}
|
||
if (!class_exists($class)) {
|
||
continue;
|
||
}
|
||
$classCount++;
|
||
try {
|
||
$reflect = $container->getReflectionClass($class);
|
||
if (!$reflect->isInstantiable() || $reflect->isTrait() || $reflect->isEnum() || $reflect->isInterface() || $reflect->isAbstract()) {
|
||
continue;
|
||
}
|
||
foreach ($reflect->getMethods() as $method) {
|
||
if ($method->isStatic() || $method->getDeclaringClass()->getName() !== $class) {
|
||
continue;
|
||
}
|
||
foreach ($method->getAttributes() as $attribute) {
|
||
$attrName = $attribute->getName();
|
||
if (!class_exists($attrName)) {
|
||
continue;
|
||
}
|
||
try {
|
||
$instance = $attribute->newInstance();
|
||
if ($instance instanceof Kiri\Di\Interface\InjectMethodInterface) {
|
||
$instance->dispatch($class, $method->getName());
|
||
$dispatchCount++;
|
||
}
|
||
} catch (\Throwable $e) {
|
||
$errorCount++;
|
||
\Kiri::getLogger()->error("Annotation rebuild error [{$class}::{$method->getName()} @ {$attrName}]: {$e->getMessage()}");
|
||
}
|
||
}
|
||
}
|
||
} catch (\Throwable $e) {
|
||
$errorCount++;
|
||
\Kiri::getLogger()->error("Annotation rebuild class [{$class}]: {$e->getMessage()}");
|
||
}
|
||
}
|
||
|
||
$router = $container->get(DataGrip::class)->get(static::$type);
|
||
$routeCount = count($router->dump());
|
||
|
||
\Kiri::getLogger()->info("Annotation route rebuild: {$classCount} user classes processed, {$dispatchCount} annotation routes dispatched, {$routeCount} total routes, {$errorCount} errors");
|
||
|
||
// 搜索特定路径的诊断日志
|
||
$searchPaths = ['/headers'];
|
||
foreach ($searchPaths as $searchPath) {
|
||
$found = [];
|
||
foreach (array_keys($manifestClasses) as $class) {
|
||
if (!class_exists($class)) continue;
|
||
try {
|
||
$reflect = $container->getReflectionClass($class);
|
||
foreach ($reflect->getMethods() as $method) {
|
||
foreach ($method->getAttributes() as $attr) {
|
||
if (in_array($attr->getName(), [
|
||
\Kiri\Router\Annotate\Get::class,
|
||
\Kiri\Router\Annotate\Post::class,
|
||
\Kiri\Router\Annotate\Put::class,
|
||
\Kiri\Router\Annotate\Delete::class,
|
||
\Kiri\Router\Annotate\Route::class,
|
||
])) {
|
||
$instance = $attr->newInstance();
|
||
$routePath = $instance->path ?? '';
|
||
if (str_contains($routePath, 'header')) {
|
||
$version = $instance->version ?? '';
|
||
$found[] = "{$class}::{$method->getName()} path={$routePath} version={$version}";
|
||
}
|
||
}
|
||
}
|
||
}
|
||
} catch (\Throwable) {}
|
||
}
|
||
if (!empty($found)) {
|
||
\Kiri::getLogger()->info("Annotation route search '{$searchPath}': " . implode(' | ', $found));
|
||
} else {
|
||
\Kiri::getLogger()->warning("Annotation route search '{$searchPath}': NO annotation found in any class");
|
||
}
|
||
}
|
||
}
|
||
|
||
|
||
/**
|
||
* @param $path
|
||
*
|
||
* @return void
|
||
* @throws
|
||
*/
|
||
private function read_dir_file($path): void
|
||
{
|
||
$files = glob($path . '/*');
|
||
for ($i = 0; $i < count($files); $i++) {
|
||
$file = $files[$i];
|
||
if (is_dir($file)) {
|
||
$this->read_dir_file($file);
|
||
} else {
|
||
$this->resolve_file($file);
|
||
}
|
||
}
|
||
}
|
||
|
||
|
||
/**
|
||
* @param $files
|
||
*
|
||
* @throws
|
||
*/
|
||
private function resolve_file($files): void
|
||
{
|
||
try {
|
||
static::$currentSourceFile = str_replace('\\', '/', realpath($files) ?: $files);
|
||
include "$files";
|
||
} catch (\Throwable $throwable) {
|
||
\Kiri::getLogger()->json_log($throwable);
|
||
} finally {
|
||
static::$currentSourceFile = null;
|
||
}
|
||
}
|
||
|
||
|
||
public static function getCurrentSourceFile(): ?string
|
||
{
|
||
return static::$currentSourceFile;
|
||
}
|
||
|
||
}
|