protected function _getEntities()
{
$options = $this->getOptions();
$query = new EntityFieldQuery();
$query->entityCondition('entity_type', $options['entity-type']);
if (!empty($options['sort-field'])) {
if (strpos($options['sort-field'], ':') === FALSE) {
$query->propertyOrderBy($options['sort-field'], strtoupper($options['sort-order']));
} else {
$sort_field = explode(':', $options['sort-field']);
$query->fieldOrderBy($sort_field[0], $sort_field[1], strtoupper($options['sort-order']));
}
}
if (!empty($options['bundle'])) {
$query->propertyCondition('type', $options['bundle']);
}
$query->range(0, $options['limit']);
$results = $query->execute();
if (empty($results[$options['entity-type']])) {
return array();
}
return entity_load($options['entity-type'], array_keys($results[$options['entity-type']]));
}
/**
* Set correct page (i.e. range) for the query for list.
*
* Determine the page that should be seen. Page 1, is actually offset 0 in the
* query range.
*
* @param \EntityFieldQuery $query
* The query object.
*
* @throws BadRequestException
*
* @see \RestfulEntityBase::getQueryForList
*/
protected function queryForListPagination(\EntityFieldQuery $query)
{
list($offset, $range) = $this->parseRequestForListPagination();
$query->range($offset, $range);
}
/**
* {@inheritdoc}
*/
public function rebuildBatchFetch($entity, &$context)
{
if (!isset($context['sandbox']['info'])) {
$context['sandbox']['info'] = xmlsitemap_get_link_info($entity);
$context['sandbox']['progress'] = 0;
$context['sandbox']['last_id'] = 0;
}
$info = $context['sandbox']['info'];
$query = new EntityFieldQuery();
$query->entityCondition('entity_type', $entity);
$query->entityCondition('entity_id', $context['sandbox']['last_id'], '>');
$query->addTag('xmlsitemap_link_bundle_access');
$query->addTag('xmlsitemap_rebuild');
$query->addMetaData('entity', $entity);
$query->addMetaData('entity_info', $info);
if (!isset($context['sandbox']['max'])) {
$count_query = clone $query;
$count_query->count();
$context['sandbox']['max'] = $count_query->execute();
if (!$context['sandbox']['max']) {
// If there are no items to process, skip everything else.
return;
}
}
// PostgreSQL cannot have the ORDERED BY in the count query.
$query->entityOrderBy('entity_id');
// get batch limit
$limit = $this->config > get('batch_limit');
$query->range(0, $limit);
$result = $query->execute();
$ids = array_keys($result[$entity]);
$info['xmlsitemap']['process callback']($ids);
$context['sandbox']['last_id'] = end($ids);
$context['sandbox']['progress'] += count($ids);
$context['message'] = t('Now processing %entity @last_id (@progress of @count).', array('%entity' => $entity, '@last_id' => $context['sandbox']['last_id'], '@progress' => $context['sandbox']['progress'], '@count' => $context['sandbox']['max']));
if ($context['sandbox']['progress'] >= $context['sandbox']['max']) {
$context['finished'] = 1;
} else {
$context['finished'] = $context['sandbox']['progress'] / $context['sandbox']['max'];
}
}
