Merge pull request #16 from netlogix/feat/postgresql

Author: paxuclus

Classes/Command/MigrationsCommandController.php

@@ -4,6 +4,7 @@
 namespace Netlogix\Migrations\Command;
 
 use Doctrine\DBAL\Connection;
+use Doctrine\DBAL\Platforms\MySqlPlatform;
 use Doctrine\ORM\EntityManagerInterface;
 use Neos\Flow\Annotations as Flow;
 use Neos\Flow\Cli\CommandController;
@@ -121,12 +122,19 @@ protected function increaseDatabaseTimeout($timeout = 3600): void
     {
         ini_set('default_socket_timeout', (string)$timeout);
         if (!$this->entityManager instanceof EntityManagerInterface) {
-            throw new RuntimeException('No Doctrine EntityManager found, cannot increase MySQL timeout');
+            throw new RuntimeException('No Doctrine EntityManager found, cannot increase database timeout');
         }
         $connection = $this->entityManager->getConnection();
         if (!$connection || !$connection instanceof Connection) {
-            throw new RuntimeException('No Doctrine Connection found, cannot increase MySQL timeout');
+            throw new RuntimeException('No Doctrine Connection found, cannot increase database timeout');
+        }
+
+        if ($connection->getDatabasePlatform() instanceof MySqlPlatform) {
+            $connection->executeStatement(sprintf('SET SESSION wait_timeout = %d;', $timeout));
+        } elseif ($connection->getDatabasePlatform()->getName() === 'postgresql') {
+            $connection->executeStatement(sprintf('SET SESSION statement_timeout = %d;', $timeout * 1000));
+        } else {
+            $this->outputLine('<error>Unsupported database platform "%s", cannot increase database timeout</error>', [$connection->getDatabasePlatform()->getName()]);
         }
-        $connection->exec(sprintf('SET SESSION wait_timeout = %d;', $timeout));
     }
 }

Migrations/Postgresql/Version20190930132259.php

@@ -0,0 +1,41 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Neos\Flow\Persistence\Doctrine\Migrations;
+
+use Doctrine\DBAL\Schema\Schema;
+use Doctrine\Migrations\AbstractMigration;
+
+/**
+ * Auto-generated Migration: Please modify to your needs!
+ */
+final class Version20190930132259 extends AbstractMigration
+{
+    public function getDescription(): string
+    {
+        return '';
+    }
+
+    public function up(Schema $schema): void
+    {
+        // this up() migration is auto-generated, please modify it to your needs
+        $this->abortIf(
+            !$this->connection->getDatabasePlatform() instanceof \Doctrine\DBAL\Platforms\PostgreSQL100Platform,
+            "Migration can only be executed safely on '\Doctrine\DBAL\Platforms\PostgreSQL100Platform'."
+        );
+
+        $this->addSql('CREATE TABLE netlogix_migrationstatus (version VARCHAR(255) NOT NULL, PRIMARY KEY(version))');
+    }
+
+    public function down(Schema $schema): void
+    {
+        // this down() migration is auto-generated, please modify it to your needs
+        $this->abortIf(
+            !$this->connection->getDatabasePlatform() instanceof \Doctrine\DBAL\Platforms\PostgreSQL100Platform,
+            "Migration can only be executed safely on '\Doctrine\DBAL\Platforms\PostgreSQL100Platform'."
+        );
+
+        $this->addSql('DROP TABLE netlogix_migrationstatus');
+    }
+}