GetAssignedStockIdsBySku.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventorySalesAdminUi\Model\ResourceModel;

use Magento\InventoryApi\Api\GetSourceItemsBySkuInterface;
use Magento\InventorySalesAdminUi\Model\GetStockSourceLinksBySourceCode;

class GetAssignedStockIdsBySku
{
    private $getSourceItemsBySku;

    private $getStockSourceLinksBySourceCode;

    public function __construct(
        GetSourceItemsBySkuInterface $getSourceItemsBySku,
        GetStockSourceLinksBySourceCode $getStockSourceLinksBySourceCode
    ) {
        $this->getSourceItemsBySku = $getSourceItemsBySku;
        $this->getStockSourceLinksBySourceCode = $getStockSourceLinksBySourceCode;
    }

    public function execute(string $sku): array
    {
        $sourceItems = $this->getSourceItemsBySku->execute($sku);

        $stocksIds = [];
        foreach ($sourceItems as $sourceItem) {
            $stockSourceLinks = $this->getStockSourceLinksBySourceCode->execute($sourceItem->getSourceCode());
            foreach ($stockSourceLinks as $stockSourceLink) {
                $stocksIds[] = (int)$stockSourceLink->getStockId();
            }
        }

        return array_unique($stocksIds);
    }
}