Symfony从实体获取连接的用户ID

Question

In a Symfony2.8/Doctrine2 application, I need to store in each row of my SQL tables the id of the user who created or updated the row (users can connect with Ldap).

So all my entities inherited of a GenericEntity wich contains this variable (type would be string if I want to store Ldap username):

/**
 * @var integer
 *
 * @ORM\Column(name="zzCreationId", type="string", nullable=false)
 */
private $creationId;

And I use the prePersistCallback() to automatically assign this value:

/**
 * @ORM\PrePersist
 */
public function prePersistCallback()
{
    $currentUser = /* ...... ????? ....... */ ;
    if ($currentUser->getId() != null) {
        $this->creationId = $currentUser->getId() ;
    } else {
        $this->creationId = 'unknown' ;
    }
    return $this;
}

But I don't know how to retreive the connected user, or how to automatically inject it in the entity... How can I do it?

Answer 1

You can use a Doctrine entity listener/subscriber instead and to inject the security token and gets the current user logged:

// src/AppBundle/EventListener/EntityListener.php
namespace AppBundle\EventListener;

use Doctrine\ORM\Event\LifecycleEventArgs;
use AppBundle\Entity\GenericEntity;

class EntityListener
{
    private $tokenStorage;

    public function __construct(TokenStorageInterface $tokenStorage = null) 
    {
        $this->tokenStorage = $tokenStorage;
    }

    public function prePersist(LifecycleEventArgs $args)
    {
        $entity = $args->getEntity();

        // only act on some "GenericEntity" entity
        if (!$entity instanceof GenericEntity) {
            return;
        }

        if (null !== $currentUser = $this->getUser()) {
            $entity->setCreationId($currentUser->getId());
        } else {
            $entity->setCreationId(0);
        }
    }

    public function getUser()
    {
        if (!$this->tokenStorage) {
            throw new \LogicException('The SecurityBundle is not registered in your application.');
        }

        if (null === $token = $this->tokenStorage->getToken()) {
            return;
        }

        if (!is_object($user = $token->getUser())) {
            // e.g. anonymous authentication
            return;
        }

        return $user;
    }
}

Next register you listener:

# app/config/services.yml
services:
    my.listener:
        class: AppBundle\EventListener\EntityListener
        arguments: ['@security.token_storage']
        tags:
            - { name: doctrine.event_listener, event: prePersist }