/**
* @param string $sort
* @param mixed $order Description
* @return \Doctrine\ORM\Query\Expr\OrderBy
*/
public function add($sort, $order = 0)
{
if (is_int($order)) {
$order = $order == 0 ? self::$asc : self::$desc;
} else {
$order = strtoupper($order) == self::$asc ? self::$asc : self::$desc;
}
parent::add($sort, $order);
return $this;
}
/**
* @param array $orderBy
* @param QueryBuilder $qb
* @return OrderBy
*/
protected function normalizeOrderBy(array $orderBy, QueryBuilder $qb)
{
$expr = new OrderBy();
foreach ($orderBy as $field => $direction) {
if (false === strpos($field, '.')) {
$rootAlias = $qb->getRootAlias();
$field = "{$rootAlias}.{$field}";
}
$expr->add($field, $direction);
}
return $expr;
}
/**
* @param string $entityName
* @param integer|null $offset
* @param integer|null $limit
* @return int
*/
protected function reindexSingleEntity($entityName, $offset = null, $limit = null)
{
/** @var EntityManager $entityManager */
$entityManager = $this->registry->getManagerForClass($entityName);
$entityManager->getConnection()->getConfiguration()->setSQLLogger(null);
$pk = $entityManager->getClassMetadata($entityName)->getIdentifier();
$orderingsExpr = new OrderBy();
foreach ($pk as $fieldName) {
$orderingsExpr->add('entity.' . $fieldName);
}
$queryBuilder = $entityManager->getRepository($entityName)->createQueryBuilder('entity')->orderBy($orderingsExpr);
if (null !== $offset) {
$queryBuilder->setFirstResult($offset);
}
if (null !== $limit) {
$queryBuilder->setMaxResults($limit);
}
$iterator = new BufferedQueryResultIterator($queryBuilder);
$iterator->setBufferSize(static::BATCH_SIZE);
$itemsCount = 0;
$entities = [];
foreach ($iterator as $entity) {
$entities[] = $entity;
$itemsCount++;
if (0 == $itemsCount % static::BATCH_SIZE) {
$this->save($entities, true);
$entityManager->clear();
$entities = [];
gc_collect_cycles();
}
}
if ($itemsCount % static::BATCH_SIZE > 0) {
$this->save($entities, true);
$entityManager->clear();
}
return $itemsCount;
}
