e09225eb09
* Created AbstractProvider class (for all DB services), which handles the default batch_size option. * The logger Closure is now optional for populate(). * Removed unused Elastica_Type argument from Provider constructors. * Added unit tests for Doctrine's AbstractProvider class. * The extra argument (ManagerRegistry) for Doctrine providers is now an appended constructor argument, so the extension no longer needs to use different replacement indexes for Propel/Doctrine providers.
53 lines
1.5 KiB
PHP
53 lines
1.5 KiB
PHP
<?php
|
|
|
|
namespace FOQ\ElasticaBundle\Doctrine\MongoDB;
|
|
|
|
use Doctrine\ODM\MongoDB\Query\Builder;
|
|
use FOQ\ElasticaBundle\Doctrine\AbstractProvider;
|
|
use FOQ\ElasticaBundle\Exception\InvalidArgumentTypeException;
|
|
|
|
class Provider extends AbstractProvider
|
|
{
|
|
/**
|
|
* @see FOQ\ElasticaBundle\Doctrine\AbstractProvider::countObjects()
|
|
*/
|
|
protected function countObjects($queryBuilder)
|
|
{
|
|
if (!$queryBuilder instanceof Builder) {
|
|
throw new InvalidArgumentTypeException($queryBuilder, 'Doctrine\ODM\MongoDB\Query\Builder');
|
|
}
|
|
|
|
return $queryBuilder
|
|
->getQuery()
|
|
->count();
|
|
}
|
|
|
|
/**
|
|
* @see FOQ\ElasticaBundle\Doctrine\AbstractProvider::fetchSlice()
|
|
*/
|
|
protected function fetchSlice($queryBuilder, $limit, $offset)
|
|
{
|
|
if (!$queryBuilder instanceof Builder) {
|
|
throw new InvalidArgumentTypeException($queryBuilder, 'Doctrine\ODM\MongoDB\Query\Builder');
|
|
}
|
|
|
|
return $queryBuilder
|
|
->limit($limit)
|
|
->skip($offset)
|
|
->getQuery()
|
|
->execute()
|
|
->toArray();
|
|
}
|
|
|
|
/**
|
|
* @see FOQ\ElasticaBundle\Doctrine\AbstractProvider::createQueryBuilder()
|
|
*/
|
|
protected function createQueryBuilder()
|
|
{
|
|
return $this->managerRegistry
|
|
->getManagerForClass($this->objectClass)
|
|
->getRepository($this->objectClass)
|
|
->{$this->options['query_builder_method']}();
|
|
}
|
|
}
|