src/AppBundle/Doctrine/EntityListenerResolver.php line 28

Open in your IDE?
  1. <?php
  2. namespace AppBundle\Doctrine;
  4. use Doctrine\ORM\Mapping\DefaultEntityListenerResolver;
  5. use Symfony\Component\DependencyInjection\ContainerInterface;
  7. class EntityListenerResolver extends DefaultEntityListenerResolver
  8. {
  9.     const DIC_ID = 'app.doctrine.entity_listener_resolver';
  11.     private $container;
  12.     private $mapping;
  14.     public function __construct(ContainerInterface $container)
  15.     {
  16.         $this->container = $container;
  17.         $this->mapping = [];
  18.     }
  20.     public function addMapping($className, $service)
  21.     {
  22.         $this->mapping[$className] = $service;
  23.     }
  25.     public function resolve($className)
  26.     {
  27.         if (isset($this->mapping[$className]) && $this->container->has($this->mapping[$className])) {
  28.             return $this->container->get($this->mapping[$className]);
  29.         }
  31.         return parent::resolve($className);
  32.     }
  33. }