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

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.     private Connection $db;
  28.     public function __construct(Connection $db)
  29.     {
  30.         $this->db = $db;
  31.     }
  33.     public static function getSubscribedEvents(): array
  34.     {
  35.         return [
  36.             ProductEvents::PRODUCT_WRITTEN_EVENT => 'productWritten',
  37.             WarehouseDefinition::ENTITY_WRITTEN_EVENT => 'warehouseWritten',
  38.         ];
  39.     }
  41.     public function productWritten(EntityWrittenEvent $entityWrittenEvent): void
  42.     {
  43.         $this->ensureProductWarehouseStockForProductsExist(
  44.             $this->getNewlyCreatedEntityIds($entityWrittenEvent),
  45.         );
  46.     }
  48.     public function warehouseWritten(EntityWrittenEvent $entityWrittenEvent): void
  49.     {
  50.         $this->ensureProductWarehouseStocksExist(
  51.             'warehouse.id',
  52.             $this->getNewlyCreatedEntityIds($entityWrittenEvent),
  53.         );
  54.     }
  56.     public function ensureProductWarehouseStockForProductsExist(array $productIds): void
  57.     {
  58.         $this->ensureProductWarehouseStocksExist('product.id', $productIds);
  59.     }
  61.     private function ensureProductWarehouseStocksExist(string $idFieldName, array $ids): void
  62.     {
  63.         if (count($ids) === 0) {
  64.             return;
  65.         }
  67.         // This query is a potential deadlock candidate and this is why it is wrapped in a retryable transaction
  68.         RetryableTransaction::retryable($this->db, fn() => $this->db->executeStatement(
  69.             'INSERT INTO pickware_erp_warehouse_stock (
  70.                 id,
  71.                 product_id,
  72.                 product_version_id,
  73.                 quantity,
  74.                 warehouse_id,
  75.                 created_at
  76.             ) SELECT
  77.                 ' . SqlUuid::UUID_V4_GENERATION . ',
  78.                 product.id,
  79.                 product.version_id,
  80.                 0,
  81.                 warehouse.id,
  82.                 NOW(3)
  83.             FROM product
  84.             CROSS JOIN pickware_erp_warehouse warehouse
  85.             WHERE ' . $idFieldName . ' IN (:ids) AND product.version_id = :liveVersionId
  86.             ON DUPLICATE KEY UPDATE pickware_erp_warehouse_stock.id = pickware_erp_warehouse_stock.id',
  87.             [
  88.                 'ids' => array_map('hex2bin', $ids),
  89.                 'liveVersionId' => hex2bin(Defaults::LIVE_VERSION),
  90.             ],
  91.             [
  92.                 'ids' => Connection::PARAM_STR_ARRAY,
  93.             ],
  94.         ));
  95.     }
  97.     public function ensureProductWarehouseStocksExistsForAllProducts(): void
  98.     {
  99.         $this->db->executeStatement(
  100.             'INSERT INTO pickware_erp_warehouse_stock (
  101.                 id,
  102.                 product_id,
  103.                 product_version_id,
  104.                 quantity,
  105.                 warehouse_id,
  106.                 created_at
  107.             ) SELECT
  108.                 ' . SqlUuid::UUID_V4_GENERATION . ',
  109.                 product.id,
  110.                 product.version_id,
  111.                 0,
  112.                 warehouse.id,
  113.                 NOW(3)
  114.             FROM product
  115.             CROSS JOIN pickware_erp_warehouse warehouse
  116.             WHERE product.version_id = :liveVersionId
  117.             ON DUPLICATE KEY UPDATE pickware_erp_warehouse_stock.id = pickware_erp_warehouse_stock.id',
  118.             ['liveVersionId' => hex2bin(Defaults::LIVE_VERSION)],
  119.         );
  120.     }
  122.     private function getNewlyCreatedEntityIds(EntityWrittenEvent $entityWrittenEvent): array
  123.     {
  124.         if ($entityWrittenEvent->getContext()->getVersionId() !== Defaults::LIVE_VERSION) {
  125.             return [];
  126.         }
  128.         $ids = [];
  129.         foreach ($entityWrittenEvent->getWriteResults() as $writeResult) {
  130.             // $writeResult->getExistence() can be null, but we have no idea why and also not what this means.
  131.             $existence = $writeResult->getExistence();
  132.             if (($existence === null && $writeResult->getOperation() === EntityWriteResult::OPERATION_INSERT)
  133.                 || ($existence !== null && !$existence->exists())
  134.             ) {
  135.                 $ids[] = $writeResult->getPrimaryKey();
  136.             }
  137.         }
  139.         return $ids;
  140.     }
  141. }