BulkInventoryTransfer.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;
use Magento\InventoryCatalogApi\Model\GetProductTypesBySkusInterface;
use Magento\InventoryConfigurationApi\Model\IsSourceItemManagementAllowedForProductTypeInterface;

class BulkInventoryTransfer
{
    private $resourceConnection;

    private $getProductTypesBySkus;

    private $isSourceItemManagementAllowedForProductType;

    private $defaultSourceProvider;

    private $setDataToLegacyStockItem;

    private $bulkZeroLegacyStockItem;

    public function __construct(
        ResourceConnection $resourceConnection,
        GetProductTypesBySkusInterface $getProductTypesBySkus,
        IsSourceItemManagementAllowedForProductTypeInterface $isSourceItemManagementAllowedForProductType,
        DefaultSourceProviderInterface $defaultSourceProvider,
        SetDataToLegacyStockItem $setDataToLegacyStockItem,
        BulkZeroLegacyStockItem $bulkZeroLegacyStockItem
    ) {
        $this->resourceConnection = $resourceConnection;
        $this->getProductTypesBySkus = $getProductTypesBySkus;
        $this->isSourceItemManagementAllowedForProductType = $isSourceItemManagementAllowedForProductType;
        $this->defaultSourceProvider = $defaultSourceProvider;
        $this->setDataToLegacyStockItem = $setDataToLegacyStockItem;
        $this->bulkZeroLegacyStockItem = $bulkZeroLegacyStockItem;
    }

    private function getSourceItemData(string $sku, string $source): ?array
    {
        $connection = $this->resourceConnection->getConnection();
        $tableName = $this->resourceConnection->getTableName(SourceItem::TABLE_NAME_SOURCE_ITEM);

        $query = $connection->select()->from($tableName)
            ->where(SourceItemInterface::SOURCE_CODE . ' = ?', $source)
            ->where(SourceItemInterface::SKU . ' = ?', $sku);

        $res = $connection->fetchRow($query);
        if ($res === false) {
            return null;
        }

        return $res;
    }

    private function transferInventory(
        string $sku,
        string $originSource,
        string $destinationSource
    ): void {
        $connection = $this->resourceConnection->getConnection();
        $tableName = $this->resourceConnection->getTableName(SourceItem::TABLE_NAME_SOURCE_ITEM);

        $orgSourceItem = $this->getSourceItemData($sku, $originSource);
        $dstSourceItem = $this->getSourceItemData($sku, $destinationSource);

        $orgSourceItemQty = $orgSourceItem === null ? 0.0 : (float) $orgSourceItem[SourceItemInterface::QUANTITY];
        $dstSourceItemQty = $dstSourceItem === null ? 0.0 : (float) $dstSourceItem[SourceItemInterface::QUANTITY];

        $finalQuantity = $orgSourceItemQty + $dstSourceItemQty;

        if ($orgSourceItem !== null) {
            $status = (int) $orgSourceItem[SourceItemInterface::STATUS];
        } elseif ($dstSourceItemQty !== null) {
            $status = (int) $dstSourceItem[SourceItemInterface::STATUS];
        } else {
            $status = (int) SourceItemInterface::STATUS_OUT_OF_STOCK;
        }

        $updateOperation = [
            SourceItemInterface::QUANTITY => $finalQuantity,
            SourceItemInterface::STATUS => $status,
        ];

        if ($dstSourceItem === null) {
            $updateOperation[SourceItemInterface::SOURCE_CODE] = $destinationSource;
            $updateOperation[SourceItemInterface::SKU] = $sku;

            $connection->insert($tableName, $updateOperation);
        } elseif ($orgSourceItem !== null) {
            $connection->update($tableName, $updateOperation, [
                SourceItemInterface::SOURCE_CODE . '=?' => $destinationSource,
                SourceItemInterface::SKU . '=?' => $sku,
            ]);
        }

        // Align legacy stock
        if ($destinationSource === $this->defaultSourceProvider->getCode()) {
            $this->setDataToLegacyStockItem->execute($sku, $finalQuantity, $status);
        }
    }

    private function clearSource(array $skus, string $source, bool $unassign)
    {
        $connection = $this->resourceConnection->getConnection();
        $tableName = $this->resourceConnection->getTableName(SourceItem::TABLE_NAME_SOURCE_ITEM);

        if ($unassign) {
            $connection->delete($tableName, [
                SourceItemInterface::SOURCE_CODE . '=?' => $source,
                SourceItemInterface::SKU . ' IN(?)' => $skus,
            ]);
        } else {
            $connection->update($tableName, [
                SourceItemInterface::QUANTITY => 0,
                SourceItemInterface::STATUS => SourceItemInterface::STATUS_OUT_OF_STOCK,
            ], [
                SourceItemInterface::SOURCE_CODE . '=?' => $source,
                SourceItemInterface::SKU . ' IN(?)' => $skus,
            ]);
        }

        // Align legacy stock
        if ($source === $this->defaultSourceProvider->getCode()) {
            $this->bulkZeroLegacyStockItem->execute($skus);
        }
    }

    public function execute(
        array $skus,
        string $originSource,
        string $destinationSource,
        bool $unassignFromOrigin
    ): void {
        $connection = $this->resourceConnection->getConnection();
        $types = $this->getProductTypesBySkus->execute($skus);

        $connection->beginTransaction();
        foreach ($types as $sku => $type) {
            if ($this->isSourceItemManagementAllowedForProductType->execute($type)) {
                $this->transferInventory($sku, $originSource, $destinationSource);
            }
        }

        $this->clearSource($skus, $originSource, $unassignFromOrigin);
        $connection->commit();
    }
}