Collection.php

Go to the documentation of this file.

<?php
namespace Magento\Framework\Pricing\Price;

use Magento\Framework\Pricing\SaleableInterface;

class Collection implements \Iterator
{
    protected $pool;

    protected $saleableItem;

    protected $priceFactory;

    protected $quantity;

    protected $contains;

    protected $excludes;

    protected $priceModels;

    public function __construct(
        SaleableInterface $saleableItem,
        Factory $priceFactory,
        Pool $pool,
        $quantity
    ) {
        $this->saleableItem = $saleableItem;
        $this->priceFactory = $priceFactory;
        $this->pool = $pool;
        $this->quantity = $quantity;
        $this->priceModels = [];
    }

    public function rewind()
    {
        return $this->pool->rewind();
    }

    public function current()
    {
        return $this->get($this->key());
    }

    public function key()
    {
        return $this->pool->key();
    }

    public function next()
    {
        return $this->pool->next();
    }

    public function valid()
    {
        return $this->pool->valid();
    }

    public function get($code)
    {
        if (!isset($this->priceModels[$code])) {
            $this->priceModels[$code] = $this->priceFactory->create(
                $this->saleableItem,
                $this->pool[$code],
                $this->quantity
            );
        }
        return $this->priceModels[$code];
    }
}