91 lines
3.1 KiB
PHP
91 lines
3.1 KiB
PHP
<?php
|
|
|
|
namespace FOQ\ElasticaBundle\Manager;
|
|
|
|
use Doctrine\Common\Annotations\Reader;
|
|
use Doctrine\Common\Persistence\ManagerRegistry;
|
|
use FOQ\ElasticaBundle\Finder\FinderInterface;
|
|
use FOQ\ElasticaBundle\Repository;
|
|
use RuntimeException;
|
|
/**
|
|
* @author Richard Miller <info@limethinking.co.uk>
|
|
*
|
|
* Allows retrieval of basic or custom repository for mapped Doctrine
|
|
* entities/documents.
|
|
*/
|
|
class RepositoryManager implements RepositoryManagerInterface
|
|
{
|
|
protected $entities = array();
|
|
protected $repositories = array();
|
|
protected $managerRegistry;
|
|
protected $reader;
|
|
|
|
public function __construct(ManagerRegistry $managerRegistry, Reader $reader)
|
|
{
|
|
$this->managerRegistry = $managerRegistry;
|
|
$this->reader = $reader;
|
|
}
|
|
|
|
public function addEntity($entityName, FinderInterface $finder, $repositoryName = null)
|
|
{
|
|
$this->entities[$entityName]= array();
|
|
$this->entities[$entityName]['finder'] = $finder;
|
|
$this->entities[$entityName]['repositoryName'] = $repositoryName;
|
|
}
|
|
|
|
/**
|
|
* Return repository for entity
|
|
*
|
|
* Returns custom repository if one specified otherwise
|
|
* returns a basic respository.
|
|
*/
|
|
public function getRepository($entityName)
|
|
{
|
|
$realEntityName = $entityName;
|
|
if (strpos($entityName, ':') !== false) {
|
|
list($namespaceAlias, $simpleClassName) = explode(':', $entityName);
|
|
$realEntityName = $this->managerRegistry->getAliasNamespace($namespaceAlias) . '\\' . $simpleClassName;
|
|
}
|
|
|
|
if (isset($this->repositories[$realEntityName])) {
|
|
return $this->repositories[$realEntityName];
|
|
}
|
|
|
|
if (!isset($this->entities[$realEntityName])) {
|
|
throw new RuntimeException(sprintf('No search finder configured for %s', $realEntityName));
|
|
}
|
|
|
|
$this->setRepositoryFromAnnotation($realEntityName);
|
|
|
|
if (isset($this->entities[$realEntityName]['repositoryName'])) {
|
|
|
|
$repositoryName = $this->entities[$realEntityName]['repositoryName'];
|
|
if (!class_exists($repositoryName)) {
|
|
throw new RuntimeException(sprintf('%s repository for %s does not exist', $repositoryName, $realEntityName));
|
|
}
|
|
$repository = new $repositoryName($this->entities[$realEntityName]['finder']);
|
|
$this->repositories[$realEntityName] = $repository;
|
|
return $repository;
|
|
}
|
|
|
|
$repository = new Repository($this->entities[$realEntityName]['finder']);
|
|
$this->repositories[$realEntityName] = $repository;
|
|
return $repository;
|
|
}
|
|
|
|
private function setRepositoryFromAnnotation($realEntityName)
|
|
{
|
|
if (isset($this->entities[$realEntityName]['repositoryName'])) {
|
|
return;
|
|
}
|
|
|
|
$refClass = new \ReflectionClass($realEntityName);
|
|
$annotation = $this->reader->getClassAnnotation($refClass, 'FOQ\\ElasticaBundle\\Configuration\\Search');
|
|
if ($annotation) {
|
|
$this->entities[$realEntityName]['repositoryName']
|
|
= $annotation->repositoryClass;
|
|
}
|
|
}
|
|
|
|
}
|