[EventDispatcher] make listeners removable from an executed listener

This commit is contained in:
Christian Flothmann 2015-04-14 22:37:49 +02:00
parent 5d428b7091
commit 54bb3995ab
2 changed files with 25 additions and 0 deletions

View File

@ -88,6 +88,18 @@ class TraceableEventDispatcher implements EventDispatcherInterface, TraceableEve
*/
public function removeListener($eventName, $listener)
{
if (isset($this->wrappedListeners[$this->lastEventId])) {
foreach ($this->wrappedListeners[$this->lastEventId] as $wrappedListener) {
$originalListener = $this->wrappedListeners[$this->lastEventId][$wrappedListener];
if ($originalListener === $listener) {
unset($this->wrappedListeners[$this->lastEventId][$wrappedListener]);
return $this->dispatcher->removeListener($eventName, $wrappedListener);
}
}
}
return $this->dispatcher->removeListener($eventName, $listener);
}

View File

@ -223,6 +223,19 @@ class TraceableEventDispatcherTest extends \PHPUnit_Framework_TestCase
$kernel->handle($request);
}
public function testListenerCanRemoveItselfWhenExecuted()
{
$eventDispatcher = new TraceableEventDispatcher(new EventDispatcher(), new Stopwatch());
$listener1 = function () use ($eventDispatcher, &$listener1) {
$eventDispatcher->removeListener('foo', $listener1);
};
$eventDispatcher->addListener('foo', $listener1);
$eventDispatcher->addListener('foo', function () {});
$eventDispatcher->dispatch('foo');
$this->assertCount(1, $eventDispatcher->getListeners('foo'), 'expected listener1 to be removed');
}
protected function getHttpKernel($dispatcher, $controller)
{
$resolver = $this->getMock('Symfony\Component\HttpKernel\Controller\ControllerResolverInterface');