reservequantitycollection.php

<?php
 
namespace Bitrix\Sale;
 
use Bitrix\Main;
use Bitrix\Main\DI\ServiceLocator;
use Bitrix\Sale\Reservation\BasketReservationService;
 
class ReserveQuantityCollection extends Internals\EntityCollection
{
    protected $basketItem;
 
    protected function getEntityParent()
    {
        return $this->getBasketItem();
    }
 
    protected function setBasketItem(BasketItemBase $item)
    {
        $this->basketItem = $item;
    }
 
    public function getBasketItem()
    {
        return $this->basketItem;
    }
 
    public static function getRegistryType()
    {
        return Registry::REGISTRY_TYPE_ORDER;
    }
 
    protected static function createCollectionObject() : self
    {
        $registry = Registry::getInstance(static::getRegistryType());
        $className = $registry->get(Registry::ENTITY_BASKET_RESERVE_COLLECTION);
 
        return new $className;
    }
 
    public function create()
    {
        $registry  = Registry::getInstance(static::getRegistryType());
 
        $reservedItemClassName = $registry->getReservedItemClassName();
 
        $reservedItem = $reservedItemClassName::create($this);
        $this->addItem($reservedItem);
 
        return $reservedItem;
    }
 
    public static function load(BasketItemBase $basketItem) : self
    {
        if (!$basketItem->isReservableItem())
        {
            throw new Main\SystemException('Basket item is not available for reservation');
        }
 
        $collection = static::createCollectionObject();
        $collection->setBasketItem($basketItem);
 
        if ($basketItem->getId() > 0)
        {
            $registry = Registry::getInstance(static::getRegistryType());
 
            $reservedItemClassName = $registry->getReservedItemClassName();
 
            $reservedQuantityList = $reservedItemClassName::loadForBasketItem($basketItem->getId());
            foreach ($reservedQuantityList as $item)
            {
                $item->setCollection($collection);
                $collection->addItem($item);
            }
        }
 
        return $collection;
    }
 
    public function save()
    {
        $result = new Result();
 
        if (!$this->isChanged())
        {
            return $result;
        }
 
        $basketItem = $this->getEntityParent();
 
        if ($basketItem->getId() > 0)
        {
            $itemsFromDbList = static::getList([
                'filter' => ["BASKET_ID" => $basketItem->getId()],
            ]);
            while ($item = $itemsFromDbList->fetch())
            {
                if (!$this->getItemById($item['ID']))
                {
                    static::deleteInternal($item['ID']);
                }
            }
        }
 
        foreach ($this->collection as $entity)
        {
            $r = $entity->save();
            if (!$r->isSuccess())
            {
                $result->addErrors($r->getErrors());
            }
        }
 
        $this->clearChanged();
 
        return $result;
    }
 
    public static function deleteNoDemand($basketId)
    {
        $result = new Result();
 
        $dbRes = static::getList([
            'select' => ['ID'],
            'filter' => ['=BASKET_ID' => $basketId],
        ]);
 
        while ($entity = $dbRes->fetch())
        {
            $r = static::deleteInternal($entity['ID']);
            if (!$r->isSuccess())
            {
                $result->addErrors($r->getErrors());
            }
        }
 
        return $result;
    }
 
    public function getQuantity() : float
    {
        $quantity = 0;
 
        foreach ($this->collection as $reservation)
        {
            $quantity += $reservation->getQuantity();
        }
 
        return $quantity;
    }
 
    public function getQuantityByStoreId(int $storeId) : float
    {
        $quantity = 0;
 
        foreach ($this->collection as $reservation)
        {
            if ($reservation->getStoreId() === $storeId)
            {
                $quantity += $reservation->getQuantity();
            }
        }
 
        return $quantity;
    }
 
    protected static function deleteInternal($primary)
    {
        $service = ServiceLocator::getInstance()->get('sale.basketReservation');
        return $service->delete($primary);
    }
 
    public static function getList(array $parameters = [])
    {
        return Reservation\Internals\BasketReservationTable::getList($parameters);
    }
}