Dispatch events around publishing

Event/AMQPEvent.php

@@ -3,6 +3,7 @@
 namespace OldSound\RabbitMqBundle\Event;
 
 use OldSound\RabbitMqBundle\RabbitMq\Consumer;
+use OldSound\RabbitMqBundle\RabbitMq\Producer;
 use PhpAmqpLib\Message\AMQPMessage;
 use Symfony\Component\EventDispatcher\Event;
 
@@ -18,6 +19,8 @@ class AMQPEvent extends Event
     const ON_IDLE                   = 'on_idle';
     const BEFORE_PROCESSING_MESSAGE = 'before_processing';
     const AFTER_PROCESSING_MESSAGE  = 'after_processing';
+    const BEFORE_PUBLISHING_MESSAGE = 'before_publishing';
+    const AFTER_PUBLISHING_MESSAGE  = 'after_publishing';
 
     /**
      * @var AMQPMessage
@@ -29,6 +32,11 @@ class AMQPEvent extends Event
      */
     protected $consumer;
 
+    /**
+     * @var Producer
+     */
+    protected $producer;
+
     /**
      * @return AMQPMessage
      */
@@ -68,4 +76,24 @@ public function setConsumer(Consumer $consumer)
 
         return $this;
     }
+
+    /**
+     * @return Producer
+     */
+    public function getProducer()
+    {
+        return $this->producer;
+    }
+
+    /**
+     * @param Producer $producer
+     *
+     * @return AMQPEvent
+     */
+    public function setProducer(Producer $producer)
+    {
+        $this->producer = $producer;
+
+        return $this;
+    }
 }

Event/AfterPublishingMessageEvent.php

@@ -0,0 +1,28 @@
+<?php
+
+namespace OldSound\RabbitMqBundle\Event;
+
+use OldSound\RabbitMqBundle\RabbitMq\Producer;
+use PhpAmqpLib\Message\AMQPMessage;
+
+/**
+ * Class AfterPublishingMessageEvent
+ *
+ * @package OldSound\RabbitMqBundle\Command
+ */
+class AfterPublishingMessageEvent extends AMQPEvent
+{
+    const NAME = AMQPEvent::AFTER_PUBLISHING_MESSAGE;
+
+    /**
+     * AfterPublishingMessageEvent constructor.
+     *
+     * @param Producer $producer
+     * @param AMQPMessage $AMQPMessage
+     */
+    public function __construct(Producer $producer, AMQPMessage $AMQPMessage)
+    {
+        $this->setProducer($producer);
+        $this->setAMQPMessage($AMQPMessage);
+    }
+}

Event/BeforePublishingMessageEvent.php

@@ -0,0 +1,28 @@
+<?php
+
+namespace OldSound\RabbitMqBundle\Event;
+
+use OldSound\RabbitMqBundle\RabbitMq\Producer;
+use PhpAmqpLib\Message\AMQPMessage;
+
+/**
+ * Class BeforePublishingMessageEvent
+ *
+ * @package OldSound\RabbitMqBundle\Command
+ */
+class BeforePublishingMessageEvent extends AMQPEvent
+{
+    const NAME = AMQPEvent::BEFORE_PUBLISHING_MESSAGE;
+
+    /**
+     * BeforePublishingMessageEvent constructor.
+     *
+     * @param Producer $producer
+     * @param AMQPMessage $AMQPMessage
+     */
+    public function __construct(Producer $producer, AMQPMessage $AMQPMessage)
+    {
+        $this->setProducer($producer);
+        $this->setAMQPMessage($AMQPMessage);
+    }
+}

README.md

@@ -261,6 +261,10 @@ If you need to use a custom class for a producer (which should inherit from `Old
     ...
 ```
 
+#### Producer Events ####
+
+TODO
+
 
 The next piece of the puzzle is to have a consumer that will take the message out of the queue and process it accordingly.
 

RabbitMq/Producer.php

@@ -2,6 +2,8 @@
 
 namespace OldSound\RabbitMqBundle\RabbitMq;
 
+use OldSound\RabbitMqBundle\Event\AfterPublishingMessageEvent;
+use OldSound\RabbitMqBundle\Event\BeforePublishingMessageEvent;
 use PhpAmqpLib\Message\AMQPMessage;
 use PhpAmqpLib\Wire\AMQPTable;
 
@@ -53,6 +55,7 @@ public function publish($msgBody, $routingKey = '', $additionalProperties = arra
             $msg->set('application_headers', $headersTable);
         }
 
+        $this->dispatchEvent(BeforePublishingMessageEvent::NAME, new BeforePublishingMessageEvent($this, $msg));
         $this->getChannel()->basic_publish($msg, $this->exchangeOptions['name'], (string)$routingKey);
         $this->logger->debug('AMQP message published', array(
             'amqp' => array(
@@ -62,5 +65,6 @@ public function publish($msgBody, $routingKey = '', $additionalProperties = arra
                 'headers' => $headers
             )
         ));
+        $this->dispatchEvent(AfterPublishingMessageEvent::NAME, new AfterPublishingMessageEvent($this, $msg));
     }
 }