Merge pull request #527 from utopia-php/feat-pdo-proxy

Add PDO proxy

Author: abnegate

bin/tasks/index.php

@@ -13,6 +13,7 @@
 use Utopia\Database\Adapter\Mongo;
 use Utopia\Database\Adapter\MySQL;
 use Utopia\Database\Database;
+use Utopia\Database\PDO;
 use Utopia\Mongo\Client;
 use Utopia\Validator\Text;
 

bin/tasks/load.php

@@ -20,6 +20,7 @@
 use Utopia\Database\Document;
 use Utopia\Database\Helpers\Permission;
 use Utopia\Database\Helpers\Role;
+use Utopia\Database\PDO;
 use Utopia\Database\Validator\Authorization;
 use Utopia\Mongo\Client;
 use Utopia\Validator\Numeric;

composer.lock

@@ -0,0 +1,58 @@
+<?php
+
+namespace Utopia\Database;
+
+/**
+ * A PDO wrapper that forwards method calls to the internal PDO instance.
+ *
+ * @mixin \PDO
+ */
+class PDO
+{
+    protected \PDO $pdo;
+
+    /**
+     * @param string $dsn
+     * @param ?string $username
+     * @param ?string $password
+     * @param array<mixed> $config
+     */
+    public function __construct(
+        protected string $dsn,
+        protected ?string $username,
+        protected ?string $password,
+        protected array $config = []
+    ) {
+        $this->pdo = new \PDO(
+            $this->dsn,
+            $this->username,
+            $this->password,
+            $this->config
+        );
+    }
+
+    /**
+     * @param string $method
+     * @param array<mixed> $args
+     * @return mixed
+     */
+    public function __call(string $method, array $args): mixed
+    {
+        return $this->pdo->{$method}(...$args);
+    }
+
+    /**
+     * Create a new connection to the database
+     *
+     * @return void
+     */
+    public function reconnect(): void
+    {
+        $this->pdo = new \PDO(
+            $this->dsn,
+            $this->username,
+            $this->password,
+            $this->config
+        );
+    }
+}

src/Database/PDO.php

@@ -2,12 +2,12 @@
 
 namespace Tests\E2E\Adapter;
 
-use PDO;
 use Redis;
 use Utopia\Cache\Adapter\Redis as RedisAdapter;
 use Utopia\Cache\Cache;
 use Utopia\Database\Adapter\MariaDB;
 use Utopia\Database\Database;
+use Utopia\Database\PDO;
 
 class MariaDBTest extends Base
 {

tests/e2e/Adapter/MariaDBTest.php

@@ -2,7 +2,6 @@
 
 namespace Tests\E2E\Adapter;
 
-use PDO;
 use Redis;
 use Utopia\Cache\Adapter\Redis as RedisAdapter;
 use Utopia\Cache\Cache;
@@ -18,6 +17,7 @@
 use Utopia\Database\Helpers\Permission;
 use Utopia\Database\Helpers\Role;
 use Utopia\Database\Mirror;
+use Utopia\Database\PDO;
 
 class MirrorTest extends Base
 {

tests/e2e/Adapter/MirrorTest.php

@@ -2,12 +2,12 @@
 
 namespace Tests\E2E\Adapter;
 
-use PDO;
 use Redis;
 use Utopia\Cache\Adapter\Redis as RedisAdapter;
 use Utopia\Cache\Cache;
 use Utopia\Database\Adapter\MySQL;
 use Utopia\Database\Database;
+use Utopia\Database\PDO;
 
 class MySQLTest extends Base
 {

tests/e2e/Adapter/MySQLTest.php

@@ -2,12 +2,12 @@
 
 namespace Tests\E2E\Adapter;
 
-use PDO;
 use Redis;
 use Utopia\Cache\Adapter\Redis as RedisAdapter;
 use Utopia\Cache\Cache;
 use Utopia\Database\Adapter\Postgres;
 use Utopia\Database\Database;
+use Utopia\Database\PDO;
 
 class PostgresTest extends Base
 {

tests/e2e/Adapter/PostgresTest.php

@@ -2,12 +2,12 @@
 
 namespace Tests\E2E\Adapter;
 
-use PDO;
 use Redis;
 use Utopia\Cache\Adapter\Redis as RedisAdapter;
 use Utopia\Cache\Cache;
 use Utopia\Database\Adapter\SQLite;
 use Utopia\Database\Database;
+use Utopia\Database\PDO;
 
 class SQLiteTest extends Base
 {

tests/e2e/Adapter/SQLiteTest.php

@@ -2,13 +2,13 @@
 
 namespace Tests\E2E\Adapter\SharedTables;
 
-use PDO;
 use Redis;
 use Tests\E2E\Adapter\Base;
 use Utopia\Cache\Adapter\Redis as RedisAdapter;
 use Utopia\Cache\Cache;
 use Utopia\Database\Adapter\MariaDB;
 use Utopia\Database\Database;
+use Utopia\Database\PDO;
 
 class MariaDBTest extends Base
 {