Wikibase\QueryEngine\SQLStore\Setup\Updater

Complexity

Total Complexity

9

Size/Duplication

Total Lines	55
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	6

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	update()	0	10	3
A	handleTable()	0	8	2
A	createTable()	0	3	1
A	migrateTable()	0	12	2

<?php

namespace Wikibase\QueryEngine\SQLStore\Setup;

use Doctrine\DBAL\DBALException;
use Doctrine\DBAL\Schema\AbstractSchemaManager;
use Doctrine\DBAL\Schema\Comparator;
use Doctrine\DBAL\Schema\Table;
use Psr\Log\LoggerInterface;
use Wikibase\QueryEngine\QueryEngineException;
use Wikibase\QueryEngine\QueryStoreUpdater;
use Wikibase\QueryEngine\SQLStore\StoreSchema;

/**
 * @licence GNU GPL v2+
 * @author Jeroen De Dauw < jeroendedauw@gmail.com >
 */
class Updater implements QueryStoreUpdater {

	private $logger;
	private $storeSchema;
	private $schemaManager;

	public function __construct( LoggerInterface $logger, StoreSchema $storeSchema, AbstractSchemaManager $schemaManager ) {
		$this->logger = $logger;
		$this->storeSchema = $storeSchema;
		$this->schemaManager = $schemaManager;
	}

	/**
	 * @see QueryStoreUpdater::update
	 *
	 * @throws QueryEngineException
	 */
	public function update() {
		foreach ( $this->storeSchema->getTables() as $table ) {
			try {
				$this->handleTable( $table );
			}
			catch ( DBALException $ex ) {
				$this->logger->alert( $ex->getMessage(), array( 'exception' => $ex ) );
			}
		}
	}

	private function handleTable( Table $table ) {
		if ( $this->schemaManager->tablesExist( array( $table->getName() ) ) ) {
			$this->migrateTable( $table );
		}
		else {
			$this->createTable( $table );
		}
	}

	private function createTable( Table $table ) {
		$this->schemaManager->createTable( $table );
	}

	private function migrateTable( Table $table ) {
		$comparator = new Comparator();

		$tableDiff = $comparator->diffTable(
			$this->schemaManager->listTableDetails( $table->getName() ),
			$table
		);

		if ( $tableDiff !== false ) {
			$this->schemaManager->alterTable( $tableDiff );
		}
	}

}