Adding a messenger middleware and stamp.

Author: floriankraemer

src/Messenger/Middleware/CorrelationIdMiddleware.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Phauthentic\CorrelationIdBundle\Messenger\Middleware;
+
+use Phauthentic\CorrelationIdBundle\Messenger\Stamp\CorrelationIdStamp;
+use Phauthentic\Infrastructure\Utils\CorrelationID;
+use Symfony\Component\Messenger\Envelope;
+use Symfony\Component\Messenger\Middleware\MiddlewareInterface;
+use Symfony\Component\Messenger\Middleware\StackInterface;
+
+/**
+ *
+ */
+class CorrelationIdMiddleware implements MiddlewareInterface
+{
+    /**
+     * @throws \Exception
+     */
+    public function handle(Envelope $envelope, StackInterface $stack): Envelope
+    {
+        if (isset($envelope->all()[CorrelationIdStamp::class])) {
+            return $stack->next()->handle($envelope, $stack);
+        }
+
+        $envelope = $envelope->with(
+            new CorrelationIdStamp(CorrelationID::toString())
+        );
+
+        return $stack->next()->handle($envelope, $stack);
+    }
+}

src/Messenger/Stamp/CorrelationIdStamp.php

@@ -0,0 +1,25 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Phauthentic\CorrelationIdBundle\Messenger\Stamp;
+
+use Symfony\Component\Messenger\Stamp\StampInterface;
+
+/**
+ *
+ */
+class CorrelationIdStamp implements StampInterface
+{
+    private string $correlationId;
+
+    public function __construct(string $correlationId)
+    {
+        $this->correlationId = $correlationId;
+    }
+
+    public function getCorrelationId(): string
+    {
+        return $this->correlationId;
+    }
+}

tests/Messenger/Middleware/CorrelationIdMiddlewareTest.php

@@ -0,0 +1,29 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Phauthentic\CorrelationIdBundle\Tests\Messenger\Middleware;
+
+use Phauthentic\CorrelationIdBundle\Messenger\Middleware\CorrelationIdMiddleware;
+use Phauthentic\CorrelationIdBundle\Messenger\Stamp\CorrelationIdStamp;
+use PHPUnit\Framework\TestCase;
+use stdClass;
+use Symfony\Component\Messenger\Envelope;
+use Symfony\Component\Messenger\Middleware\StackMiddleware;
+
+/**
+ *
+ */
+class CorrelationIdMiddlewareTest extends TestCase
+{
+    public function testHandle(): void
+    {
+        $middleware = new CorrelationIdMiddleware();
+        $envelope = new Envelope(new stdClass());
+        $stack = new StackMiddleware();
+
+        $envelope = $middleware->handle($envelope, $stack);
+
+        $this->assertArrayHasKey(CorrelationIdStamp::class, $envelope->all());
+    }
+}

tests/Messenger/Stamp/CorrelationIdStampTest.php

@@ -0,0 +1,20 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Phauthentic\CorrelationIdBundle\Tests\Messenger\Stamp;
+
+use Phauthentic\CorrelationIdBundle\Messenger\Stamp\CorrelationIdStamp;
+use PHPUnit\Framework\TestCase;
+
+/**
+ *
+ */
+class CorrelationIdStampTest extends TestCase
+{
+    public function testGetCorrelationId(): void
+    {
+        $stamp = new CorrelationIdStamp('foo');
+        $this->assertEquals('foo', $stamp->getCorrelationId());
+    }
+}