[consumption] Add onResult extension point.

makasim · makasim · commit da4dfcdc6fd9 · 2017-04-06T18:38:15.000+03:00
diff --git a/pkg/enqueue/Consumption/EmptyExtensionTrait.php b/pkg/enqueue/Consumption/EmptyExtensionTrait.php
@@ -25,6 +25,13 @@ public function onPreReceived(Context $context)
     {
     }
 
+    /**
+     * @param Context $context
+     */
+    public function onResult(Context $context)
+    {
+    }
+
     /**
      * @param Context $context
      */
diff --git a/pkg/enqueue/Consumption/ExtensionInterface.php b/pkg/enqueue/Consumption/ExtensionInterface.php
@@ -31,6 +31,15 @@ public function onBeforeReceive(Context $context);
      */
     public function onPreReceived(Context $context);
 
+    /**
+     * Executed when a message is processed by a processor or a result was set in onPreReceived method.
+     * BUT before the message status was sent to the broker
+     * The consumption could be interrupted at this step but it exits after the message is processed.
+     *
+     * @param Context $context
+     */
+    public function onResult(Context $context);
+
     /**
      * Executed when a message is processed by a processor.
      * The context contains a status, which could not be changed.
diff --git a/pkg/enqueue/Consumption/QueueConsumer.php b/pkg/enqueue/Consumption/QueueConsumer.php
@@ -204,6 +204,8 @@ protected function doConsume(ExtensionInterface $extension, Context $context)
                 $context->setResult($result);
             }
 
+            $extension->onResult($context);
+
             switch ($context->getResult()) {
                 case Result::ACK:
                     $consumer->acknowledge($message);

Original file line number	Diff line number	Diff line change
`@@ -25,6 +25,13 @@ public function onPreReceived(Context $context)`
`25`	`25`	`{`
`26`	`26`	`}`
`27`	`27`
	`28`	`+ /**`
	`29`	`+ * @param Context $context`
	`30`	`+ */`
	`31`	`+ public function onResult(Context $context)`
	`32`	`+ {`
	`33`	`+ }`
	`34`	`+`
`28`	`35`	`/**`
`29`	`36`	`* @param Context $context`
`30`	`37`	`*/`
Original file line number	Diff line number	Diff line change
`@@ -204,6 +204,8 @@ protected function doConsume(ExtensionInterface $extension, Context $context)`
`204`	`204`	`$context->setResult($result);`
`205`	`205`	`}`
`206`	`206`
	`207`	`+ $extension->onResult($context);`
	`208`	`+`
`207`	`209`	`switch ($context->getResult()) {`
`208`	`210`	`case Result::ACK:`
`209`	`211`	`$consumer->acknowledge($message);`