通过 Twig 扩展中的自定义存储库获取数据

Question

我想在 symfony 2 Web 应用程序的每个页面上显示新通知。 有人建议我为此使用 Twig Extension。我在该扩展中创建了一个函数 getFriendRequests，但我不知道通过 twig 扩展中的自定义存储库获取数据是否是一个好习惯：现在它给了我错误，它找不到 getDoctrine 方法。

<?php

namespace Tennisconnect\DashboardBundle\Extension;

class NotificationTwigExtension extends \Twig_Extension
{
    public function getFriendRequests($user)
    {
        $users = $this->getDoctrine()
            ->getRepository('TennisconnectUserBundle:User')
            ->getFriendRequests();
        return count($users);
    }

    public function getName()
    {
        return 'notification';
    }

    public function getFunctions()
    {
        return array(
            'getFriendRequests' => new \Twig_Function_Method($this, 'getFriendRequests'));
    }
}

Answer 1

我认为直接从树枝扩展中获取数据并没有那么糟糕。毕竟，如果您不在这里执行此操作，则需要之前获取这些记录，然后将它们传递给扩展程序以进行显示。

重要的一点是像您已经在做的那样在存储库中执行 DQL/SQL 操作。这对于将数据库语句与项目的其他部分分开非常重要。

您遇到的问题是该类中不存在方法 getDoctrine 。据我了解，您从扩展了 FrameworkBundle 基本控制器的控制器中获取了此代码。 FrameworkBundle 的基本控制器定义了此方法。

为了解决这个问题，您必须将正确的服务注入您的扩展中。这是基于依赖注入容器的。您当然为您的 twig 扩展定义了一个服务，类似于以下定义：

services:
  acme.twig.extension.notification:
    class: Acme\WebsiteBundle\Twig\Extension\NotificationExtension
    tags:
      - { name: twig.extension }

现在的技巧是像这样注入您需要的依赖项：

services:
  acme.twig.extension.notification:
    class: Acme\WebsiteBundle\Twig\Extension\NotificationExtension
    arguments:
      doctrine: "@doctrine"
    tags:
      - { name: twig.extension }

然后，在您的扩展中，您定义一个接收学说依赖项的构造函数：

    use Symfony\Bridge\Doctrine\RegistryInterface;

    class NotificationTwigExtension extends \Twig_Extension
    {
        protected $doctrine;

        public function __construct(RegistryInterface $doctrine)
        {
            $this->doctrine = $doctrine;
        }

        // Now you can do $this->doctrine->getRepository('TennisconnectUserBundle:User')

        // Rest of twig extension
    }

这是依赖注入。您可以看到我之前回答的有关访问控制器外部服务的另一个问题：此处

希望这会有所帮助。

问候，
马特

Answer 2

与 mongo 相同：

在 config.yml

services:
    user.twig.extension:
        class: MiProject\CoreBundle\Twig\Extension\MiFileExtension
        arguments:
          doctrine: "@doctrine.odm.mongodb.document_manager"
        tags:
          -  { name: twig.extension }

和 Twig\Extensions\MiFile.php 中

<?php

namespace MiProject\CoreBundle\Twig\Extension;

use Symfony\Component\HttpKernel\KernelInterface;

class MiFileExtension extends \Twig_Extension
{
    protected $doctrine;

    public function __construct( $doctrine){
        $this->doctrine = $doctrine;
    }
    public function getTransactionsAmount($user_id){
    return $results = $this->doctrine
    ->createQueryBuilder('MiProjectCoreBundle:Transaction')             
    ->hydrate(false)
    ->getQuery()            
    ->count();
    }

    Rest of mi code ... 

}