AddStockItemsObserver.php

Go to the documentation of this file.

<?php
namespace Magento\CatalogInventory\Observer;

use Magento\Catalog\Model\Product;
use Magento\Catalog\Model\ResourceModel\Product\Collection;
use Magento\CatalogInventory\Api\StockConfigurationInterface;
use Magento\CatalogInventory\Api\StockItemCriteriaInterfaceFactory;
use Magento\CatalogInventory\Api\StockItemRepositoryInterface;
use Magento\Framework\Event\Observer;
use Magento\Framework\Event\ObserverInterface;

class AddStockItemsObserver implements ObserverInterface
{
    private $criteriaInterfaceFactory;

    private $stockItemRepository;

    private $stockConfiguration;

    public function __construct(
        StockItemCriteriaInterfaceFactory $criteriaInterfaceFactory,
        StockItemRepositoryInterface $stockItemRepository,
        StockConfigurationInterface $stockConfiguration
    ) {
        $this->criteriaInterfaceFactory = $criteriaInterfaceFactory;
        $this->stockItemRepository = $stockItemRepository;
        $this->stockConfiguration = $stockConfiguration;
    }

    public function execute(Observer $observer)
    {
        $productCollection = $observer->getData('collection');
        $productIds = array_keys($productCollection->getItems());
        $criteria = $this->criteriaInterfaceFactory->create();
        $criteria->setProductsFilter($productIds);
        $criteria->setScopeFilter($this->stockConfiguration->getDefaultScopeId());
        $stockItemCollection = $this->stockItemRepository->getList($criteria);
        foreach ($stockItemCollection->getItems() as $item) {
            $product = $productCollection->getItemById($item->getProductId());
            $productExtension = $product->getExtensionAttributes();
            $productExtension->setStockItem($item);
            $product->setExtensionAttributes($productExtension);
        }
    }
}