Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -18,15 +18,14 @@ class LimitConsumedMessagesExtension implements PreConsumeExtensionInterface, Po
/**
* @var int
*/
protected $messageConsumed;
protected $messageConsumed = 0;

/**
* @param int $messageLimit
*/
public function __construct(int $messageLimit)
{
$this->messageLimit = $messageLimit;
$this->messageConsumed = 0;
}

public function onPreConsume(PreConsume $context): void
Expand All @@ -37,7 +36,7 @@ public function onPreConsume(PreConsume $context): void
}
}

public function onPostMessageReceived(PostMessageReceived $context): void
public function onPostConsume(PostConsume $context): void
{
++$this->messageConsumed;

Expand Down