src/Core/Framework/Increment/MySQLIncrementer.php line 35

  1. <?php declare(strict_types=1);
  3. namespace Shopware\Core\Framework\Increment;
  5. use Doctrine\DBAL\Connection;
  6. use Shopware\Core\Defaults;
  7. use Shopware\Core\Framework\DataAbstractionLayer\Doctrine\RetryableQuery;
  8. use Shopware\Core\Framework\Log\Package;
  9. use Shopware\Core\Framework\Plugin\Exception\DecorationPatternException;
  11. #[Package('core')]
  12. class MySQLIncrementer extends AbstractIncrementer
  13. {
  14.     /**
  15.      * @internal
  16.      */
  17.     public function __construct(private readonly Connection $connection)
  18.     {
  19.     }
  21.     public function getDecorated(): AbstractIncrementer
  22.     {
  23.         throw new DecorationPatternException(self::class);
  24.     }
  26.     public function increment(string $cluster, string $key): void
  27.     {
  28.         $payload = [
  29.             'pool' => $this->poolName,
  30.             'cluster' => $cluster,
  31.             'key' => $key,
  32.             'createdAt' => (new \DateTime())->format(Defaults::STORAGE_DATE_TIME_FORMAT),
  33.         ];
  35.         $this->connection->executeStatement('
  36.             INSERT INTO `increment` (`pool`, `cluster`, `key`, `count`, `created_at`)
  37.             VALUES (:pool, :cluster, :key, 1, :createdAt)
  38.             ON DUPLICATE KEY UPDATE `count` = `count` + 1, `updated_at` = :createdAt
  39.         ', $payload);
  40.     }
  42.     public function decrement(string $cluster, string $key): void
  43.     {
  44.         $payload = [
  45.             'pool' => $this->poolName,
  46.             'cluster' => $cluster,
  47.             'key' => $key,
  48.             'updatedAt' => (new \DateTime())->format(Defaults::STORAGE_DATE_TIME_FORMAT),
  49.         ];
  51.         $this->connection->executeStatement('
  52.             UPDATE `increment`
  53.             SET `count` = `count` - 1, `updated_at` = :updatedAt
  54.             WHERE `pool` = :pool AND `cluster` = :cluster AND `key` = :key AND `count` > 0;
  55.         ', $payload);
  56.     }
  58.     public function reset(string $cluster, ?string $key = null): void
  59.     {
  60.         $query = $this->connection->createQueryBuilder()
  61.             ->update('increment')
  62.             ->set('count', ':count')
  63.             ->set('updated_at', ':updatedAt')
  64.             ->where('pool = :pool')
  65.             ->andWhere('cluster = :cluster')
  66.             ->setParameter('updatedAt', (new \DateTime())->format(Defaults::STORAGE_DATE_TIME_FORMAT))
  67.             ->setParameter('cluster', $cluster)
  68.             ->setParameter('count', 0)
  69.             ->setParameter('pool', $this->poolName);
  71.         if ($key !== null) {
  72.             $query->andWhere('`key` = :key')
  73.                 ->setParameter('key', $key);
  74.         }
  76.         RetryableQuery::retryable($this->connection, function () use ($query): void {
  77.             $query->execute();
  78.         });
  79.     }
  81.     public function list(string $cluster, int $limit = 5, int $offset = 0): array
  82.     {
  83.         $sql = 'SELECT `key` as array_key, `pool`, `cluster`, `key`, `count`
  84.             FROM `increment`  WHERE `cluster` = :cluster AND `pool` = :pool
  85.             ORDER BY `count` DESC, `updated_at` DESC';
  87.         $payload = [
  88.             'pool' => $this->poolName,
  89.             'cluster' => $cluster,
  90.         ];
  92.         $types = [];
  94.         if ($limit > -1) {
  95.             $sql .= ' LIMIT :limit OFFSET :offset';
  96.             $payload['limit'] = $limit;
  97.             $payload['offset'] = $offset;
  98.             $types = [
  99.                 'offset' => \PDO::PARAM_INT,
  100.                 'limit' => \PDO::PARAM_INT,
  101.             ];
  102.         }
  104.         return $this->connection->fetchAllAssociativeIndexed($sql, $payload, $types);
  105.     }
  106. }