Nimble\ElasticBundle\Populator\Populator

Complexity

Total Complexity

9

Size/Duplication

Total Lines	99
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	5

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	8	1
A	transformEntitiesToDocuments()	0	14	2
B	populate()	0	36	6

<?php

namespace Nimble\ElasticBundle\Populator;

use Nimble\ElasticBundle\Document;
use Nimble\ElasticBundle\Transformer\TransformerManager;
use Nimble\ElasticBundle\Type\Type;
use Symfony\Component\Console\Helper\ProgressBar;

class Populator
{
    /**
     * @var Type
     */
    private $type;

    /**
     * @var PopulationFetcherInterface
     */
    private $fetcher;

    /**
     * @var TransformerManager
     */
    private $transformer;

    /**
     * @var int
     */
    private $entityCount;

    /**
     * @param Type $type
     * @param PopulationFetcherInterface $fetcher
     * @param TransformerManager $transformer
     */
    public function __construct(Type $type, PopulationFetcherInterface $fetcher, TransformerManager $transformer)
    {
        $this->type = $type;
        $this->fetcher = $fetcher;
        $this->transformer = $transformer;

        $this->entityCount = $fetcher->getEntityCount();
    }

    /**
     * @param array $entities
     * @return Document[]
     */
    protected function transformEntitiesToDocuments(array $entities)
    {
        $documents = [];

        foreach ($entities as $entity) {
            $documents = array_merge($documents, $this->transformer->transformToDocuments(
                $entity,
                $this->type->getIndex()->getId(),
                $this->type->getName()
            ));
        }

        return $documents;
    }

    /**
     * TODO: make a progress bar interface an an adapter for symfony's ProgressBar
     *
     * @param int $batchSize
     * @param ProgressBar $progress
     * @return int Number of documents created/updated.
     */
    public function populate($batchSize, ProgressBar $progress = null)
    {
        $count = $this->fetcher->getEntityCount();

        if ($count === 0) {
            return 0;
        }

        $documentCount = 0;

        $offset = 0;

        if (null !== $progress) {
            $progress->start($count);
        }

        while ($offset < $count) {
            $entities = $this->fetcher->fetchEntities($offset, $batchSize);
            $batch = $this->transformEntitiesToDocuments($entities);

            $this->type->putDocuments($batch);

            if (null !== $progress) {
                $progress->advance(count($entities));
            }

            $documentCount += count($batch);
            $offset += $batchSize;
        }

        if (null !== $progress) {
            $progress->finish();
        }

        return $documentCount;
    }
}