Rules\LazyRegistry, Collectors\Registry: Cache extension-class-names

phpstan-baseline.neon

@@ -669,18 +669,6 @@ parameters:
 			count: 1
 			path: src/Rules/Generics/TemplateTypeCheck.php
 
-		-
-			rawMessage: 'Function class_implements() is a runtime reflection concept that might not work in PHPStan because it uses fully static reflection engine. Use objects retrieved from ReflectionProvider instead.'
-			identifier: phpstanApi.runtimeReflection
-			count: 1
-			path: src/Rules/LazyRegistry.php
-
-		-
-			rawMessage: 'Function class_parents() is a runtime reflection concept that might not work in PHPStan because it uses fully static reflection engine. Use objects retrieved from ReflectionProvider instead.'
-			identifier: phpstanApi.runtimeReflection
-			count: 1
-			path: src/Rules/LazyRegistry.php
-
 		-
 			rawMessage: 'Method PHPStan\Rules\LazyRegistry::getRulesByNodeType() return type with generic interface PHPStan\Rules\Rule does not specify its types: TNodeType'
 			identifier: missingType.generics

src/Collectors/Registry.php

@@ -4,8 +4,7 @@
 
 use PhpParser\Node;
 use PHPStan\DependencyInjection\AutowiredService;
-use function class_implements;
-use function class_parents;
+use PHPStan\Type\ExtensionClassHelper;
 
 #[AutowiredService(factory: '@PHPStan\Collectors\RegistryFactory::create')]
 final class Registry
@@ -35,7 +34,7 @@ public function __construct(array $collectors)
 	public function getCollectors(string $nodeType): array
 	{
 		if (!isset($this->cache[$nodeType])) {
-			$parentNodeTypes = [$nodeType] + class_parents($nodeType) + class_implements($nodeType);
+			$parentNodeTypes = ExtensionClassHelper::getExtensionClassNamesByRuntimeReflection($nodeType);
 
 			$collectors = [];
 			foreach ($parentNodeTypes as $parentNodeType) {

src/Rules/LazyRegistry.php

@@ -6,8 +6,7 @@
 use PHPStan\DependencyInjection\AutowiredExtensions;
 use PHPStan\DependencyInjection\AutowiredService;
 use PHPStan\DependencyInjection\ExtensionsCollection;
-use function class_implements;
-use function class_parents;
+use PHPStan\Type\ExtensionClassHelper;
 
 #[AutowiredService(name: 'registry', as: Registry::class)]
 final class LazyRegistry implements Registry
@@ -39,7 +38,7 @@ public function __construct(
 	public function getRules(string $nodeType): array
 	{
 		if (!isset($this->cache[$nodeType])) {
-			$parentNodeTypes = [$nodeType] + class_parents($nodeType) + class_implements($nodeType);
+			$parentNodeTypes = ExtensionClassHelper::getExtensionClassNamesByRuntimeReflection($nodeType);
 
 			$rules = [];
 			$rulesFromContainer = $this->getRulesByNodeType();

src/Type/ExtensionClassHelper.php

@@ -2,16 +2,22 @@
 
 namespace PHPStan\Type;
 
+use PhpParser\Node;
 use PHPStan\Reflection\ReflectionProvider;
 use function array_key_exists;
 use function array_merge;
+use function class_implements;
+use function class_parents;
 
 final class ExtensionClassHelper
 {
 
 	/** @var array<string, array<string>> */
 	private static array $extensionClassNames = [];
 
+	/** @var array<class-string<Node>, array<class-string<Node>>> */
+	private static array $extensionClassNamesRuntimeReflections = [];
+
 	/**
 	 * @return string[]
 	 */
@@ -25,4 +31,17 @@ public static function getExtensionClassNames(ReflectionProvider $reflectionProv
 		return self::$extensionClassNames[$className];
 	}
 
+	/**
+	 * @param class-string<Node> $className
+	 * @return array<class-string<Node>>
+	 */
+	public static function getExtensionClassNamesByRuntimeReflection(string $className): array
+	{
+		if (!array_key_exists($className, self::$extensionClassNamesRuntimeReflections)) {
+			self::$extensionClassNamesRuntimeReflections[$className] = [$className] + class_parents($className) + class_implements($className);
+		}
+
+		return self::$extensionClassNamesRuntimeReflections[$className];
+	}
+
 }