creating lazy services in PHP 8.4 WIP

nette · Nov 25, 2024 · d9d7d4c · d9d7d4c
1 parent 9b9bfb4
commit d9d7d4c
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 0 deletions.
diff --git a/src/DI/Definitions/ServiceDefinition.php b/src/DI/Definitions/ServiceDefinition.php
@@ -11,6 +11,7 @@
 
 use Nette;
 use Nette\DI\ServiceCreationException;
+use Nette\Utils\Strings;
 
 
 /**
@@ -29,6 +30,8 @@ final class ServiceDefinition extends Definition
 	/** @var Statement[] */
 	private array $setup = [];
 
+	private bool $lazy = true;
+
 
 	public function __construct()
 	{
@@ -181,6 +184,24 @@ private function prependSelf(Statement $setup): Statement
 
 	public function generateMethod(Nette\PhpGenerator\Method $method, Nette\DI\PhpGenerator $generator): void
 	{
+		if ($this->canLazy()) {
+			$rc = new \ReflectionClass($this->creator->getEntity());
+			$code = $rc->hasMethod('__construct')
+				? $generator->formatPhp("\$service->__construct(...?:);\n", [$this->creator->arguments])
+				: '';
+			foreach ($this->setup as $setup) {
+				$code .= $generator->formatStatement($setup) . ";\n";
+			}
+			$method->setBody(
+				"return new ReflectionClass({$rc->getName()}::class)->newLazyGhost(function (\$service) {\n"
+				. Strings::indent($code)
+				. '});',
+			);
+			if (!preg_match('#(?:func_get_arg|func_num_args)#i', $code)) { // latteFactory workaround
+				return;
+			}
+		}
+
 		$code = $generator->formatStatement($this->creator) . ";\n";
 		if (!$this->setup) {
 			$method->setBody('return ' . $code);
@@ -197,6 +218,22 @@ public function generateMethod(Nette\PhpGenerator\Method $method, Nette\DI\PhpGe
 	}
 
 
+	private function canLazy(): bool
+	{
+		if (!$this->lazy || !is_string($this->creator->getEntity()) || PHP_VERSION_ID < 80400) {
+			return false;
+		}
+
+		$rc = new \ReflectionClass($this->creator->getEntity());
+		do {
+			if ($rc->isInternal()) {
+				return false;
+			}
+		} while ($rc = $rc->getParentClass());
+		return true;
+	}
+
+
 	public function __clone()
 	{
 		parent::__clone();

diff --git a/src/DI/Extensions/DIExtension.php b/src/DI/Extensions/DIExtension.php
@@ -36,6 +36,7 @@ public function __construct(bool $debugMode = false)
 			public array $excluded = [];
 			public ?string $parentClass = null;
 			public object $export;
+			public bool $lazy = false;
 		};
 		$this->config->export = new class {
 			public bool $parameters = true;
@@ -62,6 +63,7 @@ public function afterCompile(Nette\PhpGenerator\ClassType $class): void
 			$class->setExtends($this->config->parentClass);
 		}
 
+		$this->initializeLazy($class);
 		$this->restrictParameters($class);
 		$this->restrictTags($class);
 		$this->restrictTypes($class);