Merge pull request #44 from enumag/tag

Added tags to all connections and entity managers
Kdyby · Sep 6, 2013 · e5084e5 · e5084e5
2 parents 32dd68e + fe1898b
commit e5084e5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/Kdyby/Doctrine/DI/OrmExtension.php b/src/Kdyby/Doctrine/DI/OrmExtension.php
@@ -38,6 +38,8 @@ class OrmExtension extends Nette\DI\CompilerExtension
 
 	const ANNOTATION_DRIVER = 'annotations';
 	const PHP_NAMESPACE = '[a-zA-Z_\x7f-\xff][a-zA-Z0-9_\x7f-\xff\\\\]*';
+	const TAG_CONNECTION = 'kdyby.doctrine.connection';
+	const TAG_ENTITY_MANAGER = 'kdyby.doctrine.entityManager';
 
 	/**
 	 * @var array
@@ -297,6 +299,7 @@ protected function processEntityManager($name, array $defaults)
 				$connectionService = $this->processConnection($name, $defaults, $isDefault),
 				$this->prefix('@' . $name . '.ormConfiguration')
 			))
+			->addTag(self::TAG_ENTITY_MANAGER)
 			->setAutowired($isDefault)
 			->setInject(FALSE);
 	}
@@ -348,6 +351,7 @@ protected function processConnection($name, array $defaults, $isDefault = FALSE)
 				3 => $dbalTypes,
 				$schemaTypes
 			))
+			->addTag(self::TAG_CONNECTION)
 			->setAutowired($isDefault)
 			->setInject(FALSE);
 		/** @var Nette\DI\ServiceDefinition $connection */