Fix cycle/orm#60

Author: roxblnfk

src/DatabaseManager.php

@@ -13,6 +13,7 @@
 
 use Psr\Container\ContainerExceptionInterface;
 use Psr\Log\LoggerAwareInterface;
+use Psr\Log\LoggerInterface;
 use Psr\Log\NullLogger;
 use Spiral\Core\Container;
 use Spiral\Core\FactoryInterface;
@@ -89,7 +90,9 @@ final class DatabaseManager implements
     Container\SingletonInterface,
     Container\InjectorInterface
 {
-    use LoggerTrait;
+    use LoggerTrait {
+        setLogger as protected internalSetLogger;
+    }
 
     /** @var DatabaseConfig */
     private $config;
@@ -103,6 +106,20 @@ final class DatabaseManager implements
     /** @var DriverInterface[] */
     private $drivers = [];
 
+    /**
+     * Set logger for all drivers
+     */
+    public function setLogger(LoggerInterface $logger): void
+    {
+        $this->internalSetLogger($logger);
+        // Assign the logger to all initialized drivers
+        foreach ($this->drivers as $driver) {
+            if ($driver instanceof LoggerAwareInterface) {
+                $driver->setLogger($this->logger);
+            }
+        }
+    }
+
     /**
      * @param DatabaseConfig   $config
      * @param FactoryInterface $factory

tests/Database/ManagerTest.php

@@ -12,6 +12,8 @@
 
 use Mockery as m;
 use PHPUnit\Framework\TestCase;
+use Psr\Log\LoggerAwareInterface;
+use Psr\Log\LoggerInterface;
 use Spiral\Core\Container;
 use Spiral\Database\Config\DatabaseConfig;
 use Spiral\Database\Database;
@@ -130,6 +132,27 @@ public function testGetDrivers(): void
         $this->assertCount(2, $dbal->getDrivers());
     }
 
+    public function testSetLogger(): void
+    {
+        $logger = m::mock(LoggerInterface::class);
+
+        $driverWithoutLogger = m::mock(DriverInterface::class);
+
+        $driverWithLogger = m::mock(DriverInterface::class, LoggerAwareInterface::class)
+            ->shouldReceive('setLogger')->with($logger)->once()->andReturnNull()
+            ->getMock();
+        self::assertInstanceOf(LoggerAwareInterface::class, $driverWithLogger);
+
+        $dbal = new DatabaseManager(new DatabaseConfig(self::DEFAULT_OPTIONS));
+
+        $dbal->addDriver('read', $driverWithoutLogger);
+        $dbal->addDriver('write', $driverWithLogger);
+
+        $dbal->setLogger($logger);
+
+        m::close();
+    }
+
     public function testGetDatabases(): void
     {
         $read = m::mock(DriverInterface::class);