Speedup schema cloning for MySQL

1 files changed, 23 insertions, 0 deletions

diff --git a/lib/private/DB/MySQLMigrator.php b/lib/private/DB/MySQLMigrator.php
index 1b3f70a817d..7e1194865f8 100644
--- a/lib/private/DB/MySQLMigrator.php
+++ b/lib/private/DB/MySQLMigrator.php
@@ -24,6 +24,7 @@
 namespace OC\DB;
 
 use Doctrine\DBAL\Schema\Schema;
+use Doctrine\DBAL\Schema\Table;
 
 class MySQLMigrator extends Migrator {
 	/**
@@ -48,4 +49,26 @@ class MySQLMigrator extends Migrator {
 
 		return $schemaDiff;
 	}
+	
+        /**
+         * Speed up migration test by disabling autocommit and unique indexes check
+         *
+         * @param \Doctrine\DBAL\Schema\Table $table
+         * @throws \OC\DB\MigrationException
+         */
+        protected function checkTableMigrate(Table $table) {
+                $this->connection->exec('SET autocommit=0');
+                $this->connection->exec('SET unique_checks=0');
+
+                try {
+                        parent::checkTableMigrate($table);
+                } catch (\Exception $e) {
+                        $this->connection->exec('SET unique_checks=1');
+                        $this->connection->exec('SET autocommit=1');
+                        throw new MigrationException($table->getName(), $e->getMessage());
+                }
+                $this->connection->exec('SET unique_checks=1');
+                $this->connection->exec('SET autocommit=1');
+        }
+
 }