Fix incompatibility with Nette/DI 2.4

Kdyby · Apr 17, 2016 · bde1a7a · bde1a7a
1 parent 823fbae
commit bde1a7a
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/Kdyby/Doctrine/DI/OrmExtension.php b/src/Kdyby/Doctrine/DI/OrmExtension.php
@@ -601,17 +601,17 @@ protected function processMetadataDriver(Nette\DI\ServiceDefinition $metadataDri
 			$driver = new Statement(self::ANNOTATION_DRIVER, is_array($paths) ? $paths : [$paths]);
 		}
 
-		$impl = $driver instanceof \stdClass ? $driver->value : ($driver instanceof Statement ? $driver->entity : (string) $driver);
+		$impl = $driver instanceof \stdClass ? $driver->value : ($driver instanceof Statement ? $driver->getEntity() : (string) $driver);
 		list($driver) = CacheHelpers::filterArgs($driver);
 		/** @var Statement $driver */
 
 		if (isset($this->metadataDriverClasses[$impl])) {
-			$driver->entity = $this->metadataDriverClasses[$impl];
+			$driver->setEntity($this->metadataDriverClasses[$impl]);
 		}
 
-		if (is_string($driver->entity) && substr($driver->entity, 0, 1) === '@') {
-			$metadataDriver->addSetup('addDriver', [$driver->entity, $namespace]);
-			return $driver->entity;
+		if (is_string($driver->getEntity()) && substr($driver->getEntity(), 0, 1) === '@') {
+			$metadataDriver->addSetup('addDriver', [$driver->getEntity(), $namespace]);
+			return $driver->getEntity();
 		}
 
 		if ($impl === self::ANNOTATION_DRIVER) {
@@ -625,7 +625,7 @@ protected function processMetadataDriver(Nette\DI\ServiceDefinition $metadataDri
 
 		$this->getContainerBuilder()->addDefinition($serviceName)
 			->setClass('Doctrine\Common\Persistence\Mapping\Driver\MappingDriver')
-			->setFactory($driver->entity, $driver->arguments)
+			->setFactory($driver->getEntity(), $driver->arguments)
 			->setAutowired(FALSE)
 			->setInject(FALSE);