[Messenger] Add AMQP adapter

Author: sroze

Adapter/AmqpExt/AmqpAdapterFactory.php

@@ -0,0 +1,50 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Component\Messenger\Adapter\AmqpExt;
+
+use Symfony\Component\Messenger\Adapter\Factory\AdapterFactoryInterface;
+use Symfony\Component\Messenger\Transport\ReceiverInterface;
+use Symfony\Component\Messenger\Transport\SenderInterface;
+use Symfony\Component\Messenger\Transport\Serialization\DecoderInterface;
+use Symfony\Component\Messenger\Transport\Serialization\EncoderInterface;
+
+/**
+ * @author Samuel Roze <[email protected]>
+ */
+class AmqpAdapterFactory implements AdapterFactoryInterface
+{
+    private $encoder;
+    private $decoder;
+    private $debug;
+
+    public function __construct(EncoderInterface $encoder, DecoderInterface $decoder, bool $debug)
+    {
+        $this->encoder = $encoder;
+        $this->decoder = $decoder;
+        $this->debug = $debug;
+    }
+
+    public function createReceiver(string $dsn, array $options): ReceiverInterface
+    {
+        return new AmqpReceiver($this->decoder, Connection::fromDsn($dsn, $options, $this->debug));
+    }
+
+    public function createSender(string $dsn, array $options): SenderInterface
+    {
+        return new AmqpSender($this->encoder, Connection::fromDsn($dsn, $options, $this->debug));
+    }
+
+    public function supports(string $dsn, array $options): bool
+    {
+        return 0 === strpos($dsn, 'amqp://');
+    }
+}

Adapter/AmqpExt/AmqpFactory.php

@@ -0,0 +1,35 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Component\Messenger\Adapter\AmqpExt;
+
+class AmqpFactory
+{
+    public function createConnection(array $credentials): \AMQPConnection
+    {
+        return new \AMQPConnection($credentials);
+    }
+
+    public function createChannel(\AMQPConnection $connection): \AMQPChannel
+    {
+        return new \AMQPChannel($connection);
+    }
+
+    public function createQueue(\AMQPChannel $channel): \AMQPQueue
+    {
+        return new \AMQPQueue($channel);
+    }
+
+    public function createExchange(\AMQPChannel $channel): \AMQPExchange
+    {
+        return new \AMQPExchange($channel);
+    }
+}

Adapter/AmqpExt/AmqpReceiver.php

@@ -0,0 +1,80 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Component\Messenger\Adapter\AmqpExt;
+
+use Symfony\Component\Messenger\Adapter\AmqpExt\Exception\RejectMessageExceptionInterface;
+use Symfony\Component\Messenger\Transport\ReceiverInterface;
+use Symfony\Component\Messenger\Transport\Serialization\DecoderInterface;
+
+/**
+ * Symfony Messenger receiver to get messages from AMQP brokers using PHP's AMQP extension.
+ *
+ * @author Samuel Roze <[email protected]>
+ */
+class AmqpReceiver implements ReceiverInterface
+{
+    private $messageDecoder;
+    private $connection;
+    private $shouldStop;
+
+    public function __construct(DecoderInterface $messageDecoder, Connection $connection)
+    {
+        $this->messageDecoder = $messageDecoder;
+        $this->connection = $connection;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function receive(callable $handler): void
+    {
+        while (!$this->shouldStop) {
+            $message = $this->connection->get();
+            if (null === $message) {
+                $handler(null);
+
+                usleep($this->connection->getConnectionCredentials()['loop_sleep'] ?? 200000);
+                if (function_exists('pcntl_signal_dispatch')) {
+                    pcntl_signal_dispatch();
+                }
+
+                continue;
+            }
+
+            try {
+                $handler($this->messageDecoder->decode(array(
+                    'body' => $message->getBody(),
+                    'headers' => $message->getHeaders(),
+                )));
+
+                $this->connection->ack($message);
+            } catch (RejectMessageExceptionInterface $e) {
+                $this->connection->reject($message);
+
+                throw $e;
+            } catch (\Throwable $e) {
+                $this->connection->nack($message, AMQP_REQUEUE);
+
+                throw $e;
+            } finally {
+                if (function_exists('pcntl_signal_dispatch')) {
+                    pcntl_signal_dispatch();
+                }
+            }
+        }
+    }
+
+    public function stop(): void
+    {
+        $this->shouldStop = true;
+    }
+}

Adapter/AmqpExt/AmqpSender.php

@@ -0,0 +1,42 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Component\Messenger\Adapter\AmqpExt;
+
+use Symfony\Component\Messenger\Transport\SenderInterface;
+use Symfony\Component\Messenger\Transport\Serialization\EncoderInterface;
+
+/**
+ * Symfony Messenger sender to send messages to AMQP brokers using PHP's AMQP extension.
+ *
+ * @author Samuel Roze <[email protected]>
+ */
+class AmqpSender implements SenderInterface
+{
+    private $messageEncoder;
+    private $connection;
+
+    public function __construct(EncoderInterface $messageEncoder, Connection $connection)
+    {
+        $this->messageEncoder = $messageEncoder;
+        $this->connection = $connection;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function send($message)
+    {
+        $encodedMessage = $this->messageEncoder->encode($message);
+
+        $this->connection->publish($encodedMessage['body'], $encodedMessage['headers']);
+    }
+}