ONGR\ElasticsearchBundle\Command\IndexImportCommand

Complexity

Total Complexity

3

Size/Duplication

Total Lines	58
Duplicated Lines	0 %

Coupling/Cohesion

Components	0
Dependencies	6

Importance

Changes	3
Bugs	0	Features	1

2 Methods

Rating	Name	Duplication	Size	Complexity
A	execute()	0	22	2
B	configure()	0	26	1

<?php

/*
 * This file is part of the ONGR package.
 *
 * (c) NFQ Technologies UAB <info@nfq.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace ONGR\ElasticsearchBundle\Command;

use ONGR\ElasticsearchBundle\Service\ImportService;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Output\OutputInterface;

/**
 * IndexImportCommand class.
 */
class IndexImportCommand extends AbstractManagerAwareCommand
{
    /**
     * {@inheritdoc}
     */
    protected function configure()
    {
        parent::configure();

        $this
            ->setName('ongr:es:index:import')
            ->setDescription('Imports data to elasticsearch index.')
            ->addArgument(
                'filename',
                InputArgument::REQUIRED,
                'Select file to store output'
            )
            ->addOption(
                'bulk-size',
                'b',
                InputOption::VALUE_REQUIRED,
                'Set bulk size for import',
                1000
            )
            ->addOption(
                'gzip',
                'z',
                InputOption::VALUE_NONE,
                'Import a gzip file'
            );
    }

    /**
     * {@inheritdoc}
     */
    protected function execute(InputInterface $input, OutputInterface $output)
    {
        $manager = $this->getManager($input->getOption('manager'));

        // Initialize options array
        $options = [];
        if ($input->getOption('gzip')) {
            $options['gzip'] = null;
        }
        $options['bulk-size'] = $input->getOption('bulk-size');

        /** @var ImportService $importService */
        $importService = $this->getContainer()->get('es.import');
        $importService->importIndex(
            $manager,
            $input->getArgument('filename'),
            $output,
            $options
        );

        $output->writeln('<info>Data import completed!</info>');
    }
}