vendor/shopware/core/Checkout/Promotion/DataAbstractionLayer/PromotionRedemptionUpdater.php line 93

  1. <?php declare(strict_types=1);
  3. namespace Shopware\Core\Checkout\Promotion\DataAbstractionLayer;
  5. use Doctrine\DBAL\ArrayParameterType;
  6. use Doctrine\DBAL\Connection;
  7. use Shopware\Core\Checkout\Cart\Event\CheckoutOrderPlacedEvent;
  8. use Shopware\Core\Checkout\Order\Aggregate\OrderLineItem\OrderLineItemEntity;
  9. use Shopware\Core\Checkout\Promotion\Cart\PromotionProcessor;
  10. use Shopware\Core\Defaults;
  11. use Shopware\Core\Framework\Context;
  12. use Shopware\Core\Framework\DataAbstractionLayer\Doctrine\RetryableQuery;
  13. use Shopware\Core\Framework\Log\Package;
  14. use Shopware\Core\Framework\Uuid\Uuid;
  15. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  17. /**
  18.  * @internal
  19.  */
  20. #[Package('checkout')]
  21. class PromotionRedemptionUpdater implements EventSubscriberInterface
  22. {
  23.     /**
  24.      * @internal
  25.      */
  26.     public function __construct(private readonly Connection $connection)
  27.     {
  28.     }
  30.     /**
  31.      * @return array<string, string|array{0: string, 1: int}|list<array{0: string, 1?: int}>>
  32.      */
  33.     public static function getSubscribedEvents(): array
  34.     {
  35.         return [
  36.             CheckoutOrderPlacedEvent::class => 'orderPlaced',
  37.         ];
  38.     }
  40.     /**
  41.      * @param array<string> $ids
  42.      */
  43.     public function update(array $ids, Context $context): void
  44.     {
  45.         $ids = array_unique(array_filter($ids));
  47.         if (empty($ids) || $context->getVersionId() !== Defaults::LIVE_VERSION) {
  48.             return;
  49.         }
  51.         $sql = <<<'SQL'
  52.                 SELECT LOWER(HEX(order_line_item.promotion_id)) as promotion_id,
  53.                        COUNT(DISTINCT order_line_item.id) as total,
  54.                        LOWER(HEX(order_customer.customer_id)) as customer_id
  55.                 FROM order_line_item
  56.                 INNER JOIN order_customer
  57.                     ON order_customer.order_id = order_line_item.order_id
  58.                     AND order_customer.version_id = order_line_item.version_id
  59.                 WHERE order_line_item.type = :type
  60.                 AND order_line_item.promotion_id IN (:ids)
  61.                 AND order_line_item.version_id = :versionId
  62.                 GROUP BY order_line_item.promotion_id, order_customer.customer_id
  63. SQL;
  65.         $promotions = $this->connection->fetchAllAssociative(
  66.             $sql,
  67.             ['type' => PromotionProcessor::LINE_ITEM_TYPE, 'ids' => Uuid::fromHexToBytesList($ids), 'versionId' => Uuid::fromHexToBytes(Defaults::LIVE_VERSION)],
  68.             ['ids' => ArrayParameterType::STRING]
  69.         );
  71.         if (empty($promotions)) {
  72.             return;
  73.         }
  74.         $update = new RetryableQuery(
  75.             $this->connection,
  76.             $this->connection->prepare('UPDATE promotion SET order_count = :count, orders_per_customer_count = :customerCount WHERE id = :id')
  77.         );
  79.         // group the promotions to update each promotion with a single update statement
  80.         $promotions = $this->groupByPromotion($promotions);
  82.         foreach ($promotions as $id => $totals) {
  83.             $total = array_sum($totals);
  85.             $update->execute([
  86.                 'id' => Uuid::fromHexToBytes($id),
  87.                 'count' => (int) $total,
  88.                 'customerCount' => json_encode($totals, \JSON_THROW_ON_ERROR),
  89.             ]);
  90.         }
  91.     }
  93.     public function orderPlaced(CheckoutOrderPlacedEvent $event): void
  94.     {
  95.         $lineItems = $event->getOrder()->getLineItems();
  96.         $customer = $event->getOrder()->getOrderCustomer();
  98.         if (!$lineItems || !$customer) {
  99.             return;
  100.         }
  102.         $promotionIds = [];
  103.         /** @var OrderLineItemEntity $lineItem */
  104.         foreach ($lineItems as $lineItem) {
  105.             if ($lineItem->getType() !== PromotionProcessor::LINE_ITEM_TYPE) {
  106.                 continue;
  107.             }
  109.             $promotionId = $lineItem->getPromotionId();
  110.             if ($promotionId === null) {
  111.                 continue;
  112.             }
  114.             $promotionIds[] = $promotionId;
  115.         }
  117.         if (!$promotionIds) {
  118.             return;
  119.         }
  121.         $allCustomerCounts = $this->getAllCustomerCounts($promotionIds);
  123.         $update = new RetryableQuery(
  124.             $this->connection,
  125.             $this->connection->prepare('UPDATE promotion SET order_count = order_count + 1, orders_per_customer_count = :customerCount WHERE id = :id')
  126.         );
  128.         foreach ($promotionIds as $promotionId) {
  129.             $customerId = $customer->getCustomerId();
  130.             if ($customerId !== null) {
  131.                 $allCustomerCounts[$promotionId][$customerId] = 1 + ($allCustomerCounts[$promotionId][$customerId] ?? 0);
  132.             }
  134.             $update->execute([
  135.                 'id' => Uuid::fromHexToBytes($promotionId),
  136.                 'customerCount' => json_encode($allCustomerCounts[$promotionId], \JSON_THROW_ON_ERROR),
  137.             ]);
  138.         }
  139.     }
  141.     /**
  142.      * @param array<mixed> $promotions
  143.      *
  144.      * @return array<mixed>
  145.      */
  146.     private function groupByPromotion(array $promotions): array
  147.     {
  148.         $grouped = [];
  149.         foreach ($promotions as $promotion) {
  150.             $id = $promotion['promotion_id'];
  151.             $customerId = $promotion['customer_id'];
  152.             $grouped[$id][$customerId] = (int) $promotion['total'];
  153.         }
  155.         return $grouped;
  156.     }
  158.     /**
  159.      * @param array<string> $promotionIds
  160.      *
  161.      * @return array<string>
  162.      */
  163.     private function getAllCustomerCounts(array $promotionIds): array
  164.     {
  165.         $allCustomerCounts = [];
  166.         $countResult = $this->connection->fetchAllAssociative(
  167.             'SELECT `id`, `orders_per_customer_count` FROM `promotion` WHERE `id` IN (:ids)',
  168.             ['ids' => Uuid::fromHexToBytesList($promotionIds)],
  169.             ['ids' => ArrayParameterType::STRING]
  170.         );
  172.         foreach ($countResult as $row) {
  173.             if (!\is_string($row['orders_per_customer_count'])) {
  174.                 $allCustomerCounts[Uuid::fromBytesToHex($row['id'])] = [];
  176.                 continue;
  177.             }
  179.             $customerCount = json_decode($row['orders_per_customer_count'], true, 512, \JSON_THROW_ON_ERROR);
  180.             if (!$customerCount) {
  181.                 $allCustomerCounts[Uuid::fromBytesToHex($row['id'])] = [];
  183.                 continue;
  184.             }
  186.             $allCustomerCounts[Uuid::fromBytesToHex($row['id'])] = $customerCount;
  187.         }
  189.         return $allCustomerCounts;
  190.     }
  191. }