Implement aliased connections for balancing

src/Pckg/Database/Entity.php

@@ -88,7 +88,7 @@ public function __construct(Repository $repository = null, $alias = null, $boot
         $this->repository = $repository;
 
         $this->guessDefaults();
-        
+
         if ($boot) {
             $this->initExtensions();
             $this->preboot();
@@ -539,7 +539,9 @@ public function one()
     {
         $this->applyExtensions();
 
-        $one = $this->repository->one($this);
+        $repository = $this->getRepository()->aliased('read');
+
+        $one = $repository->one($this);
 
         $this->resetQuery();
 
@@ -577,7 +579,9 @@ public function all()
     {
         $this->applyExtensions();
 
-        $all = $this->repository->all($this);
+        $repository = $this->getRepository()->aliased('read');
+
+        $all = $repository->all($this);
 
         $this->resetQuery();
 
@@ -663,6 +667,8 @@ public function delete(Repository $repository = null)
             $repository = $this->getRepository();
         }
 
+        $repository = $repository->aliased('write');
+
         $delete = $this->getQuery()->transformToDelete();
 
         $prepare = $repository->prepareQuery($delete);
@@ -681,6 +687,8 @@ public function insert(Repository $repository = null)
             $repository = $this->getRepository();
         }
 
+        $repository = $repository->aliased('write');
+
         $insert = $this->getQuery()->transformToInsert();
 
         $prepare = $repository->prepareQuery($insert);
@@ -711,6 +719,8 @@ public function update(Repository $repository = null)
             $repository = $this->getRepository();
         }
 
+        $repository = $repository->aliased('write');
+
         $update = $this->getQuery()->transformToUpdate();
 
         $update->setSet($this->setData);

src/Pckg/Database/Record/Extension/Translatable.php

@@ -28,6 +28,8 @@ public function deleteTranslation($language, Entity $entity = null, Repository $
             $repository = $entity->getRepository();
         }
 
+        $repository = $repository->aliased('write');
+
         $deleted = $repository->deleteTranslation($this, $entity, $language);
 
         return $deleted;

src/Pckg/Database/Repository/AbstractRepository.php

@@ -22,6 +22,11 @@ abstract class AbstractRepository implements Repository
      */
     protected $cache;
 
+    /**
+     * @var array
+     */
+    protected $aliases = [];
+
     /**
      * @return PDO
      */
@@ -30,6 +35,29 @@ public function getConnection()
         return $this->connection;
     }
 
+    /**
+     * @param            $alias
+     * @param Repository $repository
+     *
+     * @return $this
+     */
+    public function addAlias($alias, Repository $repository)
+    {
+        $this->aliases[$alias] = $repository;
+
+        return $this;
+    }
+
+    /**
+     * @param $alias
+     *
+     * @return mixed|AbstractRepository
+     */
+    public function aliased($alias)
+    {
+        return $this->aliases[$alias] ?? $this;
+    }
+
     /**
      * @param \PDO $connection
      */

src/Pckg/Database/Repository/PDO.php

@@ -199,16 +199,6 @@ public function executePrepared($prepare)
     {
         $execute = $prepare->execute();
 
-        /*$caller = db(0, 6, false)[0];
-        message(
-            'query in ' .
-            ($caller['class'] ?? null) .
-            ($caller['type'] ?? null) .
-            ($caller['function'] ?? null) . ':' .
-            ($caller['line'] ?? null),
-            'optimize'
-        );*/
-
         if (!$execute) {
             $errorInfo = $prepare->errorInfo();
 

src/Pckg/Database/Repository/PDO/Command/DeleteRecord.php

@@ -48,7 +48,7 @@ public function __construct(Record $record, Entity $entity, Repository $reposito
     {
         $this->record = $record;
         $this->entity = $entity;
-        $this->repository = $repository;
+        $this->repository = $repository->aliased('write');
     }
 
     /**

src/Pckg/Database/Repository/PDO/Command/GetRecords.php

@@ -32,8 +32,8 @@ class GetRecords
     public function __construct(Entity $entity, Repository $repository = null)
     {
         $this->entity = $entity;
-        $this->repository = $repository
-            ?: $this->entity->getRepository();
+        $this->repository = ($repository
+                             ?? $this->entity->getRepository())->aliased('read');
     }
 
     /**
@@ -61,6 +61,7 @@ public function executeAll()
             $collection->setEntity($entity)->setSaved()->setOriginalFromData();
 
             stopMeasure('Executing ' . $measure);
+
             return $entity->fillCollectionWithRelations($collection);
         }
         stopMeasure('Executing ' . $measure);
@@ -86,6 +87,7 @@ public function executeOne()
             $record->setEntity($entity)->setSaved()->setOriginalFromData();
 
             stopMeasure('Executing ' . $measure);
+
             return $entity->fillRecordWithRelations($record);
         }
         stopMeasure('Executing ' . $measure);

src/Pckg/Database/Repository/PDO/Command/InsertRecord.php

@@ -44,7 +44,7 @@ public function __construct(Record $record, Entity $entity, Repository $reposito
     {
         $this->record = $record;
         $this->entity = $entity;
-        $this->repository = $repository;
+        $this->repository = $repository->aliased('write');
     }
 
     /**

src/Pckg/Database/Repository/PDO/Command/UpdateRecord.php

@@ -44,7 +44,7 @@ public function __construct(Record $record, Entity $entity, Repository $reposito
     {
         $this->record = $record;
         $this->entity = $entity;
-        $this->repository = $repository;
+        $this->repository = $repository->aliased('write');
     }
 
     /**

src/Pckg/Database/Repository/RepositoryFactory.php

@@ -44,7 +44,8 @@ public static function getOrCreateRepository($name)
         if (array_key_exists($name, static::$repositories)) {
             $repository = static::$repositories[$name];
         } else {
-            if (!context()->exists($name)) {
+            $fullName = Repository::class . '.' . $name;
+            if (!context()->exists($fullName)) {
                 /**
                  * Lazy load.
                  */
@@ -53,10 +54,10 @@ public static function getOrCreateRepository($name)
                     throw new Exception("No config found for database connection " . $name);
                 }
                 $repository = RepositoryFactory::initPdoDatabase($config, $name);
-                context()->bind($name, $repository);
+                context()->bind($fullName, $repository);
             }
 
-            $repository = context()->get($name);
+            $repository = context()->get($fullName);
         }
 
         if (!$repository) {
@@ -121,7 +122,7 @@ protected static function checkDebugBar($pdo, $name)
                 $debugBar->addCollector($pdoCollector = new PDOCollector());
             }
 
-            $pdoCollector->addConnection($tracablePdo, $name);
+            $pdoCollector->addConnection($tracablePdo, str_replace(':', '-', $name));
         }
     }
 
@@ -168,7 +169,19 @@ public static function createRepositoryConnection($config, $name)
         /**
          * Bind repository to context so we can reuse it later.
          */
-        context()->bindIfNot(Repository::class, $repository);
+        if ($pos = strpos($name, ':')) {
+            /**
+             * We're probably initializing write connection.
+             */
+            $originalAlias = Repository::class . '.' . substr($name, 0, $pos);
+            $originalRepository = context()->get($originalAlias);
+            $alias = substr($name, $pos + 1);
+            $originalRepository->addAlias($alias, $repository);
+            $repository->addAlias($alias == 'write' ? 'read' : 'write', $originalRepository);
+        } else {
+            context()->bindIfNot(Repository::class, $repository);
+        }
+
         context()->bind(Repository::class . '.' . $name, $repository);
 
         return $repository;