From adbf201bcb3b2e0783bf92c06f25b55e057fd702 Mon Sep 17 00:00:00 2001 From: Alex Mayer Date: Sat, 26 Mar 2022 20:26:24 -0400 Subject: [PATCH] Use onPostConsume Hook for Calculating Messages Consumed Fixes issue where no messages are consumed if value is set to 1. The first messages will be received and processing will stop. --- .../Extension/LimitConsumedMessagesExtension.php | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/pkg/enqueue/Consumption/Extension/LimitConsumedMessagesExtension.php b/pkg/enqueue/Consumption/Extension/LimitConsumedMessagesExtension.php index e89957a72..4989e14b9 100644 --- a/pkg/enqueue/Consumption/Extension/LimitConsumedMessagesExtension.php +++ b/pkg/enqueue/Consumption/Extension/LimitConsumedMessagesExtension.php @@ -2,13 +2,13 @@ namespace Enqueue\Consumption\Extension; -use Enqueue\Consumption\Context\PostMessageReceived; +use Enqueue\Consumption\Context\PostConsume; use Enqueue\Consumption\Context\PreConsume; -use Enqueue\Consumption\PostMessageReceivedExtensionInterface; +use Enqueue\Consumption\PostConsumeExtensionInterface; use Enqueue\Consumption\PreConsumeExtensionInterface; use Psr\Log\LoggerInterface; -class LimitConsumedMessagesExtension implements PreConsumeExtensionInterface, PostMessageReceivedExtensionInterface +class LimitConsumedMessagesExtension implements PreConsumeExtensionInterface, PostConsumeExtensionInterface { /** * @var int @@ -18,7 +18,7 @@ class LimitConsumedMessagesExtension implements PreConsumeExtensionInterface, Po /** * @var int */ - protected $messageConsumed; + protected $messageConsumed = 0; /** * @param int $messageLimit @@ -26,7 +26,6 @@ class LimitConsumedMessagesExtension implements PreConsumeExtensionInterface, Po public function __construct(int $messageLimit) { $this->messageLimit = $messageLimit; - $this->messageConsumed = 0; } public function onPreConsume(PreConsume $context): void @@ -37,7 +36,7 @@ public function onPreConsume(PreConsume $context): void } } - public function onPostMessageReceived(PostMessageReceived $context): void + public function onPostConsume(PostConsume $context): void { ++$this->messageConsumed;