Wikibase\QueryEngine\SQLStore\Writer

Complexity

Total Complexity

5

Size/Duplication

Total Lines	58
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	4

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	insertEntity()	0	3	1
A	updateEntity()	0	14	2
A	deleteEntity()	0	3	1

<?php

namespace Wikibase\QueryEngine\SQLStore;

use Doctrine\DBAL\Connection;
use Wikibase\DataModel\Entity\EntityDocument;
use Wikibase\QueryEngine\QueryEngineException;
use Wikibase\QueryEngine\QueryStoreWriter;
use Wikibase\QueryEngine\SQLStore\EntityStore\EntityInserter;
use Wikibase\QueryEngine\SQLStore\EntityStore\EntityRemover;

/**
 * Class responsible for writing information to the SQLStore.
 *
 * @since 0.1
 *
 * @licence GNU GPL v2+
 * @author Jeroen De Dauw < jeroendedauw@gmail.com >
 */
class Writer implements QueryStoreWriter {

	private $connection;
	private $entityInserter;
	private $entityRemover;

	/**
	 * @param Connection $connection
	 * @param EntityInserter $inserter
	 * @param EntityRemover $remover
	 */
	public function __construct( Connection $connection, EntityInserter $inserter, EntityRemover $remover ) {
		$this->connection = $connection;
		$this->entityInserter = $inserter;
		$this->entityRemover = $remover;
	}

	/**
	 * @see QueryStoreUpdater::insertEntity
	 *
	 * @param EntityDocument $entity
	 */
	public function insertEntity( EntityDocument $entity ) {
		$this->entityInserter->insertEntity( $entity );
	}

	/**
	 * @see QueryStoreUpdater::updateEntity
	 *
	 * @param EntityDocument $entity
	 *
	 * @throws QueryEngineException
	 */
	public function updateEntity( EntityDocument $entity ) {
		$this->connection->beginTransaction();

		try {
			$this->entityRemover->removeEntity( $entity );
			$this->entityInserter->insertEntity( $entity );
		}
		catch ( QueryEngineException $ex ) {
			$this->connection->rollBack();
			throw $ex;
		}

		$this->connection->commit();
	}

	/**
	 * @see QueryStoreUpdater::deleteEntity
	 *
	 * @param EntityDocument $entity
	 */
	public function deleteEntity( EntityDocument $entity ) {
		$this->entityRemover->removeEntity( $entity );
	}

}