Wikibase\EntityStore\Internal\IncrementalXmlDumpReader

Complexity

Total Complexity

18

Size/Duplication

Total Lines	113
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	4

Test Coverage

Coverage

81.25%

Importance

Changes	3
Bugs	0	Features	1

7 Methods

Rating	Name	Duplication	Size	Complexity
A	__destruct()	0	3	1
A	current()	0	3	1
A	valid()	0	3	1
A	rewind()	0	4	1
A	__construct()	0	5	1
C	next()	0	39	11
A	key()	0	7	2

<?php

namespace Wikibase\EntityStore\Internal;

use Deserializers\Deserializer;
use Deserializers\Exceptions\DeserializationException;
use Iterator;
use Psr\Log\LoggerInterface;
use Wikibase\DataModel\Entity\EntityDocument;

/**
 * Internal class

 * @licence GPLv2+
 * @author Thomas Pellissier Tanon
 */
class IncrementalXmlDumpReader implements Iterator {

	/**
	 * @var resource
	 */
	private $fileStream;

	/**
	 * @var Deserializer
	 */
	private $entityDeserializer;

	/**
	 * @var EntityDocument|null
	 */
	private $currentEntity = null;

	/**
	 * @var LoggerInterface
	 */
	private $logger;

	/**
	 * @param string $fileName
	 * @param Deserializer $entityDeserializer
	 * @param LoggerInterface $logger
	 */
	public function __construct( $fileName, Deserializer $entityDeserializer, LoggerInterface $logger ) {
		$this->fileStream = fopen( $fileName, 'r' );
		$this->entityDeserializer = $entityDeserializer;
		$this->logger = $logger;
	}

	public function __destruct() {
		fclose( $this->fileStream );
	}

	/**
	 * @see Iterator::current
	 */
	public function current() {
		return $this->currentEntity;
	}

	/**
	 * @see Iterator::next
	 */
	public function next() {
		$this->currentEntity = null;

		$isEntity = false;
		$text = '';
		while( true ) {
			$line = fgets( $this->fileStream );

			if( $line === false ) {
				return;
			}

			if( preg_match( '/<model>wikibase-[a-z]+<\/model>/', $line ) ) {
				$isEntity = true;
			} elseif( preg_match( '/<text.*>(.*)<\/text>/', $line, $m ) ) {
				$text = $m[1];
			} elseif( preg_match( '/<\/page>/', $line, $m ) && $isEntity ) {

				$json = json_decode( html_entity_decode( $text ), true );
				if( $json === null ) {
					$this->logger->error( 'The serialization is not a valid JSON document.' );
					continue;
				}

				if( array_key_exists( 'redirect', $json ) ) {
					$this->logger->info( 'Entity redirection not supported.' );
					continue;
				}

				try {
					$this->currentEntity = $this->entityDeserializer->deserialize( $json );
					return;
				} catch( DeserializationException $e ) {
					$id = array_key_exists( 'id', $json ) ? $json['id'] : '';
					$this->logger->error( 'Deserialization of entity ' . $id . ' failed: ' . $e->getMessage() );
				}
			}
		}
	}

	/**
	 * @see Iterator::key
	 */
	public function key() {
		if ( $this->currentEntity === null ) {
			return null;
		}

		return $this->currentEntity->getId()->getSerialization();
	}

	/**
	 * @see Iterator::valid
	 */
	public function valid() {
		return $this->currentEntity !== null;
	}

	/**
	 * @see Iterator::rewind
	 */
	public function rewind() {
		fseek( $this->fileStream, 0 );
		$this->next();
	}
}