platform/src/Core/Framework/DataAbstractionLayer/Write/Validation/LockValidator.php line 38

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Shopware\Core\Framework\DataAbstractionLayer\Write\Validation;
  5. use Doctrine\DBAL\Connection;
  6. use Shopware\Core\Framework\DataAbstractionLayer\Dbal\EntityDefinitionQueryHelper;
  7. use Shopware\Core\Framework\DataAbstractionLayer\Write\Command\InsertCommand;
  8. use Shopware\Core\Framework\DataAbstractionLayer\Write\Command\WriteCommand;
  9. use Shopware\Core\Framework\Validation\WriteConstraintViolationException;
  10. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  11. use Symfony\Component\Validator\ConstraintViolation;
  12. use Symfony\Component\Validator\ConstraintViolationList;
  14. class LockValidator implements EventSubscriberInterface
  15. {
  16.     public const VIOLATION_LOCKED = 'FRAMEWORK__ENTITY_IS_LOCKED';
  18.     /**
  19.      * @var Connection
  20.      */
  21.     private $connection;
  23.     public function __construct(Connection $connection)
  24.     {
  25.         $this->connection = $connection;
  26.     }
  28.     public static function getSubscribedEvents(): array
  29.     {
  30.         return [
  31.             PreWriteValidationEvent::class => 'preValidate',
  32.         ];
  33.     }
  35.     /**
  36.      * @throws WriteConstraintViolationException
  37.      */
  38.     public function preValidate(PreWriteValidationEvent $event): void
  39.     {
  40.         $violations = new ConstraintViolationList();
  41.         $writeCommands = $event->getCommands();
  42.         $lockedEntities = $this->containsLockedEntities($writeCommands);
  44.         if (empty($lockedEntities)) {
  45.             return;
  46.         }
  48.         $message = 'The %s entity is locked and can neither be modified nor deleted.';
  50.         foreach ($lockedEntities as $entity => $_isLocked) {
  51.             $violations->add(new ConstraintViolation(
  52.                 sprintf($message, $entity),
  53.                 sprintf($message, '{{ entity }}'),
  54.                 ['{{ entity }}' => $entity],
  55.                 null,
  56.                 '/',
  57.                 null,
  58.                 null,
  59.                 self::VIOLATION_LOCKED
  60.             ));
  61.         }
  63.         $event->getExceptions()->add(new WriteConstraintViolationException($violations));
  64.     }
  66.     /**
  67.      * @param WriteCommand[] $writeCommands
  68.      */
  69.     private function containsLockedEntities(array $writeCommands): array
  70.     {
  71.         $ids = [];
  72.         $locked = [];
  74.         foreach ($writeCommands as $command) {
  75.             if ($command instanceof InsertCommand) {
  76.                 continue;
  77.             }
  79.             if (!$command->getDefinition()->isLockAware()) {
  80.                 continue;
  81.             }
  83.             $ids[$command->getDefinition()->getEntityName()][] = $command->getPrimaryKey()['id'];
  84.         }
  86.         foreach ($ids as $entityName => $primaryKeys) {
  87.             $locked[$entityName] = $this->connection->createQueryBuilder()
  88.                 ->select('1')
  89.                 ->from(EntityDefinitionQueryHelper::escape($entityName))
  90.                 ->where('`id` IN (:ids) AND `locked` = 1')
  91.                 ->setParameter('ids', $primaryKeys, Connection::PARAM_STR_ARRAY)
  92.                 ->execute()
  93.                 ->rowCount() > 0;
  94.         }
  96.         return array_filter($locked);
  97.     }
  98. }