Completed
Pull Request — 0.4 (#35)
by jean
12:50
created

AmqpMessagePublisher::buildMessage()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 3
Code Lines 1

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 1
eloc 1
nc 1
nop 1
dl 0
loc 3
rs 10
c 0
b 0
f 0
1
<?php
2
3
declare(strict_types=1);
4
5
namespace Darkilliant\MqProcessBundle\Message\Publisher;
6
7
use Darkilliant\MqProcessBundle\Message\Message;
8
use PhpAmqpLib\Channel\AMQPChannel;
9
use PhpAmqpLib\Message\AMQPMessage;
10
11
class AmqpMessagePublisher implements MessagePublisherInterface
12
{
13
    /** @var AMQPChannel */
14
    private $channel;
15
    /** @var string */
16
    private $queue;
17
    /** @var string */
18
    private $exchange;
19
    /** @var int */
20
    private $batch = 0;
21
    /** @var bool */
22
    private $persistant;
23
24
    public function __construct(AMQPChannel $channel, string $queue, string $exchange, bool $persistant)
25
    {
26
        $this->channel = $channel;
27
        $this->queue = $queue;
28
        $this->exchange = $exchange;
29
        $this->persistant = $persistant;
30
31
        $this->channel->queue_declare($this->queue, false, $this->persistant, false, false);
32
        $this->channel->queue_bind($this->queue, $this->exchange, $this->queue);
33
    }
34
35
    public function publish(Message $message, int $batchCount)
36
    {
37
        ++$this->batch;
38
        $this->channel->batch_basic_publish($this->buildMessage($message), $this->exchange, $this->queue);
39
40
        if ($this->batch >= $batchCount) {
41
            $this->flush();
42
        }
43
    }
44
45
    private function buildMessage(Message $message)
46
    {
47
        return new AMQPMessage($message->getBody());
48
    }
49
50
    public function finalize()
51
    {
52
        $this->flush();
53
    }
54
55
    public function flush()
56
    {
57
        $this->channel->publish_batch();
58
    }
59
60
    public function countMessages(): int
61
    {
62
        return $this->channel->queue_declare($this->queue, true)[1];
63
    }
64
}
65