在ZF2中创建具有依赖关系(依赖注入)的doctrine存储库

时间:2015-11-10 13:16:13

标签: php doctrine-orm dependency-injection zend-framework2 doctrine

我想创建一个具有硬依赖关系的存储库。我找到了 this blog post by Jurian Sluisman ,但他建议从服务管理器获取存储库并将其注入服务中。

如果我能够通过EntityManagerObjectManager实例通常使用getRepository方法获取我的自定义存储库并注入依赖项,那会好得多:

$objectManager->getRepository('My\Entity\Class');

如何在我的存储库中使用构造函数注入,并且仍然使用ObjectManager方法直接从getRepository获取它们?

1 个答案:

答案 0 :(得分:12)

Doctrine使用a factory class Doctrine\ORM\EntityManagerInterface\DefaultRepositoryFactory创建存储库实例。如果未设置自定义工厂,则会在here in the getRepositoryFactory method中创建此默认工厂the Doctrine\ORM\Configuration class

通过定义自定义repository_factory,我们可以覆盖此默认工厂类,并将自定义逻辑添加到将注入硬依赖项的工厂:

为了说明如何执行此操作,我将展示一个示例,其中存储库工厂类通过构造函数注入创建依赖于ServiceLocator实例的存储库。

1)创建一个实现学说RepositoryFactory接口的自定义工厂类

此类看起来非常类似于学说DefaultRepositoryFactory 类。

<?php

namespace My\ORM\Repository;

use Doctrine\Common\Persistence\ObjectRepository;    
use Doctrine\ORM\Repository\RepositoryFactory;
use Doctrine\ORM\EntityManagerInterface;
use Zend\ServiceManager\ServiceLocatorAwareInterface;
use Zend\ServiceManager\ServiceLocatorAwareTrait;
use Zend\ServiceManager\ServiceLocatorInterface;

class CustomRepositoryFactory implements RepositoryFactory, ServiceLocatorAwareInterface
{
    use ServiceLocatorAwareTrait;

    /**
     * @var ObjectRepository[]
     */
    private $repositoryList = array();

    /**
     * @var ServiceLocator
     */
    protected $serviceLocator;

    /**
     * @param ServiceLocatorInterface $serviceLocator
     */
    public function __construct(ServiceLocatorInterface $serviceLocator)
    {
        $this->serviceLocator = $serviceLocator;
    }

    /**
     * {@inheritdoc}
     */
    public function getRepository(EntityManagerInterface $entityManager, $entityName)
    {
        $repositoryHash = $entityManager->getClassMetadata($entityName)->getName() . spl_object_hash($entityManager);

        if (isset($this->repositoryList[$repositoryHash])) {
            return $this->repositoryList[$repositoryHash];
        }

        return $this->repositoryList[$repositoryHash] = $this->createRepository($entityManager, $entityName);
    }

    /**
     * @param EntityManagerInterface $entityManager The EntityManager instance.
     * @param string                 $entityName    The name of the entity.
     * @return ObjectRepository
     */
    private function createRepository(EntityManagerInterface $entityManager, $entityName)
    {
        /* @var $metadata \Doctrine\ORM\Mapping\ClassMetadata */
        $metadata            = $entityManager->getClassMetadata($entityName);
        $repositoryClassName = $metadata->customRepositoryClassName
            ?: $entityManager->getConfiguration()->getDefaultRepositoryClassName();

        // Constructor injection, I check with subclass of but it is just an example
        if(is_subclass_of($repositoryClassName, ServiceLocatorAwareInterface::class)){
            $serviceLocator = $this->getServiceLocator()
            $repository = new $repositoryClassName($entityManager, $metadata, $serviceLocator);
        }else{
            $repository = new $repositoryClassName($entityManager, $metadata);
        }
        return $repository;
    }
}

2)为存储库工厂创建工厂

<?php
namespace My\ORM\Repository\Factory;

use My\ORM\Repository\CustomRepositoryFactory;
use Zend\Cache\Storage\StorageInterface;
use Zend\ServiceManager\FactoryInterface;
use Zend\ServiceManager\ServiceLocatorInterface;

class CustomRepositoryFactoryFactory implements FactoryInterface
{
    /**
     * @param  ServiceLocatorInterface $serviceLocator
     * @return StorageInterface
     */
    public function createService(ServiceLocatorInterface $serviceLocator)
    {
        return new CustomRepositoryFactory($serviceLocator);
    }
}

3)在service_manager config

中注册存储库工厂的工厂
'service_manager' => array(
    'factories' => array(
        'My\ORM\Repository\CustomRepositoryFactory' => 'My\ORM\Repository\Factory\CustomRepositoryFactoryFactory'
    )
)

4)在doctrine config

中注册存储库工厂
'doctrine' => array(
    'configuration' => array(
        'orm_default' => array(
            'repository_factory' => 'My\ORM\Repository\CustomRepositoryFactory'
        )
    )
)