Werkspot\MessageBus\Bus\DeliveryChain\Middleware\OrmTransactionMiddleware

Complexity

Total Complexity

4

Size/Duplication

Total Lines	28
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	3	1
A	deliver()	0	16	3

<?php

declare(strict_types=1);

namespace Werkspot\MessageBus\Bus\DeliveryChain\Middleware;

use Doctrine\ORM\EntityManagerInterface;
use Exception;
use Werkspot\MessageBus\Bus\DeliveryChain\MiddlewareInterface;
use Werkspot\MessageBus\Message\MessageInterface;

final class OrmTransactionMiddleware implements MiddlewareInterface
{
    /**
     * @var EntityManagerInterface
     */
    private $entityManager;

    public function __construct(EntityManagerInterface $entityManager)
    {
        $this->entityManager = $entityManager;
    }

    public function deliver(MessageInterface $message, callable $next): void
    {
        if ($this->entityManager->getConnection()->isTransactionActive()) {
            $next($message);

            return;
        }

        try {
            $this->entityManager->beginTransaction();
            $next($message);
            $this->entityManager->flush();
            $this->entityManager->commit();
        } catch (Exception $exception) {
            $this->entityManager->rollback();
            throw $exception;
        }
    }
}