add full-rollback option for production migrations

src/Bridges/SymfonyConsole/ContinueCommand.php

@@ -25,13 +25,16 @@ protected function configure()
 		$this->setDescription('Updates database schema by running all new migrations');
 		$this->setHelp("If table 'migrations' does not exist in current database, it is created automatically.");
 		$this->addOption('production', NULL, InputOption::VALUE_NONE, 'Will not import dummy data');
+		$this->addOption('full-rollback', 'r', InputOption::VALUE_NONE, 'Upon failing, rollback all migrations, not only the failed on. <comment>Only works reliably with PostgreSQL.</comment>');
 	}
 
 
 	protected function execute(InputInterface $input, OutputInterface $output)
 	{
+		$mode = $input->getOption('full-rollback') ? Runner::MODE_CONTINUE_FULL_ROLLBACK : Runner::MODE_CONTINUE;
+
 		$withDummy = !$input->getOption('production');
-		$this->runMigrations(Runner::MODE_CONTINUE, $withDummy);
+		$this->runMigrations($mode, $withDummy);
 	}
 
 }

src/Engine/Runner.php

@@ -25,6 +25,7 @@ class Runner
 {
 	/** @const modes */
 	const MODE_CONTINUE = 'continue';
+	const MODE_CONTINUE_FULL_ROLLBACK = 'continue-full-rollback';
 	const MODE_RESET = 'reset';
 	const MODE_INIT = 'init';
 
@@ -80,7 +81,7 @@ public function addExtensionHandler($extension, IExtensionHandler $handler)
 
 
 	/**
-	 * @param  string $mode self::MODE_CONTINUE|self::MODE_RESET|self::MODE_INIT
+	 * @param  string $mode self::MODE_CONTINUE|self::MODE_CONTINUE_FULL_ROLLBACK|self::MODE_RESET|self::MODE_INIT
 	 * @return void
 	 */
 	public function run($mode = self::MODE_CONTINUE)
@@ -103,15 +104,30 @@ public function run($mode = self::MODE_CONTINUE)
 				$this->printer->printReset();
 			}
 
-			$this->driver->createTable();
-			$migrations = $this->driver->getAllMigrations();
-			$files = $this->finder->find($this->groups, array_keys($this->extensionsHandlers));
-			$toExecute = $this->orderResolver->resolve($migrations, $this->groups, $files, $mode);
-			$this->printer->printToExecute($toExecute);
-
-			foreach ($toExecute as $file) {
-				$queriesCount = $this->execute($file);
-				$this->printer->printExecute($file, $queriesCount);
+			$this->driver->beginTransaction();
+			try {
+				$this->driver->createTable();
+				$migrations = $this->driver->getAllMigrations();
+				$files = $this->finder->find($this->groups, array_keys($this->extensionsHandlers));
+				$toExecute = $this->orderResolver->resolve($migrations, $this->groups, $files, $mode);
+				$this->printer->printToExecute($toExecute);
+
+				foreach ($toExecute as $file) {
+					$queriesCount = $this->execute($file);
+					$this->printer->printExecute($file, $queriesCount);
+				}
+				$this->driver->commitTransaction();
+
+			} catch (\Exception $e) {
+				if ($mode === self::MODE_CONTINUE_FULL_ROLLBACK) {
+					// rollback all migrations executed in this run
+					$this->driver->rollbackTransaction();
+
+				} else if ($mode === self::MODE_CONTINUE) {
+					// commit migrations not including the failing one
+					$this->driver->commitTransaction();
+				}
+				throw $e;
 			}
 
 			$this->driver->unlock();

tests/cases/integration/Runner.Rollback.phpt

@@ -0,0 +1,73 @@
+<?php
+
+/**
+ * @testCase
+ * @dataProvider ../../dbals.ini
+ */
+
+namespace NextrasTests\Migrations;
+
+use Mockery;
+use Nextras\Migrations\Engine\Runner;
+use Nextras\Migrations\Entities\Group;
+use Tester;
+use Tester\Assert;
+
+require __DIR__ . '/../../bootstrap.php';
+
+
+class RollbackTest extends IntegrationTestCase
+{
+
+	protected function getGroups($dir)
+	{
+		$rollback = new Group();
+		$rollback->enabled = TRUE;
+		$rollback->name = 'rollback';
+		$rollback->directory = $dir . '/rollback';
+		$rollback->dependencies = [];
+
+		return [$rollback];
+	}
+
+
+	/**
+	 * @param $mode
+	 * @return bool table exists
+	 */
+	private function runInMode($mode)
+	{
+		try {
+			$this->runner->run($mode);
+		} catch (\Exception $e) {
+		}
+
+		$res = $this->dbal->query('
+			SELECT Count(*) ' . $this->dbal->escapeIdentifier('count') . '
+			FROM information_schema.tables
+			WHERE table_name = ' . $this->dbal->escapeString('rollback') . '
+				AND table_schema = ' . $this->dbal->escapeString($this->dbName) . '
+		');
+		return (bool) $res[0]['count'];
+	}
+
+
+	public function testContinueRollbacksFailingOnly()
+	{
+		Assert::true($this->runInMode(Runner::MODE_CONTINUE));
+		Assert::count(2, $this->driver->getAllMigrations());
+	}
+
+
+	public function testFullRollback()
+	{
+		$this->driver->createTable();
+
+		Assert::false($this->runInMode(Runner::MODE_CONTINUE_FULL_ROLLBACK));
+		Assert::count(0, $this->driver->getAllMigrations());
+	}
+
+}
+
+
+(new RollbackTest)->run();

tests/fixtures/mysql/rollback/001.sql

@@ -0,0 +1,4 @@
+CREATE TABLE `rollback` (
+  `id` bigint NOT NULL,
+  PRIMARY KEY (`id`)
+);

tests/fixtures/mysql/rollback/002.sql

@@ -0,0 +1 @@
+INSERT INTO `rollback` (`id`) VALUES (1), (2), (3);

tests/fixtures/mysql/rollback/003.sql

@@ -0,0 +1 @@
+INSERT INTO `rollback` (`id`) VALUES (3); -- duplicate key

tests/fixtures/pgsql/rollback/001.sql

@@ -0,0 +1,4 @@
+CREATE TABLE "rollback" (
+  "id" serial4 NOT NULL,
+  PRIMARY KEY ("id")
+);

tests/fixtures/pgsql/rollback/002.sql

@@ -0,0 +1 @@
+INSERT INTO "rollback" ("id") VALUES (1), (2), (3);

tests/fixtures/pgsql/rollback/003.sql

@@ -0,0 +1 @@
+INSERT INTO "rollback" ("id") VALUES (3); -- duplicate key