Wikibase\EntityStore\Console\ImportJsonDumpCommand

Complexity

Total Complexity

4

Size/Duplication

Total Lines	41
Duplicated Lines	0 %

Coupling/Cohesion

Components	0
Dependencies	9

Test Coverage

Coverage

93.75%

Importance

Changes	5
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	configure()	0	6	1
B	execute()	0	31	3

<?php

namespace Wikibase\EntityStore\Console;

use Symfony\Component\Console\Command\Command;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Logger\ConsoleLogger;
use Symfony\Component\Console\Output\OutputInterface;
use Wikibase\DataModel\Deserializers\EntityIdDeserializer;
use Wikibase\DataModel\Entity\BasicEntityIdParser;
use Wikibase\EntityStore\Config\EntityStoreFromConfigurationBuilder;
use Wikibase\EntityStore\DataModel\Deserializers\SerializedEntityDeserializer;
use Wikibase\EntityStore\Internal\JsonDumpReader;

/**
 * @licence GPLv2+
 * @author Thomas Pellissier Tanon
 */
class ImportJsonDumpCommand extends Command {

	protected function configure() {
		$this->setName( 'import-json-dump' )
			->setDescription( 'Import a JSON dump in an entity store' )
			->addArgument( 'file', InputArgument::REQUIRED, 'JSON dump file' )
			->addArgument( 'configuration', InputArgument::REQUIRED, 'Configuration file' );
	}

	protected function execute( InputInterface $input, OutputInterface $output ) {
		$configurationBuilder = new EntityStoreFromConfigurationBuilder();
		$store = $configurationBuilder->buildEntityStore( $input->getArgument( 'configuration' ) );

		$output->writeln( 'Setup store.' );
		$store->setupStore();
		$output->writeln( 'Setup store done.' );

		$output->writeln( 'Import data.' );
		$entitySaver = $store->getEntityDocumentSaver();

		$dumpReader = new JsonDumpReader(
			$input->getArgument( 'file' ),
			new SerializedEntityDeserializer( new EntityIdDeserializer( new BasicEntityIdParser() ) ),
			new ConsoleLogger( $output )
		);
		$count = 0;
		foreach( $dumpReader as $entity ) {
			$entitySaver->saveEntityDocument( $entity );
			$count++;

			if($count % 1000 === 0 ) {
				$output->write( '.' );
			}
		}
		$output->writeln( 'Importation done.' );

		$output->writeln( 'Setup indexes.' );
		$store->setupIndexes();
		$output->writeln( 'Setup indexes done.' );
	}
}