Collection.php

Go to the documentation of this file.

<?php
namespace Magento\Framework\Data\Tree\Node;

use Magento\Framework\Data\Tree;
use Magento\Framework\Data\Tree\Node;

class Collection implements \ArrayAccess, \IteratorAggregate, \Countable
{
    private $_nodes;

    private $_container;

    public function __construct($container)
    {
        $this->_nodes = [];
        $this->_container = $container;
    }

    public function getNodes()
    {
        return $this->_nodes;
    }

    public function getIterator()
    {
        return new \ArrayIterator($this->_nodes);
    }

    public function offsetSet($key, $value)
    {
        $this->_nodes[$key] = $value;
    }

    public function offsetGet($key)
    {
        return $this->_nodes[$key];
    }

    public function offsetUnset($key)
    {
        unset($this->_nodes[$key]);
    }

    public function offsetExists($key)
    {
        return isset($this->_nodes[$key]);
    }

    public function add(Node $node)
    {
        $node->setParent($this->_container);

        // Set the Tree for the node
        if ($this->_container->getTree() instanceof Tree) {
            $node->setTree($this->_container->getTree());
        }

        $this->_nodes[$node->getId()] = $node;

        return $node;
    }

    public function delete($node)
    {
        if (isset($this->_nodes[$node->getId()])) {
            unset($this->_nodes[$node->getId()]);
        }
        return $this;
    }

    public function count()
    {
        return count($this->_nodes);
    }

    public function lastNode()
    {
        if (!empty($this->_nodes)) {
            $result = end($this->_nodes);
            reset($this->_nodes);
        } else {
            $result = null;
        }

        return $result;
    }

    public function searchById($nodeId)
    {
        if (isset($this->_nodes[$nodeId])) {
            return $this->_nodes[$nodeId];
        }
        return null;
    }
}