ServiceLocator/EventManager Traits

EventManagerAwareTrait.php

<?php

namespace Protec\Stdlib;

use \Zend\EventManager\EventManagerInterface;
use \Zend\ServiceManager\ServiceLocatorAwareInterface;

trait EventManagerAwareTrait
{
    protected $event_manager;

    /**
     * getEventManager
     *
     * @return EventManagerInterface
     */
    public function getEventManager()
    {
        if (is_null($this->event_manager)
            && $this instanceof ServiceLocatorAwareInterface
            && ! is_null($this->getServiceLocator())
            && $this->getServiceLocator()->has('EventManager')
        ) {
            $this->setEventManager(
                $this->getServiceLocator()->get('EventManager'));
        }

        return $this->event_manager;
    }

    /**
     * setEventManager
     *
     * @param EventManagerInterface $eventManager
     */
    public function setEventManager(EventManagerInterface $eventManager)
    {
        $this->event_manager = $eventManager;

        $this->event_manager->addIdentifiers(
            array(
                get_called_class(),
                __CLASS__
            )
        );

        return $this;
    }
}