Added tags to all connections and entity managers

Kdyby · Sep 6, 2013 · aa25c84 · aa25c84
1 parent 54d787f
commit aa25c84
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
@@ -281,6 +283,7 @@ protected function processEntityManager($name, array $defaults)
 				$connectionService = $this->processConnection($name, $defaults, $isDefault),
 				$this->prefix('@' . $name . '.ormConfiguration')
 			))
+			->addTag(self::TAG_ENTITY_MANAGER)
 			->setAutowired($isDefault);
 	}
 
@@ -329,6 +332,7 @@ protected function processConnection($name, array $defaults, $isDefault = FALSE)
 				3 => $dbalTypes,
 				$schemaTypes
 			))
+			->addTag(self::TAG_CONNECTION)
 			->setAutowired($isDefault);
 		/** @var Nette\DI\ServiceDefinition $connection */