queue/src/Middleware/Push/Implementation/IdMiddleware.php at 221018ea2d8be4f35b11961b3d1e7caa3162cca2 · yiisoft/queue · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
<?php

declare(strict_types=1);

namespace Yiisoft\Queue\Middleware\Push\Implementation;

use Yiisoft\Queue\Message\IdEnvelope;
use Yiisoft\Queue\Message\MessageInterface;
use Yiisoft\Queue\Middleware\Push\MessageHandlerPushInterface;
use Yiisoft\Queue\Middleware\Push\MiddlewarePushInterface;

/**
 * A middleware for message ID setting.
 */
final class IdMiddleware implements MiddlewarePushInterface
{
    public function processPush(MessageInterface $message, MessageHandlerPushInterface $handler): MessageInterface
    {
        $meta = $message->getMetadata();
        if (empty($meta[IdEnvelope::MESSAGE_ID_KEY])) {
            $message = new IdEnvelope($message, uniqid('yii3-message-', true));
        }

        return $handler->handlePush($message);
    }
}