Added support for getting documents in unified Doctrine Listener

This commit is contained in:
Josh Worden 2014-04-02 10:06:24 -05:00
parent ff95945819
commit 1628413e65

View file

@ -197,7 +197,7 @@ class Listener implements EventSubscriber
public function postPersist(EventArgs $eventArgs)
{
$entity = $eventArgs->getEntity();
$entity = (method_exists($eventArgs, 'getEntity')) ? $eventArgs->getEntity() : $eventArgs->getDocument();
if ($entity instanceof $this->objectClass && $this->isObjectIndexable($entity)) {
$this->scheduledForInsertion[] = $entity;
@ -206,7 +206,7 @@ class Listener implements EventSubscriber
public function postUpdate(EventArgs $eventArgs)
{
$entity = $eventArgs->getEntity();
$entity = (method_exists($eventArgs, 'getEntity')) ? $eventArgs->getEntity() : $eventArgs->getDocument();
if ($entity instanceof $this->objectClass) {
if ($this->isObjectIndexable($entity)) {
@ -224,7 +224,7 @@ class Listener implements EventSubscriber
*/
public function preRemove(EventArgs $eventArgs)
{
$entity = $eventArgs->getEntity();
$entity = (method_exists($eventArgs, 'getEntity')) ? $eventArgs->getEntity() : $eventArgs->getDocument();
if ($entity instanceof $this->objectClass) {
$this->scheduleForDeletion($entity);