Locking

locking.php

<?php declare(strict_types=1);
final class DoctrineSagaPersistence implements SagaPersistence
{
    private ManagerRegistry $mr;
    private SagaFinding $sagaFinding;

    public function __construct(ManagerRegistry $mr, SagaFinding $sagaFinding)
    {
        $this->mr          = $mr;
        $this->sagaFinding = $sagaFinding;
    }

    public function retrieveSaga(DomainMessage $domainMessage, string $sagaType): Saga
    {
        /** @var object $message */
        $message = $domainMessage->getPayload();
        /** @var EntityManagerInterface $em */
        $em      = $this->mr->getManager();
        $em->beginTransaction();
        foreach ($this->sagaFinding->sagasForMessage($message) as $finderDefinition) {
            if ($finderDefinition->sagaType !== $sagaType) {
                continue;
            }
            /** @var Id<Saga> $id */
            $id = ($finderDefinition->messageProperty)($message, $domainMessage);
            /** @psalm-suppress TooManyArguments */
            $saga = $em->find(
                $sagaType,
                $id,
                LockMode::PESSIMISTIC_WRITE
            );
            if ($saga === null) {
                $saga = new $finderDefinition->sagaType($id);
                $em->persist($saga);
            }

            return $saga;
        }

        throw new \Exception("Did not find saga $sagaType for message ".get_class($message));
    }

    public function saveSaga(Saga $saga): void
    {
        /** @var EntityManager $em */
        $em = $this->mr->getManager();
        $em->getUnitOfWork()->commit($saga);
        $em->commit();
    }

    public function cleanSaga(Saga $saga): void
    {
        /** @var EntityManager $em */
        $em = $this->mr->getManager();
        $em->rollback();
        $this->mr->resetManager();
    }
}