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)) {
            if ($this instanceof ServiceLocatorAwareInterface) {
                if (!is_null($this->getServiceLocator())) {
                    if ($this->getServiceLocator()->has('EventManager')) {
                        $event_manager = $this->getServiceLocator()->get('EventManager');
                        $this->setEventManager($event_manager);
                    }
                }
            }
        }

        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;
    }
}