View source
<?php
namespace Drupal\entity_usage\Form;
use Drupal\Core\Entity\EntityTypeManagerInterface;
use Drupal\Core\Form\FormBase;
use Drupal\Core\Form\FormStateInterface;
use Drupal\entity_usage\EntityUsageSourceLevel;
use Symfony\Component\DependencyInjection\ContainerInterface;
class BatchUpdateForm extends FormBase {
protected $entityTypeManager;
public function __construct(EntityTypeManagerInterface $entity_manager) {
$this->entityTypeManager = $entity_manager;
}
public static function create(ContainerInterface $container) {
return new static($container
->get('entity_type.manager'));
}
public function getFormId() {
return 'entity_update_batch_update_form';
}
public function buildForm(array $form, FormStateInterface $form_state) {
$form['description'] = [
'#markup' => $this
->t("This page allows you to delete and re-generate again all entity usage statistics in your system.<br /><br />You may want to check the settings page to fine-tune what entities should be tracked, and other options."),
];
$form['actions'] = [
'#type' => 'actions',
];
$form['actions']['submit'] = [
'#type' => 'submit',
'#button_type' => 'primary',
'#value' => $this
->t('Recreate all entity usage statistics'),
];
return $form;
}
public function submitForm(array &$form, FormStateInterface $form_state) {
$batch = $this
->generateBatch();
batch_set($batch);
}
public function generateBatch() {
$operations = [];
$to_track = \Drupal::config('entity_usage.settings')
->get('track_enabled_source_entity_types');
foreach ($this->entityTypeManager
->getDefinitions() as $entity_type_id => $entity_type) {
$track_this_entity_type = FALSE;
if (!is_array($to_track) && $entity_type
->entityClassImplements('\\Drupal\\Core\\Entity\\ContentEntityInterface')) {
if (!in_array($entity_type_id, [
'file',
'user',
])) {
$track_this_entity_type = TRUE;
}
}
elseif (is_array($to_track) && in_array($entity_type_id, $to_track, TRUE)) {
$track_this_entity_type = TRUE;
}
if ($track_this_entity_type) {
$operations[] = [
'Drupal\\entity_usage\\Form\\BatchUpdateForm::updateSourcesBatchWorker',
[
$entity_type_id,
],
];
}
}
$batch = [
'operations' => $operations,
'finished' => 'Drupal\\entity_usage\\Form\\BatchUpdateForm::batchFinished',
'title' => $this
->t('Updating entity usage statistics.'),
'progress_message' => $this
->t('Processed @current of @total entity types.'),
'error_message' => $this
->t('This batch encountered an error.'),
];
return $batch;
}
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'])) {
\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();
$entity = $entity_storage
->load(reset($entity_ids));
if ($entity) {
if (EntityUsageSourceLevel::isTopLevel($entity)) {
if ($entity
->getEntityType()
->isRevisionable()) {
$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) {
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'],
]);
}
public static function batchFinished($success, array $results, array $operations) {
if ($success) {
drupal_set_message(t('Recreated entity usage for @count entities.', [
'@count' => count($results),
]));
\Drupal::state()
->set('entity_usage_needs_regeneration', FALSE);
}
else {
$error_operation = reset($operations);
drupal_set_message(t('An error occurred while processing @operation with arguments : @args', [
'@operation' => $error_operation[0],
'@args' => print_r($error_operation[0], TRUE),
]));
}
}
}