Add PostgreSQL connection

leomarquine · Aug 17, 2016 · 2c96730 · 2c96730
1 parent f39e6de
commit 2c96730
Show file tree

Hide file tree

Showing 3 changed files with 115 additions and 2 deletions.
diff --git a/docs/index.md b/docs/index.md
@@ -33,7 +33,7 @@ MySQL connection
 ```php
 $connection = [
     'host' => localhost,
-    'port' => 3306,
+    'port' => '3306',
     'database' => dbname,
     'username' => user,
     'password' => pass,
@@ -42,6 +42,20 @@ $connection = [
 ];
 ```
 
+PostgreSQL connection
+```php
+$connection = [
+    'driver' => 'pgsql',
+    'host' => 'localhost',
+    'port' => '5432',
+    'database' => 'dbname',
+    'username' => 'user',
+    'password' => 'pass',
+    'charset' => 'utf8',
+    'schema' => 'public'
+];
+```
+
 Adding connections:
 ```php
 use Marquine\Etl\Job;

diff --git a/src/Database/Connectors/ConnectionFactory.php b/src/Database/Connectors/ConnectionFactory.php
@@ -28,7 +28,7 @@ public static function make($config)
     /**
     * Select the database connector.
     *
-    * @param string driver
+    * @param string $driver
     * @return \Marquine\Etl\Database\Connectors\Connector
     */
     public static function selectConnector($driver)
@@ -38,6 +38,8 @@ public static function selectConnector($driver)
                 return new SqliteConnector;
             case 'mysql':
                 return new MySqlConnector;
+            case 'pgsql':
+                return new PostgresConnector;
         }
 
         throw new InvalidArgumentException('The specified driver is not valid.');

diff --git a/src/Database/Connectors/PostgresConnector.php b/src/Database/Connectors/PostgresConnector.php
@@ -0,0 +1,97 @@
+<?php
+
+namespace Marquine\Etl\Database\Connectors;
+
+use PDO;
+
+class PostgresConnector extends Connector
+{
+    /**
+    * Connect to a database.
+    *
+    * @param array $config
+    * @return \PDO
+    */
+    public function connect($config)
+    {
+        $dsn = $this->getDsn($config);
+
+        $connection = $this->createConnection($dsn, $config);
+
+        $this->afterConnection($connection, $config);
+
+        return $connection;
+    }
+
+    /**
+     * Get the DSN string.
+     *
+     * @param array $config
+     * @return string
+     */
+    public function getDsn($config)
+    {
+        extract($config, EXTR_SKIP);
+
+        $dsn = [];
+
+        if (isset($host)) {
+            $dsn['host'] = $host;
+        }
+
+        if (isset($port)) {
+            $dsn['port'] = $port;
+        }
+
+        if (isset($database)) {
+            $dsn['dbname'] = $database;
+        }
+
+        return 'pgsql:' . http_build_query($dsn, '', ';');
+    }
+
+    /**
+     * Handle tasks after connection.
+     *
+     * @param \PDO $connection
+     * @param array $config
+     * @return void
+     */
+    public function afterConnection($connection, $config)
+    {
+        extract($config, EXTR_SKIP);
+
+        if (isset($charset)) {
+            $connection->prepare("set names '$charset'")->execute();
+        }
+
+        if (isset($timezone)) {
+            $connection->prepare("set time zone '$timezone'")->execute();
+        }
+
+        if (isset($schema)) {
+            $schema = $this->formatSchema($schema);
+
+            $connection->prepare("set search_path to $schema")->execute();
+        }
+
+        if (isset($application_name)) {
+            $connection->prepare("set application_name to '$application_name'")->execute();
+        }
+    }
+
+    /**
+     * Format the schema.
+     *
+     * @param array|string $schema
+     * @return string
+     */
+    public function formatSchema($schema)
+    {
+        if (is_string($schema)) {
+            $schema = [$schema];
+        }
+
+        return implode(', ', $schema);
+    }
+}