EntityArrayCollection

<?php

/**
 * Custom ArrayCollection for Doctrine2 entities
 */

namespace XXX\XXX\Collections;

/**
 * Custom array collection
 *
 * Entity is only inserted once
 */
class EntityArrayCollection extends ArrayCollection
{

    /**
     * We just add value if not already exists
     *
     * If two elements are already mapped, so both have id, are equal if both ids are equal
     * 
     * Otherwise, two elements are equal if are same object without id ( same memory position )
     *
     * {@inheritDoc}
     */
    public function add($value)
    {
        if ($this->forAll(function($key, $element) use ($value) {
            
            if ($element->getId() && $value->getId()) {

                    return $element->getId() !== $value->getId();
                }

                return spl_object_hash($element) !== spl_object_hash($value);
            })) {

            parent::add($value);
        }

        return $this;
    }


    /**
     * We just remove element when exists inside collection
     *
     * If two elements are already mapped, so both have id, are equal if both ids are equal
     * 
     * Otherwise, two elements are equal if are same object without id ( same memory position )
     *
     * {@inheritDoc}
     */
    public function removeElement($value)
    {
        foreach ($this as $key => $element) {
            
            if ($element->getId() && $value->getId()) {

                if ($element->getId() === $value->getId()) {

                    $this->remove($key);

                    break;
                }
            } else {

                if (spl_object_hash($element) === spl_object_hash($value)) {

                    $this->remove($key);

                    break;
                }
            }


        }

        return $this;
    }
}