Multiple Entity Managers and Connections

Author: npmarrin

src/Console/DqlCommand.php

@@ -1,7 +1,7 @@
 <?php  namespace Mitch\LaravelDoctrine\Console;
 
 use Doctrine\Common\Util\Debug;
-use Doctrine\ORM\EntityManagerInterface;
+use Doctrine\Common\Persistence\ManagerRegistry;
 use Symfony\Component\Console\Input\InputArgument;
 use Symfony\Component\Console\Input\InputOption;
 
@@ -28,11 +28,11 @@ class DqlCommand extends Command
      */
     private $entityManager;
 
-    public function __construct(EntityManagerInterface $entityManager)
+    public function __construct(ManagerRegistry $registry)
     {
         parent::__construct();
 
-        $this->entityManager = $entityManager;
+        $this->entityManager = $registry->getManager();
     }
 
     public function fire()
@@ -53,4 +53,4 @@ protected function getOptions()
             ['hydrate', null, InputOption::VALUE_OPTIONAL, 'Hydrate type. Available: object, array, scalar, single_scalar, simpleobject']
         ];
     }
-} 
+}

src/Console/GenerateProxiesCommand.php

@@ -1,7 +1,7 @@
 <?php  namespace Mitch\LaravelDoctrine\Console;
 
 use Illuminate\Console\Command;
-use Doctrine\ORM\EntityManagerInterface;
+use Doctrine\Common\Persistence\ManagerRegistry;
 
 class GenerateProxiesCommand extends Command
 {
@@ -26,11 +26,11 @@ class GenerateProxiesCommand extends Command
      */
     private $entityManager;
 
-    public function __construct(EntityManagerInterface $entityManager)
+    public function __construct(ManagerRegistry $registry)
     {
         parent::__construct();
 
-        $this->entityManager = $entityManager;
+        $this->entityManager = $registry->getManager();;
     }
 
     public function fire()
@@ -53,4 +53,4 @@ public function fire()
         $this->entityManager->getProxyFactory()->generateProxyClasses($metadata, $directory);
         $this->info('Proxies have been created.');
     }
-} 
+}

src/DoctrineUserProvider.php

@@ -1,6 +1,6 @@
 <?php namespace Mitch\LaravelDoctrine;
 
-use Doctrine\ORM\EntityManager;
+use Doctrine\Common\Persistence\ManagerRegistry;
 use Doctrine\ORM\EntityRepository;
 use Illuminate\Auth\UserInterface;
 use Illuminate\Auth\UserProviderInterface;
@@ -13,7 +13,7 @@ class DoctrineUserProvider implements UserProviderInterface
      */
     private $hasher;
     /**
-     * @var EntityManager
+     * @var \Doctrine\ORM\EntityManager
      */
     private $entityManager;
     /**
@@ -23,13 +23,13 @@ class DoctrineUserProvider implements UserProviderInterface
 
     /**
      * @param HasherInterface $hasher
-     * @param EntityManager $entityManager
+     * @param ManagerRegistry $registry
      * @param $entity
      */
-    public function __construct(HasherInterface $hasher, EntityManager $entityManager, $entity)
+    public function __construct(HasherInterface $hasher, ManagerRegistry $registry, $entity)
     {
         $this->hasher = $hasher;
-        $this->entityManager = $entityManager;
+        $this->entityManager = $registry->getManager();
         $this->entity = $entity;
     }
     /**

src/IlluminateRegistry.php

@@ -0,0 +1,60 @@
+<?php namespace Mitch\LaravelDoctrine;
+
+use Doctrine\ORM\ORMException;
+use Illuminate\Container\Container;
+use Doctrine\Common\Persistence\AbstractManagerRegistry;
+
+final class IlluminateRegistry extends AbstractManagerRegistry
+{
+    /**
+     * @var Container
+     */
+    private $container;
+
+    public function __construct(Container $container, array $connections, array $entityManagers, $defaultConnection, $defaultEntityManager)
+    {
+        $this->container = $container;
+        parent::__construct('ORM', $connections, $entityManagers, $defaultConnection, $defaultEntityManager, 'Doctrine\ORM\Proxy\Proxy');
+    }
+
+    /**
+     * Fetches/creates the given services.
+     * A service in this context is connection or a manager instance.
+     * @param string $name The name of the service.
+     * @return object The instance of the given service.
+     */
+    protected function getService($name)
+    {
+        return $this->container->make($name);
+    }
+
+    /**
+     * Resets the given services.
+     * A service in this context is connection or a manager instance.
+     * @param string $name The name of the service.
+     * @return void
+     */
+    protected function resetService($name)
+    {
+        return $this->container->bind($name, null);
+    }
+
+    /**
+     * Resolves a registered namespace alias to the full namespace.
+     * This method looks for the alias in all registered object managers.
+     * @param string $alias The alias.
+     * @throws ORMException
+     * @return string The full namespace.
+     */
+    public function getAliasNamespace($alias)
+    {
+        foreach (array_keys($this->getManagers()) as $name) {
+            try {
+                return $this->getManager($name)->getConfiguration()->getEntityNamespace($alias);
+            } catch (ORMException $e) {
+            }
+        }
+
+        throw ORMException::unknownEntityNamespace($alias);
+    }
+}

