swarrot / swarrot

src/Swarrot/Broker/MessageProvider/StatefulStompMessageProvider.php

<?php

namespace Swarrot\Broker\MessageProvider;

use Stomp\Client;
use Stomp\StatefulStomp;
use Stomp\Transport\Message as StompMessage;
use Swarrot\Broker\Message;

class StatefulStompMessageProvider implements MessageProviderInterface
{
    /**
     * @var Client
     */
    private $client;

    /**
     * @var StatefulStomp
     */
    private $stomp;

    /**
     * @var string
     */
    private $destination;

    /**
     * @param string $destination
     * @param null   $selector
     * @param string $ack
     */
    public function __construct(
        Client $client,
        $destination,
        $selector = null,
        $ack = 'client',
        array $header = []
    ) {
        @trigger_error(sprintf('"%s" have been deprecated since Swarrot 3.6', __CLASS__), E_USER_DEPRECATED);

It seems like you do not handle an error condition here. This can introduce security issues, and is generally not recommended.

        $this->client = $client;
        $this->destination = $destination;

        $this->stomp = new StatefulStomp($client);
        $this->stomp->subscribe($destination, $selector, $ack, $header);
    }

    public function get()
    {
        if ($frame = $this->stomp->read()) {
            return new Message($frame->getBody(), $frame->getHeaders());
        }

        return null;
    }

    public function ack(Message $message)
    {
        $this->stomp->ack(new StompMessage($message->getBody(), $message->getProperties()));
    }

    /**
     * @param bool $requeue
     */
    public function nack(Message $message, $requeue = false)
    {
        $this->stomp->nack(new StompMessage($message->getBody(), $message->getProperties()), $requeue);
    }

    /**
     * @return string
     */
    public function getQueueName()
    {
        return $this->destination;
    }
}