2014-01-29 18:48:12 +04:00
|
|
|
<?php
|
|
|
|
/**
|
2015-03-26 13:44:34 +03:00
|
|
|
* @author martin-rueegg <martin.rueegg@metaworx.ch>
|
|
|
|
* @author Morris Jobke <hey@morrisjobke.de>
|
|
|
|
* @author Robin Appelman <icewind@owncloud.com>
|
|
|
|
* @author tbelau666 <thomas.belau@gmx.de>
|
|
|
|
* @author Thomas Müller <thomas.mueller@tmit.eu>
|
|
|
|
* @author Vincent Petry <pvince81@owncloud.com>
|
|
|
|
*
|
|
|
|
* @copyright Copyright (c) 2015, ownCloud, Inc.
|
|
|
|
* @license AGPL-3.0
|
|
|
|
*
|
|
|
|
* This code is free software: you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU Affero General Public License, version 3,
|
|
|
|
* as published by the Free Software Foundation.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU Affero General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU Affero General Public License, version 3,
|
|
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
|
|
*
|
2014-01-29 18:48:12 +04:00
|
|
|
*/
|
2015-02-26 13:37:37 +03:00
|
|
|
|
2014-01-29 18:48:12 +04:00
|
|
|
namespace OC\DB;
|
|
|
|
|
|
|
|
use \Doctrine\DBAL\DBALException;
|
2014-01-30 17:21:16 +04:00
|
|
|
use \Doctrine\DBAL\Schema\Index;
|
2014-01-29 18:48:12 +04:00
|
|
|
use \Doctrine\DBAL\Schema\Table;
|
|
|
|
use \Doctrine\DBAL\Schema\Schema;
|
|
|
|
use \Doctrine\DBAL\Schema\SchemaConfig;
|
2014-01-30 17:21:16 +04:00
|
|
|
use \Doctrine\DBAL\Schema\Comparator;
|
2014-12-22 12:43:56 +03:00
|
|
|
use OCP\IConfig;
|
2013-10-17 22:16:27 +04:00
|
|
|
use OCP\Security\ISecureRandom;
|
2014-01-29 18:48:12 +04:00
|
|
|
|
|
|
|
class Migrator {
|
2013-10-17 22:16:27 +04:00
|
|
|
|
2014-01-29 18:48:12 +04:00
|
|
|
/**
|
|
|
|
* @var \Doctrine\DBAL\Connection $connection
|
|
|
|
*/
|
|
|
|
protected $connection;
|
|
|
|
|
2013-10-17 22:16:27 +04:00
|
|
|
/**
|
|
|
|
* @var ISecureRandom
|
|
|
|
*/
|
|
|
|
private $random;
|
|
|
|
|
2014-12-22 12:43:56 +03:00
|
|
|
/** @var IConfig */
|
|
|
|
protected $config;
|
|
|
|
|
2014-01-29 18:48:12 +04:00
|
|
|
/**
|
|
|
|
* @param \Doctrine\DBAL\Connection $connection
|
2013-10-17 22:16:27 +04:00
|
|
|
* @param ISecureRandom $random
|
2014-12-22 12:43:56 +03:00
|
|
|
* @param IConfig $config
|
2014-01-29 18:48:12 +04:00
|
|
|
*/
|
2014-12-22 12:43:56 +03:00
|
|
|
public function __construct(\Doctrine\DBAL\Connection $connection, ISecureRandom $random, IConfig $config) {
|
2014-01-29 18:48:12 +04:00
|
|
|
$this->connection = $connection;
|
2013-10-17 22:16:27 +04:00
|
|
|
$this->random = $random;
|
2014-12-22 12:43:56 +03:00
|
|
|
$this->config = $config;
|
2014-01-29 18:48:12 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param \Doctrine\DBAL\Schema\Schema $targetSchema
|
|
|
|
*/
|
|
|
|
public function migrate(Schema $targetSchema) {
|
|
|
|
$this->applySchema($targetSchema);
|
|
|
|
}
|
|
|
|
|
2014-04-08 18:01:08 +04:00
|
|
|
/**
|
|
|
|
* @param \Doctrine\DBAL\Schema\Schema $targetSchema
|
|
|
|
* @return string
|
|
|
|
*/
|
|
|
|
public function generateChangeScript(Schema $targetSchema) {
|
|
|
|
$schemaDiff = $this->getDiff($targetSchema, $this->connection);
|
|
|
|
|
|
|
|
$script = '';
|
|
|
|
$sqls = $schemaDiff->toSql($this->connection->getDatabasePlatform());
|
|
|
|
foreach ($sqls as $sql) {
|
2013-10-17 22:16:27 +04:00
|
|
|
$script .= $this->convertStatementToScript($sql);
|
2014-04-08 18:01:08 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
return $script;
|
|
|
|
}
|
|
|
|
|
2014-01-29 18:48:12 +04:00
|
|
|
/**
|
|
|
|
* @param Schema $targetSchema
|
2014-03-27 17:44:19 +04:00
|
|
|
* @throws \OC\DB\MigrationException
|
2014-01-29 18:48:12 +04:00
|
|
|
*/
|
|
|
|
public function checkMigrate(Schema $targetSchema) {
|
|
|
|
/**
|
|
|
|
* @var \Doctrine\DBAL\Schema\Table[] $tables
|
|
|
|
*/
|
|
|
|
$tables = $targetSchema->getTables();
|
2015-01-12 15:57:46 +03:00
|
|
|
$filterExpression = $this->getFilterExpression();
|
2014-12-01 01:17:09 +03:00
|
|
|
$this->connection->getConfiguration()->
|
2015-01-12 15:57:46 +03:00
|
|
|
setFilterSchemaAssetsExpression($filterExpression);
|
2014-01-29 18:48:12 +04:00
|
|
|
$existingTables = $this->connection->getSchemaManager()->listTableNames();
|
|
|
|
|
|
|
|
foreach ($tables as $table) {
|
2014-03-14 17:32:02 +04:00
|
|
|
if (strpos($table->getName(), '.')) {
|
|
|
|
list(, $tableName) = explode('.', $table->getName());
|
|
|
|
} else {
|
|
|
|
$tableName = $table->getName();
|
|
|
|
}
|
2014-01-29 18:48:12 +04:00
|
|
|
// don't need to check for new tables
|
|
|
|
if (array_search($tableName, $existingTables) !== false) {
|
|
|
|
$this->checkTableMigrate($table);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-06-03 13:24:31 +04:00
|
|
|
/**
|
|
|
|
* Create a unique name for the temporary table
|
|
|
|
*
|
|
|
|
* @param string $name
|
|
|
|
* @return string
|
|
|
|
*/
|
|
|
|
protected function generateTemporaryTableName($name) {
|
2015-03-03 03:49:15 +03:00
|
|
|
return $this->config->getSystemValue('dbtableprefix', 'oc_') . $name . '_' . $this->random->generate(13, ISecureRandom::CHAR_LOWER . ISecureRandom::CHAR_DIGITS);
|
2014-06-03 13:24:31 +04:00
|
|
|
}
|
|
|
|
|
2014-01-29 18:48:12 +04:00
|
|
|
/**
|
|
|
|
* Check the migration of a table on a copy so we can detect errors before messing with the real table
|
|
|
|
*
|
|
|
|
* @param \Doctrine\DBAL\Schema\Table $table
|
2014-01-30 17:21:16 +04:00
|
|
|
* @throws \OC\DB\MigrationException
|
2014-01-29 18:48:12 +04:00
|
|
|
*/
|
|
|
|
protected function checkTableMigrate(Table $table) {
|
|
|
|
$name = $table->getName();
|
2014-06-03 13:24:31 +04:00
|
|
|
$tmpName = $this->generateTemporaryTableName($name);
|
2014-01-29 18:48:12 +04:00
|
|
|
|
|
|
|
$this->copyTable($name, $tmpName);
|
|
|
|
|
|
|
|
//create the migration schema for the temporary table
|
2014-01-30 17:21:16 +04:00
|
|
|
$tmpTable = $this->renameTableSchema($table, $tmpName);
|
2014-01-29 18:48:12 +04:00
|
|
|
$schemaConfig = new SchemaConfig();
|
|
|
|
$schemaConfig->setName($this->connection->getDatabase());
|
|
|
|
$schema = new Schema(array($tmpTable), array(), $schemaConfig);
|
|
|
|
|
|
|
|
try {
|
|
|
|
$this->applySchema($schema);
|
|
|
|
$this->dropTable($tmpName);
|
|
|
|
} catch (DBALException $e) {
|
2014-03-14 19:28:56 +04:00
|
|
|
// pgsql needs to commit it's failed transaction before doing anything else
|
2014-07-23 22:13:19 +04:00
|
|
|
if ($this->connection->isTransactionActive()) {
|
|
|
|
$this->connection->commit();
|
|
|
|
}
|
2014-01-29 18:48:12 +04:00
|
|
|
$this->dropTable($tmpName);
|
|
|
|
throw new MigrationException($table->getName(), $e->getMessage());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-30 17:21:16 +04:00
|
|
|
/**
|
|
|
|
* @param \Doctrine\DBAL\Schema\Table $table
|
|
|
|
* @param string $newName
|
|
|
|
* @return \Doctrine\DBAL\Schema\Table
|
|
|
|
*/
|
|
|
|
protected function renameTableSchema(Table $table, $newName) {
|
2014-03-12 17:23:00 +04:00
|
|
|
/**
|
|
|
|
* @var \Doctrine\DBAL\Schema\Index[] $indexes
|
|
|
|
*/
|
2014-01-30 17:21:16 +04:00
|
|
|
$indexes = $table->getIndexes();
|
|
|
|
$newIndexes = array();
|
|
|
|
foreach ($indexes as $index) {
|
2014-06-17 16:11:02 +04:00
|
|
|
if ($index->isPrimary()) {
|
|
|
|
// do not rename primary key
|
|
|
|
$indexName = $index->getName();
|
|
|
|
} else {
|
|
|
|
// avoid conflicts in index names
|
2015-06-08 18:59:24 +03:00
|
|
|
$indexName = $this->config->getSystemValue('dbtableprefix', 'oc_') . $this->random->generate(13, ISecureRandom::CHAR_LOWER);
|
2014-06-17 16:11:02 +04:00
|
|
|
}
|
2014-03-12 17:23:00 +04:00
|
|
|
$newIndexes[] = new Index($indexName, $index->getColumns(), $index->isUnique(), $index->isPrimary());
|
2014-01-30 17:21:16 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
// foreign keys are not supported so we just set it to an empty array
|
2014-03-14 17:32:02 +04:00
|
|
|
return new Table($newName, $table->getColumns(), $newIndexes, array(), 0, $table->getOptions());
|
2014-01-30 17:21:16 +04:00
|
|
|
}
|
|
|
|
|
2014-04-08 18:01:08 +04:00
|
|
|
protected function getDiff(Schema $targetSchema, \Doctrine\DBAL\Connection $connection) {
|
2015-01-12 15:57:46 +03:00
|
|
|
$filterExpression = $this->getFilterExpression();
|
|
|
|
$this->connection->getConfiguration()->
|
|
|
|
setFilterSchemaAssetsExpression($filterExpression);
|
2014-04-08 18:01:08 +04:00
|
|
|
$sourceSchema = $connection->getSchemaManager()->createSchema();
|
2014-01-29 18:48:12 +04:00
|
|
|
|
|
|
|
// remove tables we don't know about
|
|
|
|
/** @var $table \Doctrine\DBAL\Schema\Table */
|
|
|
|
foreach ($sourceSchema->getTables() as $table) {
|
|
|
|
if (!$targetSchema->hasTable($table->getName())) {
|
|
|
|
$sourceSchema->dropTable($table->getName());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// remove sequences we don't know about
|
|
|
|
foreach ($sourceSchema->getSequences() as $table) {
|
|
|
|
if (!$targetSchema->hasSequence($table->getName())) {
|
|
|
|
$sourceSchema->dropSequence($table->getName());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-01-30 17:21:16 +04:00
|
|
|
$comparator = new Comparator();
|
2014-04-11 17:10:09 +04:00
|
|
|
return $comparator->compare($sourceSchema, $targetSchema);
|
2014-04-08 18:01:08 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param \Doctrine\DBAL\Schema\Schema $targetSchema
|
|
|
|
* @param \Doctrine\DBAL\Connection $connection
|
|
|
|
*/
|
|
|
|
protected function applySchema(Schema $targetSchema, \Doctrine\DBAL\Connection $connection = null) {
|
|
|
|
if (is_null($connection)) {
|
|
|
|
$connection = $this->connection;
|
2014-01-29 18:48:12 +04:00
|
|
|
}
|
|
|
|
|
2014-04-08 18:01:08 +04:00
|
|
|
$schemaDiff = $this->getDiff($targetSchema, $connection);
|
|
|
|
|
2014-01-29 18:48:12 +04:00
|
|
|
$connection->beginTransaction();
|
|
|
|
foreach ($schemaDiff->toSql($connection->getDatabasePlatform()) as $sql) {
|
|
|
|
$connection->query($sql);
|
|
|
|
}
|
|
|
|
$connection->commit();
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param string $sourceName
|
|
|
|
* @param string $targetName
|
|
|
|
*/
|
|
|
|
protected function copyTable($sourceName, $targetName) {
|
|
|
|
$quotedSource = $this->connection->quoteIdentifier($sourceName);
|
|
|
|
$quotedTarget = $this->connection->quoteIdentifier($targetName);
|
|
|
|
|
2014-03-14 19:28:56 +04:00
|
|
|
$this->connection->exec('CREATE TABLE ' . $quotedTarget . ' (LIKE ' . $quotedSource . ')');
|
2014-01-29 18:48:12 +04:00
|
|
|
$this->connection->exec('INSERT INTO ' . $quotedTarget . ' SELECT * FROM ' . $quotedSource);
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param string $name
|
|
|
|
*/
|
|
|
|
protected function dropTable($name) {
|
|
|
|
$this->connection->exec('DROP TABLE ' . $this->connection->quoteIdentifier($name));
|
|
|
|
}
|
2013-10-17 22:16:27 +04:00
|
|
|
|
|
|
|
/**
|
|
|
|
* @param $statement
|
|
|
|
* @return string
|
|
|
|
*/
|
|
|
|
protected function convertStatementToScript($statement) {
|
|
|
|
$script = $statement . ';';
|
|
|
|
$script .= PHP_EOL;
|
|
|
|
$script .= PHP_EOL;
|
|
|
|
return $script;
|
|
|
|
}
|
2015-01-12 15:57:46 +03:00
|
|
|
|
|
|
|
protected function getFilterExpression() {
|
|
|
|
return '/^' . preg_quote($this->config->getSystemValue('dbtableprefix', 'oc_')) . '/';
|
|
|
|
}
|
2014-01-29 18:48:12 +04:00
|
|
|
}
|