src/LaravelDoctrineServiceProvider.php

@@ -1,6 +1,9 @@
 <?php namespace Mitch\LaravelDoctrine;
 
 use App;
+use Illuminate\Container\Container;
+use Doctrine\Common\Persistence\ManagerRegistry;
+use Doctrine\DBAL\DriverManager;
 use Doctrine\ORM\EntityManagerInterface;
 use Doctrine\ORM\Events;
 use Doctrine\ORM\EntityManager;
@@ -38,6 +41,7 @@ public function register()
     {
         $this->registerConfigurationMapper();
         $this->registerCacheManager();
+        $this->registerManagerRegistry();
         $this->registerEntityManager();
         $this->registerClassMetadataFactory();
 
@@ -76,9 +80,9 @@ public function registerCacheManager()
         });
     }
 
-    private function registerEntityManager()
+    private function registerManagerRegistry()
     {
-        $this->app->singleton(EntityManager::class, function ($app) {
+        $this->app->singleton(IlluminateRegistry::class, function ($app) {
             $config = $app['config']['doctrine::doctrine'];
             $metadata = Setup::createAnnotationMetadataConfiguration(
                 $config['metadata'],
@@ -97,9 +101,33 @@ private function registerEntityManager()
 
             $eventManager = new EventManager;
             $eventManager->addEventListener(Events::onFlush, new SoftDeletableListener);
-            $entityManager = EntityManager::create($this->mapLaravelToDoctrineConfig($app['config']), $metadata, $eventManager);
-            $entityManager->getFilters()->enable('trashed');
-            return $entityManager;
+
+            $defaultDatabase = $app['config']['database']['default'];
+            $databaseConnections = $app['config']['database']['connections'];
+
+            $registyConnections = array();
+            $registryManagers = array();
+
+            foreach ($databaseConnections as $connectionName => $databaseConfig) {
+                $connection = DriverManager::getConnection($this->mapLaravelToDoctrineConfig($databaseConfig), $metadata, $eventManager);
+                $registyConnections[$connectionName] = "doctrine.dbal.{$connectionName}_connection";
+                $app->instance($registyConnections[$connectionName], $connection);
+
+                $entityManager = EntityManager::create($connection, $metadata, $eventManager);
+                $entityManager->getFilters()->enable('trashed');
+                $registryManagers[$connectionName] = "doctrine.orm.{$connectionName}_entity_manager";
+                $app->instance($registryManagers[$connectionName], $entityManager);
+            }
+
+            return new IlluminateRegistry($app, $registyConnections, $registryManagers, $defaultDatabase, $defaultDatabase);
+        });
+        $this->app->singleton(ManagerRegistry::class, IlluminateRegistry::class);
+    }
+
+    private function registerEntityManager()
+    {
+        $this->app->singleton(EntityManager::class, function ($app) {
+            return $app->make(IlluminateRegistry::class)->getManager();
         });
         $this->app->singleton(EntityManagerInterface::class, EntityManager::class);
     }
@@ -135,19 +163,19 @@ public function provides()
             ClassMetadataFactory::class,
             DriverMapper::class,
             AuthManager::class,
+            ManagerRegistry::class,
+            IlluminateRegistry::class,
         ];
     }
 
     /**
      * Map Laravel's to Doctrine's database configuration requirements.
-     * @param $config
+     * @param $databaseConfig
      * @throws \Exception
      * @return array
      */
-    private function mapLaravelToDoctrineConfig($config)
+    private function mapLaravelToDoctrineConfig($databaseConfig)
     {
-        $default = $config['database.default'];
-        $connection = $config["database.connections.{$default}"];
-        return App::make(DriverMapper::class)->map($connection);
+        return App::make(DriverMapper::class)->map($databaseConfig);
     }
 }