/**
  * {@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'];
     }
 }