custom/plugins/PickwareErpStarter/src/Stock/WarehouseStockInitializer.php line 51

Open in your IDE?
  1. <?php
  2. /*
  3.  * Copyright (c) Pickware GmbH. All rights reserved.
  4.  * This file is part of software that is released under a proprietary license.
  5.  * You must not copy, modify, distribute, make publicly available, or execute
  6.  * its contents or parts thereof without express permission by the copyright
  7.  * holder, unless otherwise permitted by law.
  8.  */
  10. declare(strict_types=1);
  12. namespace Pickware\PickwareErpStarter\Stock;
  14. use Doctrine\DBAL\Connection;
  15. use Pickware\DalBundle\RetryableTransaction;
  16. use Pickware\DalBundle\Sql\SqlUuid;
  17. use Pickware\PickwareErpStarter\Warehouse\Model\WarehouseDefinition;
  18. use Shopware\Core\Content\Product\ProductEvents;
  19. use Shopware\Core\Defaults;
  20. use Shopware\Core\Framework\DataAbstractionLayer\EntityWriteResult;
  21. use Shopware\Core\Framework\DataAbstractionLayer\Event\EntityWrittenEvent;
  22. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  24. class WarehouseStockInitializer implements EventSubscriberInterface
  25. {
  26.     /**
  27.      * @var Connection
  28.      */
  29.     private $db;
  31.     public function __construct(Connection $db)
  32.     {
  33.         $this->db = $db;
  34.     }
  36.     public static function getSubscribedEvents(): array
  37.     {
  38.         return [
  39.             ProductEvents::PRODUCT_WRITTEN_EVENT => 'productWritten',
  40.             WarehouseDefinition::ENTITY_WRITTEN_EVENT => 'warehouseWritten',
  41.         ];
  42.     }
  44.     public function productWritten(EntityWrittenEvent $entityWrittenEvent): void
  45.     {
  46.         $this->ensureProductWarehouseStockForProductsExist(
  47.             $this->getNewlyCreatedEntityIds($entityWrittenEvent),
  48.         );
  49.     }
  51.     public function warehouseWritten(EntityWrittenEvent $entityWrittenEvent): void
  52.     {
  53.         $this->ensureProductWarehouseStocksExist(
  54.             'warehouse.id',
  55.             $this->getNewlyCreatedEntityIds($entityWrittenEvent),
  56.         );
  57.     }
  59.     public function ensureProductWarehouseStockForProductsExist(array $productIds): void
  60.     {
  61.         $this->ensureProductWarehouseStocksExist('product.id', $productIds);
  62.     }
  64.     private function ensureProductWarehouseStocksExist(string $idFieldName, array $ids): void
  65.     {
  66.         if (count($ids) === 0) {
  67.             return;
  68.         }
  70.         // This query is a potential deadlock candidate and this is why it is wrapped in a retryable transaction
  71.         RetryableTransaction::retryable($this->db, fn() => $this->db->executeStatement(
  72.             'INSERT INTO pickware_erp_warehouse_stock (
  73.                 id,
  74.                 product_id,
  75.                 product_version_id,
  76.                 quantity,
  77.                 warehouse_id,
  78.                 created_at
  79.             ) SELECT
  80.                 ' . SqlUuid::UUID_V4_GENERATION . ',
  81.                 product.id,
  82.                 product.version_id,
  83.                 0,
  84.                 warehouse.id,
  85.                 NOW(3)
  86.             FROM product
  87.             CROSS JOIN pickware_erp_warehouse warehouse
  88.             WHERE ' . $idFieldName . ' IN (:ids) AND product.version_id = :liveVersionId
  89.             ON DUPLICATE KEY UPDATE pickware_erp_warehouse_stock.id = pickware_erp_warehouse_stock.id',
  90.             [
  91.                 'ids' => array_map('hex2bin', $ids),
  92.                 'liveVersionId' => hex2bin(Defaults::LIVE_VERSION),
  93.             ],
  94.             [
  95.                 'ids' => Connection::PARAM_STR_ARRAY,
  96.             ],
  97.         ));
  98.     }
  100.     private function getNewlyCreatedEntityIds(EntityWrittenEvent $entityWrittenEvent): array
  101.     {
  102.         if ($entityWrittenEvent->getContext()->getVersionId() !== Defaults::LIVE_VERSION) {
  103.             return [];
  104.         }
  106.         $ids = [];
  107.         foreach ($entityWrittenEvent->getWriteResults() as $writeResult) {
  108.             // $writeResult->getExistence() can be null, but we have no idea why and also not what this means.
  109.             $existence = $writeResult->getExistence();
  110.             if (($existence === null && $writeResult->getOperation() === EntityWriteResult::OPERATION_INSERT)
  111.                 || ($existence !== null && !$existence->exists())
  112.             ) {
  113.                 $ids[] = $writeResult->getPrimaryKey();
  114.             }
  115.         }
  117.         return $ids;
  118.     }
  119. }