Phraseanet/lib/Alchemy/Phrasea/SearchEngine/Elastic/DataboxFetcherFactory.php

<?php

namespace Alchemy\Phrasea\SearchEngine\Elastic;

use Alchemy\Phrasea\SearchEngine\Elastic\Indexer\Record\Delegate\FetcherDelegateInterface;
use Alchemy\Phrasea\SearchEngine\Elastic\Indexer\Record\Fetcher;
use Alchemy\Phrasea\SearchEngine\Elastic\Indexer\Record\Hydrator\CoreHydrator;
use Alchemy\Phrasea\SearchEngine\Elastic\Indexer\Record\Hydrator\FlagHydrator;
use Alchemy\Phrasea\SearchEngine\Elastic\Indexer\Record\Hydrator\MetadataHydrator;
use Alchemy\Phrasea\SearchEngine\Elastic\Indexer\Record\Hydrator\SubDefinitionHydrator;
use Alchemy\Phrasea\SearchEngine\Elastic\Indexer\Record\Hydrator\ThesaurusHydrator;
use Alchemy\Phrasea\SearchEngine\Elastic\Indexer\Record\Hydrator\TitleHydrator;
use Alchemy\Phrasea\SearchEngine\Elastic\Structure\Structure;
use Alchemy\Phrasea\SearchEngine\Elastic\Thesaurus\CandidateTerms;

class DataboxFetcherFactory
{
    /**
     * @var \ArrayAccess
     */
    private $container;

    /**
     * @var string
     */
    private $structureKey;

    /**
     * @var string
     */
    private $thesaurusKey;

    /**
     * @var RecordHelper
     */
    private $recordHelper;

    /**
     * @param RecordHelper $recordHelper
     * @param \ArrayAccess $container
     * @param string $structureKey
     * @param string $thesaurusKey
     */
    public function __construct(RecordHelper $recordHelper, \ArrayAccess $container, $structureKey, $thesaurusKey)
    {
        $this->recordHelper = $recordHelper;
        $this->container = $container;
        $this->structureKey = $structureKey;
        $this->thesaurusKey = $thesaurusKey;
    }

    /**
     * @param \databox $databox
     * @param FetcherDelegateInterface $fetcherDelegate
     * @return Fetcher
     */
    public function createFetcher(\databox $databox, FetcherDelegateInterface $fetcherDelegate = null)
    {
        $connection = $databox->get_connection();

        $candidateTerms = new CandidateTerms($databox);
        $fetcher = new Fetcher($databox, array(
            new CoreHydrator($databox->get_sbas_id(), $databox->get_viewname(), $this->recordHelper),
            new TitleHydrator($connection),
            new MetadataHydrator($connection, $this->getStructure(), $this->recordHelper),
            new FlagHydrator($this->getStructure(), $databox),
            new ThesaurusHydrator($this->getStructure(), $this->getThesaurus(), $candidateTerms),
            new SubDefinitionHydrator($connection)
        ), $fetcherDelegate);

        $fetcher->setBatchSize(200);
        $fetcher->onDrain(function() use ($candidateTerms) {
            $candidateTerms->save();
        });

        return $fetcher;
    }

    /**
     * @return Structure
     */
    private function getStructure()
    {
        return $this->container[$this->structureKey];
    }

    /**
     * @return Thesaurus
     */
    private function getThesaurus()
    {
        return $this->container[$this->thesaurusKey];
    }
}