[FEATURE] Add read write connection support

src/EntityManagerFactory.php

@@ -2,6 +2,8 @@
 
 namespace LaravelDoctrine\ORM;
 
+use Doctrine\DBAL\Connections\MasterSlaveConnection;
+use Doctrine\DBAL\DriverManager;
 use Doctrine\ORM\Cache\DefaultCacheFactory;
 use Doctrine\ORM\Configuration;
 use Doctrine\ORM\EntityManager;
@@ -18,6 +20,8 @@
 use LaravelDoctrine\ORM\Configuration\MetaData\MetaDataManager;
 use LaravelDoctrine\ORM\Extensions\MappingDriverChain;
 use LaravelDoctrine\ORM\Resolvers\EntityListenerResolver;
+use LaravelDoctrine\ORM\Utilities\ArrayUtil;
+use LaravelDoctrine\ORM\Utilities\MasterSlaveConfigParser;
 use ReflectionException;
 
 class EntityManagerFactory
@@ -47,6 +51,11 @@ class EntityManagerFactory
      */
     protected $container;
 
+    /**
+     * @var array
+     */
+    protected $configCache = [];
+
     /**
      * @var Setup
      */
@@ -91,37 +100,66 @@ public function __construct(
      */
     public function create(array $settings = [])
     {
-        $configuration = $this->setup->createConfiguration(
-            array_get($settings, 'dev', false),
-            array_get($settings, 'proxies.path'),
-            $this->cache->driver()
-        );
-
-        $this->setMetadataDriver($settings, $configuration);
-
-        $driver = $this->getConnectionDriver($settings);
-
-        $connection = $this->connection->driver(
-            $driver['driver'],
-            $driver
-        );
+        $settingsHash = ArrayUtil::hashArray($settings);
 
-        $this->setNamingStrategy($settings, $configuration);
-        $this->setCustomFunctions($configuration);
-        $this->setCacheSettings($configuration);
-        $this->configureProxies($settings, $configuration);
-        $this->setCustomMappingDriverChain($settings, $configuration);
-        $this->registerPaths($settings, $configuration);
-        $this->setRepositoryFactory($settings, $configuration);
+        if (array_key_exists($settingsHash, $this->configCache)) {
+            $configuration = $this->configCache[$settingsHash]['configuration'];
+            $connectionConfig = $this->configCache[$settingsHash]['connectionConfig'];
+            $slaveConfig = $this->configCache[$settingsHash]['slaveConfig'];
+        } else {
+            $configuration = $this->setup->createConfiguration(
+                array_get($settings, 'dev', false),
+                array_get($settings, 'proxies.path'),
+                $this->cache->driver()
+            );
+            $configuration->setDefaultRepositoryClassName(
+                array_get($settings, 'repository', EntityRepository::class)
+            );
+            $configuration->setEntityListenerResolver($this->resolver);
+
+            $this->setMetadataDriver($settings, $configuration);
+            $this->setNamingStrategy($settings, $configuration);
+            $this->setCustomFunctions($configuration);
+            $this->setCacheSettings($configuration);
+            $this->configureProxies($settings, $configuration);
+            $this->setCustomMappingDriverChain($settings, $configuration);
+            $this->registerPaths($settings, $configuration);
+            $this->setRepositoryFactory($settings, $configuration);
+
+            $driverConfig = $this->getConnectionDriver($settings);
+            $slaveConfig = MasterSlaveConfigParser::hasValidConfig($driverConfig)
+                ? MasterSlaveConfigParser::parseConfig($driverConfig)
+                : [];
+
+            $connectionConfig = $this->connection->driver(
+                $driverConfig['driver'],
+                $driverConfig
+            );
 
-        $configuration->setDefaultRepositoryClassName(
-            array_get($settings, 'repository', EntityRepository::class)
-        );
+            $this->configCache[$settingsHash]['configuration'] = $configuration;
+            $this->configCache[$settingsHash]['connectionConfig'] = $connectionConfig;
+            $this->configCache[$settingsHash]['slaveConfig'] = $slaveConfig;
+        }
 
-        $configuration->setEntityListenerResolver($this->resolver);
+        if (empty($slaveConfig)) {
+            $conn = DriverManager::getConnection($connectionConfig, $configuration);
+        } else {
+            $conn = DriverManager::getConnection([
+                'wrapperClass' => MasterSlaveConnection::class,
+                'driver'       => $connectionConfig['driver'],
+                'master'       => [
+                    'user'     => array_get($slaveConfig, 'write.user', ''),
+                    'password' => array_get($slaveConfig, 'write.password', ''),
+                    'host'     => array_get($slaveConfig, 'write.host', ''),
+                    'dbname'   => array_get($slaveConfig, 'write.dbname', ''),
+                    'port'   => array_get($slaveConfig, 'write.port', ''),
+                ],
+                'slaves'       => $slaveConfig['read'],
+            ]);
+        }
 
         $manager = EntityManager::create(
-            $connection,
+            $conn,
             $configuration
         );
 

src/Utilities/ArrayUtil.php

@@ -8,4 +8,14 @@ public static function get(&$var, $default = null)
     {
         return isset($var) ? $var : $default;
     }
+
+    /**
+     * @param array $array
+     *
+     * @return string
+     */
+    public static function hashArray(array $array)
+    {
+        return md5(json_encode($array));
+    }
 }

src/Utilities/MasterSlaveConfigParser.php

@@ -0,0 +1,76 @@
+<?php
+
+namespace LaravelDoctrine\ORM\Utilities;
+
+use UnexpectedValueException;
+
+class MasterSlaveConfigParser
+{
+    /**
+     * @param array $config
+     *
+     * @throws UnexpectedValueException If config not contains valid read / write configuration.
+     *
+     * @return array If config is not valid for parse.
+     *
+     */
+    public static function parseConfig(array $config)
+    {
+        if ( ! self::hasValidConfig($config)) {
+            throw new UnexpectedValueException('Config not contains configuration for master/slave setup.');
+        }
+
+        $masterSlave = [
+            'write' => [],
+            'read'  => [],
+        ];
+
+        $masterSlave['write'] = [
+            'user'     => array_get($config, 'write.username', $config['username']),
+            'password' => array_get($config, 'write.password', $config['password']),
+            'host'     => array_get($config, 'write.host', $config['host']),
+            'dbname'   => array_get($config, 'write.database', $config['database']),
+            'port'     => array_get($config, 'write.port', $config['port']),
+        ];
+
+        foreach (array_get($config, 'read', []) as $slaveConfig) {
+            $config   = [
+                'user'     => array_get($slaveConfig, 'username', $config['username']),
+                'password' => array_get($slaveConfig, 'password', $config['password']),
+                'host'     => array_get($slaveConfig, 'host', $config['host']),
+                'dbname'   => array_get($slaveConfig, 'database', $config['database']),
+                'port'     => array_get($config, 'write.port', $config['port']),
+            ];
+
+            $masterSlave['read'][] = $config;
+        }
+
+        return $masterSlave;
+    }
+
+    /**
+     * @param array $config
+     *
+     * @return bool
+     */
+    public static function hasValidConfig(array $config)
+    {
+        if (
+            array_key_exists('read', $config)
+            && ! empty($config['read'])
+            && is_array($config['read'])
+            //All value in $config['read'] should be an array
+            && ! in_array(false, array_map(function ($readConfigValue) {
+                return is_array($readConfigValue);
+            }, $config['read']))
+            //All value in $config['read'] should have at least one config difference
+            && ! in_array(false, array_map(function (array $readConfig) {
+                return count($readConfig) > 0;
+            }, $config['read']))
+        ) {
+            return true;
+        }
+
+        return false;
+    }
+}

tests/Utilities/ArrayUtilTest.php

@@ -30,4 +30,35 @@ public function test_can_pass_custom_default_value()
 
         $this->assertEquals('default', ArrayUtil::get($values['key2'], 'default'));
     }
+
+    /**
+     * @dataProvider arrayProvider
+     *
+     * @param array $array1
+     * @param array $array2
+     * @param bool  $expectedEquals
+     */
+    public function test_hashArray_returns_unique_hash(array $array1, array $array2, $expectedEquals)
+    {
+        $hash1 = ArrayUtil::hashArray($array1);
+        $hash2 = ArrayUtil::hashArray($array2);
+
+        if ($expectedEquals) {
+            $this->assertEquals($hash1, $hash2);
+        } else {
+            $this->assertNotEquals($hash1, $hash2);
+        }
+    }
+
+    /**
+     * @return array
+     */
+    public function arrayProvider()
+    {
+        return [
+            [[], [], true],
+            [['key1' => 'value1'], ['key1' => 'value1'], true],
+            [['key1' => 'value1'], ['key1' => 'value1-NOT-EQUALS'], false],
+        ];
+    }
 }