Cycle\ORM\Relation\Pivoted\PivotedCollection

Complexity

Total Complexity

6

Size/Duplication

Total Lines	57
Duplicated Lines	0 %

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	setPivot()	0	3	1
A	hasPivot()	0	3	1
A	getPivotContext()	0	3	1
A	getPivot()	0	3	1
A	createFrom()	0	6	1

<?php

/**
 * Cycle DataMapper ORM
 *
 * @license   MIT
 * @author    Anton Titov (Wolfy-J)
 */

declare(strict_types=1);

namespace Cycle\ORM\Relation\Pivoted;

use Doctrine\Common\Collections\ArrayCollection;

/**
 * Collection with associated relation context. Attention, pivot context is lost when collection is partitioned or
 * filtered.
 */
final class PivotedCollection extends ArrayCollection implements PivotedCollectionInterface
{
    /** @var \SplObjectStorage */
    protected $pivotContext;

    /**
     * @param array                  $elements
     * @param \SplObjectStorage|null $pivotData
     */
    public function __construct(array $elements = [], \SplObjectStorage $pivotData = null)
    {
        parent::__construct($elements);
        $this->pivotContext = $pivotData ?? new \SplObjectStorage();
    }

    /**
     * @inheritdoc
     */
    public function hasPivot($element): bool
    {
        return $this->pivotContext->offsetExists($element);
    }

    /**
     * @inheritdoc
     */
    public function getPivot($element)
    {
        return $this->pivotContext[$element] ?? null;
    }

    /**
     * @inheritdoc
     */
    public function setPivot($element, $pivot): void
    {
        $this->pivotContext[$element] = $pivot;
    }

    /**
     * @inheritdoc
     */
    public function getPivotContext(): \SplObjectStorage
    {
        return $this->pivotContext;
    }

    /**
     * @param array $elements
     * @return PivotedCollection
     */
    protected function createFrom(array $elements)
    {
        $new = parent::createFrom($elements);
        $new->pivotContext = $this->pivotContext;

        return $new;
    }
}