Making ZF2 services available within Doctrine entities is not as trivial as with objects that are created via the ZF2 factory methods. The problem is that Doctrine Entities are not generated via regular ZF2 factory methods but by the Doctrine Entity Manager. Nevertheless it’s possible to inject dependencies into Doctrine entities.

Doctrine’s event manager offers a postLoad event which is called after the entity has been loaded. We can add a new listener to this event during our regular bootstrapping process.

i.e. module/Application/Module.php

public function onBootstrap(MvcEvent $e)
{
    $sm = $e->getApplication()->getServiceManager();
    $em = $sm->get('doctrine.entitymanager.orm_default');
    $dem = $em->getEventManager();
    $dem->addEventListener(array(\Doctrine\ORM\Events::postLoad), new ServiceManagerListener($sm));
}

Next we need to implement the listener class. Please note that with this implementation only entities that implement ServiceLocatorAwareInterface will get the service manager injected. The nice thing about this is that this makes the entities compatible with ZF2 factories. This way entities that are created via ZF2 factories will get the service manager injected as well.

<?php
use Zend\ServiceManager\ServiceManager;
use Zend\ServiceManager\ServiceLocatorAwareInterface;
class ServiceManagerListener
{
    protected $sm;

    public function __construct(ServiceManager $sm)
    {
        $this->sm = $sm;
    }

    public function postLoad($eventArgs)
    {
        $entity = $eventArgs->getEntity();
        $class = new \ReflectionClass($entity);
        if ($class instanceof ServiceLocatorAwareInterface)) {
            $entity->setServiceLocator($this->sm);
        }
    }
}

You could implement the ServiceLocatorAwareInterface for each entity individually but this seems to be tedious. Instead we will create a ServiceLocatorAwareEntity class which implements Zend\ServiceManager\ServiceLocatorAwareInterface and then simply make the entities that need access to the service manager inherit from this class.

<?php
use Zend\ServiceManager\ServiceLocatorAwareInterface;
use Zend\ServiceManager\ServiceLocatorInterface;

class ServiceLocatorAwareEntity implements ServiceLocatorAwareInterface
{
    protected $sm;

    /**
     * Set the service locator
     *
     * @param ServiceLocatorInterface $sm
     *
     * @return void
     */
    public function setServiceLocator(ServiceLocatorInterface $sm)
    {
        $this->sm = $sm;
    }

    /**
     * Get the service locator
     *
     * @return ServiceLocator ServiceLocator instance
     */
    public function getServiceLocator()
    {
        return $this->sm;
    }
}

As mentioned above inherit from ServiceLocatorAwareEntity. Make sure to add @ORM\HasLifecycleCallbacks to make sure doctrine triggers the postLoad event for this entity.

/**
 * Foo
 *
 * @ORM\Entity
 * @ORM\Table(name="foo")
 * @ORM\HasLifecycleCallbacks
 */
class Foo extends ServiceLocatorAwareEntity
{

}