AddStockDataToCollection.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventoryCatalog\Model\ResourceModel;

use Magento\Catalog\Model\ResourceModel\Product\Collection;
use Magento\InventoryIndexer\Indexer\IndexStructure;
use Magento\InventoryIndexer\Model\StockIndexTableNameResolverInterface;

class AddStockDataToCollection
{
    private $stockIndexTableNameResolver;

    public function __construct(StockIndexTableNameResolverInterface $stockIndexTableNameResolver)
    {
        $this->stockIndexTableNameResolver = $stockIndexTableNameResolver;
    }

    public function execute(Collection $collection, bool $isFilterInStock, int $stockId)
    {
        $stockIndexTableName = $this->stockIndexTableNameResolver->execute($stockId);

        $resource = $collection->getResource();
        $collection->getSelect()->join(
            ['product' => $resource->getTable('catalog_product_entity')],
            sprintf('product.entity_id = %s.entity_id', Collection::MAIN_TABLE_ALIAS),
            []
        );
        $collection->getSelect()
            ->join(
                ['stock_status_index' => $stockIndexTableName],
                'product.sku = stock_status_index.' . IndexStructure::SKU,
                [IndexStructure::IS_SALABLE]
            );

        if ($isFilterInStock) {
            $collection->getSelect()
                ->where('stock_status_index.' . IndexStructure::IS_SALABLE . ' = ?', 1);
        }
    }
}