public static function BatchUpdateForm::updateSourcesBatchWorker in Entity Usage 8.3

Batch operation worker for recreating statistics for source entities.

Parameters

string $entity_type_id: The entity type id, for example 'node'.

array $context: The context array.

File

src/Form/BatchUpdateForm.php, line 125

Class

BatchUpdateForm

Form to launch batch tracking of existing entities.

Namespace

Drupal\entity_usage\Form

Code

public static function updateSourcesBatchWorker($entity_type_id, array &$context) {
  $entity_storage = \Drupal::entityTypeManager()
    ->getStorage($entity_type_id);
  $entity_type = \Drupal::entityTypeManager()
    ->getDefinition($entity_type_id);
  $entity_type_key = $entity_type
    ->getKey('id');
  if (empty($context['sandbox']['total'])) {

    // Delete current usage statistics for these entities.
    \Drupal::service('entity_usage.usage')
      ->bulkDeleteSources($entity_type_id);
    $context['sandbox']['progress'] = 0;
    $context['sandbox']['current_id'] = -1;
    $context['sandbox']['total'] = (int) $entity_storage
      ->getQuery()
      ->accessCheck(FALSE)
      ->count()
      ->execute();
  }
  $entity_ids = $entity_storage
    ->getQuery()
    ->condition($entity_type_key, $context['sandbox']['current_id'], '>')
    ->range(0, 1)
    ->accessCheck(FALSE)
    ->sort($entity_type_key)
    ->execute();

  /** @var \Drupal\Core\Entity\EntityInterface $entity */
  $entity = $entity_storage
    ->load(reset($entity_ids));
  if ($entity) {
    if (EntityUsageSourceLevel::isTopLevel($entity)) {
      if ($entity
        ->getEntityType()
        ->isRevisionable()) {

        // Track all revisions and translations of the source entity. Sources
        // are tracked as if they were new entities.
        $result = $entity_storage
          ->getQuery()
          ->allRevisions()
          ->condition($entity
          ->getEntityType()
          ->getKey('id'), $entity
          ->id())
          ->sort($entity
          ->getEntityType()
          ->getKey('revision'), 'DESC')
          ->execute();
        $revision_ids = array_keys($result);
        foreach ($revision_ids as $revision_id) {

          /** @var \Drupal\Core\Entity\EntityInterface $entity_revision */
          if (!($entity_revision = $entity_storage
            ->loadRevision($revision_id))) {
            continue;
          }
          \Drupal::service('entity_usage.entity_update_manager')
            ->recalculateUsageInformation($entity_revision);
        }
      }
      else {
        \Drupal::service('entity_usage.entity_update_manager')
          ->recalculateUsageInformation($entity);
      }
    }
    else {
      $context['sandbox']['progress']++;
      $context['sandbox']['current_id'] = $entity
        ->id();
      $context['results'][] = $entity_type_id . ':' . $entity
        ->id();
    }
    $context['sandbox']['progress']++;
    $context['sandbox']['current_id'] = $entity
      ->id();
    $context['results'][] = $entity_type_id . ':' . $entity
      ->id();
  }
  if ($context['sandbox']['progress'] < $context['sandbox']['total']) {
    $context['finished'] = $context['sandbox']['progress'] / $context['sandbox']['total'];
  }
  else {
    $context['finished'] = 1;
  }
  $context['message'] = t('Updating entity usage for @entity_type: @current of @total', [
    '@entity_type' => $entity_type_id,
    '@current' => $context['sandbox']['progress'],
    '@total' => $context['sandbox']['total'],
  ]);
}