BulkSourceUnassign.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventoryCatalog\Model\ResourceModel;

use Magento\Framework\App\ResourceConnection;
use Magento\Inventory\Model\ResourceModel\SourceItem;
use Magento\InventoryApi\Api\Data\SourceItemInterface;
use Magento\InventoryCatalogApi\Api\DefaultSourceProviderInterface;

class BulkSourceUnassign
{
    private $resourceConnection;

    private $bulkZeroLegacyStockItem;

    private $defaultSourceProvider;

    public function __construct(
        ResourceConnection $resourceConnection,
        DefaultSourceProviderInterface $defaultSourceProvider,
        BulkZeroLegacyStockItem $bulkZeroLegacyStockItem
    ) {
        $this->resourceConnection = $resourceConnection;
        $this->bulkZeroLegacyStockItem = $bulkZeroLegacyStockItem;
        $this->defaultSourceProvider = $defaultSourceProvider;
    }

    public function execute(array $skus, array $sourceCodes): int
    {
        $connection = $this->resourceConnection->getConnection();
        $tableName = $this->resourceConnection->getTableName(SourceItem::TABLE_NAME_SOURCE_ITEM);

        $connection->beginTransaction();

        $count = (int) $connection->delete($tableName, [
            SourceItemInterface::SOURCE_CODE . ' IN (?)' => $sourceCodes,
            SourceItemInterface::SKU . ' IN (?)' => $skus,
        ]);

        // Legacy stock update
        if (in_array($this->defaultSourceProvider->getCode(), $sourceCodes)) {
            $this->bulkZeroLegacyStockItem->execute($skus);
        }

        $connection->commit();

        return $count;
    }
}