nsq/nsq-bundle
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Extract getMessage method from NsqTransport to NsqReceivedStamp

Browse Source
This commit is contained in:
Konstantin Grachev
2021-02-15 22:16:27 +03:00
parent ce9d1824d9
commit bbbc9da3e0
2 changed files with 16 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/Messenger/NsqReceivedStamp.php
View File

@ -4,7 +4,9 @@ declare(strict_types=1);
namespace Nsq\NsqBundle\Messenger;
use LogicException;
use Nsq\Message;
use Symfony\Component\Messenger\Envelope;
use Symfony\Component\Messenger\Stamp\StampInterface;
/**
@ -18,4 +20,16 @@ final class NsqReceivedStamp implements StampInterface
{
$this->message = $message;
}
public static function getMessageFromEnvelope(Envelope $envelope): Message
{
/** @var self|null $stamp */
$stamp = $envelope->last(self::class);
if (null === $stamp) {
throw new LogicException('Envelop doesn\'t related to NsqMessage.');
}
return $stamp->message;
}
}

21
src/Messenger/NsqTransport.php
View File

@ -6,7 +6,6 @@ namespace Nsq\NsqBundle\Messenger;
use Generator;
use JsonException;
use LogicException;
use Nsq\Message;
use Nsq\Producer;
use Nsq\Subscriber;
@ -114,10 +113,7 @@ final class NsqTransport implements TransportInterface
*/
public function ack(Envelope $envelope): void
{
$message = $this->getMessage($envelope);
if (!$message instanceof Message) {
throw new LogicException('Returned envelop doesn\'t related to NsqMessage.');
}
$message = NsqReceivedStamp::getMessageFromEnvelope($envelope);
$message->finish();
}
@ -127,21 +123,8 @@ final class NsqTransport implements TransportInterface
*/
public function reject(Envelope $envelope): void
{
$message = $this->getMessage($envelope);
if (!$message instanceof Message) {
throw new LogicException('Returned envelop doesn\'t related to NsqMessage.');
}
$message = NsqReceivedStamp::getMessageFromEnvelope($envelope);
$message->finish();
}
private function getMessage(Envelope $envelope): ?Message
{
$stamp = $envelope->last(NsqReceivedStamp::class);
if (!$stamp instanceof NsqReceivedStamp) {
return null;
}
return $stamp->message;
}
}