From adbf201bcb3b2e0783bf92c06f25b55e057fd702 Mon Sep 17 00:00:00 2001
From: Alex Mayer <amayer5125@gmail.com>
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;