Merge pull request #3 from netlogix/feature/increase-database-timeout

FEATURE: Increase database timeout before running migrations
netlogix · Jan 9, 2020 · a5b85bc · a5b85bc
2 parents b369b56 + 7729ac9
commit a5b85bc
Showing 1 changed file with 28 additions and 1 deletion.
diff --git a/Classes/Command/MigrationsCommandController.php b/Classes/Command/MigrationsCommandController.php
@@ -3,13 +3,17 @@
 
 namespace Netlogix\Migrations\Command;
 
+use Doctrine\Common\Persistence\ObjectManager as DoctrineObjectManager;
+use Doctrine\DBAL\Connection;
+use Doctrine\ORM\EntityManager as DoctrineEntityManager;
 use Neos\Flow\Cli\CommandController;
 use Neos\Flow\Log\ThrowableStorageInterface;
 use Neos\Flow\Log\Utility\LogEnvironment;
 use Netlogix\Migrations\Domain\Service\MigrationExecutor;
 use Netlogix\Migrations\Domain\Service\MigrationService;
 use Psr\Log\LoggerInterface;
 use Neos\Flow\Annotations as Flow;
+use RuntimeException;
 
 /**
  * @Flow\Scope("singleton")
@@ -36,18 +40,25 @@ class MigrationsCommandController extends CommandController
      */
     private $logger;
 
+    /**
+     * @var DoctrineObjectManager
+     */
+    private $doctrineObjectManager;
+
     public function __construct(
         MigrationService $migrationService,
         MigrationExecutor $migrationExecutor,
         ThrowableStorageInterface $throwableStorage,
-        LoggerInterface $logger
+        LoggerInterface $logger,
+        DoctrineObjectManager $doctrineObjectManager
     ) {
         parent::__construct();
 
         $this->migrationService = $migrationService;
         $this->migrationExecutor = $migrationExecutor;
         $this->throwableStorage = $throwableStorage;
         $this->logger = $logger;
+        $this->doctrineObjectManager = $doctrineObjectManager;
     }
 
     /**
@@ -64,6 +75,8 @@ public function migrateCommand(bool $quiet = false)
             $this->sendAndExit(0);
         }
 
+        $this->increaseDatabaseTimeout();
+
         foreach ($unexecutedMigrations as $version => $migration) {
             try {
                 $this->migrationExecutor->execute($migration, 'up', $this->output);
@@ -85,6 +98,7 @@ public function migrateCommand(bool $quiet = false)
     public function executeCommand(string $version, string $direction = 'up')
     {
         try {
+            $this->increaseDatabaseTimeout();
             $migration = $this->migrationService->getMigrationByVersion($version);
             $this->migrationExecutor->execute($migration, $direction, $this->output);
         } catch (\Exception $exception) {
@@ -102,4 +116,17 @@ protected function handleException(\Exception $exception)
         $this->logger->error($message, LogEnvironment::fromMethodName(__METHOD__));
         $this->quit(1);
     }
+
+    protected function increaseDatabaseTimeout($timeout = 3600): void
+    {
+        ini_set('default_socket_timeout', (string)$timeout);
+        if (!$this->doctrineObjectManager instanceof DoctrineEntityManager) {
+            throw new RuntimeException('No Doctrine EntityManager found, cannot increase MySQL timeout');
+        }
+        $connection = $this->doctrineObjectManager->getConnection();
+        if (!$connection || !$connection instanceof Connection) {
+            throw new RuntimeException('No Doctrine Connection found, cannot increase MySQL timeout');
+        }
+        $connection->exec(sprintf('SET SESSION wait_timeout = %d;', $timeout));
+    }
 }