Allow collect initial queries

Author: natanfelles

src/Database.php

@@ -93,8 +93,12 @@ public function __construct(
         string $schema = null,
         string $host = 'localhost',
         int $port = 3306,
-        Logger $logger = null
+        Logger $logger = null,
+        DatabaseCollector $collector = null
     ) {
+        if ($collector) {
+            $this->setDebugCollector($collector);
+        }
         $this->logger = $logger;
         $this->connect($username, $password, $schema, $host, $port);
     }
@@ -260,13 +264,21 @@ protected function setCollations(string $charset, string $collation) : bool
         $this->mysqli->set_charset($charset);
         $charset = $this->quote($charset);
         $collation = $this->quote($collation);
-        return $this->mysqli->real_query("SET NAMES {$charset} COLLATE {$collation}");
+        $statement = "SET NAMES {$charset} COLLATE {$collation}";
+        $this->lastQuery = $statement;
+        return isset($this->debugCollector)
+            ? $this->addToDebug(fn () => $this->mysqli->real_query($statement))
+            : $this->mysqli->real_query($statement);
     }
 
     protected function setTimezone(string $timezone) : bool
     {
         $timezone = $this->quote($timezone);
-        return $this->mysqli->real_query("SET time_zone = {$timezone}");
+        $statement = "SET time_zone = {$timezone}";
+        $this->lastQuery = $statement;
+        return isset($this->debugCollector)
+            ? $this->addToDebug(fn () => $this->mysqli->real_query($statement))
+            : $this->mysqli->real_query($statement);
     }
 
     /**

tests/DatabaseTest.php

@@ -10,6 +10,7 @@
 namespace Tests\Database;
 
 use Framework\Database\Database;
+use Framework\Database\Debug\DatabaseCollector;
 use Framework\Database\Definition\AlterSchema;
 use Framework\Database\Definition\AlterTable;
 use Framework\Database\Definition\CreateSchema;
@@ -115,6 +116,24 @@ public function testConnectionFailWithLogger() : void
         }
     }
 
+    public function testConnectionWithDebugCollector() : void
+    {
+        $collector = new DatabaseCollector();
+        self::assertCount(0, $collector->getData());
+        $config = [
+            'username' => \getenv('DB_USERNAME'),
+            'password' => \getenv('DB_PASSWORD'),
+            'host' => \getenv('DB_HOST'),
+            'port' => \getenv('DB_PORT'),
+        ];
+        $database = new Database($config, collector: $collector);
+        self::assertCount(2, $collector->getData());
+        self::assertSame(
+            "SET time_zone = '+00:00'",
+            $database->getLastQuery()
+        );
+    }
+
     protected function mariadbHasSSL() : bool
     {
         $version = static::$database->getConnection()->get_server_info();