src/Core/Framework/DataAbstractionLayer/Doctrine/RetryableQuery.php line 32

  1. <?php declare(strict_types=1);
  3. namespace Shopware\Core\Framework\DataAbstractionLayer\Doctrine;
  5. use Doctrine\DBAL\Connection;
  6. use Doctrine\DBAL\Exception\RetryableException;
  7. use Doctrine\DBAL\Statement;
  8. use Shopware\Core\Framework\Log\Package;
  10. #[Package('core')]
  11. class RetryableQuery
  12. {
  13.     public function __construct(private readonly ?Connection $connection, private readonly Statement $query)
  14.     {
  15.     }
  17.     public function execute(array $params = []): int
  18.     {
  19.         return self::retry($this->connection, fn () => $this->query->executeStatement($params), 0);
  20.     }
  22.     public static function retryable(Connection $connection, \Closure $closure)
  23.     {
  24.         return self::retry($connection, $closure, 0);
  25.     }
  27.     public function getQuery(): Statement
  28.     {
  29.         return $this->query;
  30.     }
  32.     private static function retry(?Connection $connection, \Closure $closure, int $counter)
  33.     {
  34.         ++$counter;
  36.         try {
  37.             return $closure();
  38.         } catch (RetryableException $e) {
  39.             if ($connection && $connection->getTransactionNestingLevel() > 0) {
  40.                 // If this closure was executed inside a transaction, do not retry. Remember that the whole (outermost)
  41.                 // transaction was already rolled back by the database when any RetryableException is thrown. Rethrow
  42.                 // the exception here so only the outermost transaction is retried which in turn includes this closure.
  43.                 throw $e;
  44.             }
  46.             if ($counter > 10) {
  47.                 throw $e;
  48.             }
  50.             // randomize sleep to prevent same execution delay for multiple statements
  51.             usleep(20 * $counter);
  53.             return self::retry($connection, $closure, $counter);
  54.         }
  55.     }
  56. }