Question

是否可以通过某种类型的walker运行所有doctrine查询，以便我可以根据当前用户的凭据修改查询？理想情况下，我不必在每个查询上为自定义walker显式调用setHint，因为这会限制我将当前SecurityContext传递给walker的能力。

另外，我不想使用Doctrine Filter，因为我无法使用过滤器修改连接条件，并且我将被迫使用“IN”子句，这会严重影响性能

目前，我正在使用一种基于用户凭据修改QueryBuilder的服务，但这变得乏味，因为我需要在每次创建新的QueryBuilder时调用该服务，而且当存储库更加困难时发挥作用（因为我需要将服务注入需要修改查询的每个存储库。

希望我已经清楚地解释了这一点。感谢任何反馈！

Answer 1

我想我已经解决了自己的问题。如果其他人有更优雅的方式来实现这些结果，请随时解释。为了修改我的所有查询，我创建了一个自定义的EntityManager和自定义的EntityRepository。

在我的自定义EntityManager中，我覆盖了2种方法。 create（）和getRepository（）

public static function create($conn, Configuration $config, EventManager $eventManager = null)
{
    if ( ! $config->getMetadataDriverImpl()) {
        throw ORMException::missingMappingDriverImpl();
    }

    switch (true) {
        case (is_array($conn)):
            $conn = \Doctrine\DBAL\DriverManager::getConnection(
                $conn, $config, ($eventManager ?: new EventManager())
            );
            break;

        case ($conn instanceof Connection):
            if ($eventManager !== null && $conn->getEventManager() !== $eventManager) {
                 throw ORMException::mismatchedEventManager();
            }
            break;

        default:
            throw new \InvalidArgumentException("Invalid argument: " . $conn);
    }

    return new MyCustomEntityManager($conn, $config, $conn->getEventManager());
}

此方法中唯一改变的是我返回自己的EntityManger（ MyCustomEntityManager ）。然后，我按如下方式覆盖了getRepository方法：

public function getRepository($entityName)
{
    $entityName = ltrim($entityName, '\\');

    if (isset($this->repositories[$entityName])) {
        return $this->repositories[$entityName];
    }

    $metadata = $this->getClassMetadata($entityName);
    $repositoryClassName = $metadata->customRepositoryClassName;

    if ($repositoryClassName === null) {
        $repositoryClassName = "Acme\DemoBundle\Doctrine\ORM\MyCustomEntityRepository";
    }

    $repository = new $repositoryClassName($this, $metadata);

    $this->repositories[$entityName] = $repository;

    return $repository;
}

在这里，我只修改了一行。我没有依赖DBAL配置来检索默认的$ repositoryClassName，而是指定了我自己的默认存储库 Acme \ DemoBundle \ Doctrine \ ORM \ MyCustomEntityRepository 。

创建自己的自定义EntityRepository后，天空就是极限。您可以将服务注入存储库（我目前使用JMS Di注释，如下所述），或者在createQueryBuilder方法中对QueryBuilder执行自定义操作，如下所示：

use JMS\DiExtraBundle\Annotation as DI;

class MyCustomEntityRepository extends EntityRepository
{
    private $myService;

    public function createQueryBuilder($alias)
    {
         $queryBuilder = parent::createQueryBuilder($alias);

          /** INSERT CUSTOM CODE HERE **/

          return $queryBuilder;
    }

    /**
    * @DI\InjectParams({
    *     "myService" = @DI\Inject("my_service_id")
    * })
    */
    public function setMyService(MyServiceInterface $myService)
    {
        $this->myService = $myService;
    }
}

创建自己的EntityRepository之后，您应该拥有所有需要此自定义功能的存储库来扩展MyCustomEntityRepository。你甚至可以更进一步，创建自己的QueryBuilder来进一步扩展它。

Answer 2

您可以使用defaultQueryHint（Doctrine 2.5新功能）配置选项编写custom AST Walker并设置应用程序以使用此walker进行所有查询：

<?php
/** @var \Doctrine\ORM\EntityManager $em */
$em->getConfiguration()->setDefaultQueryHint(
    Query::HINT_CUSTOM_TREE_WALKERS,
    ['YourWalkerFQClassName']
)

Symfony2 / Doctrine - 修改所有查询

2 个答案: