Dsantang\DomainEventsDoctrine\EventsRecorder\OrderedDoctrineEventsRecorder

Complexity

Total Complexity

4

Size/Duplication

Total Lines	40
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	getEventAwareEntities()	0	10	1
A	onFlush()	0	15	2
A	__construct()	0	3	1

<?php

declare(strict_types=1);

namespace Dsantang\DomainEventsDoctrine\EventsRecorder;

use Doctrine\ORM\Event\OnFlushEventArgs;
use Doctrine\ORM\UnitOfWork;
use Dsantang\DomainEvents\Counter;
use Dsantang\DomainEvents\EventAware;
use Dsantang\DomainEventsDoctrine\Aggregator;
use function array_filter;
use function array_merge;
use function ksort;

final class OrderedDoctrineEventsRecorder
{
    /** @var Aggregator */
    private $eventsAggregator;

    public function __construct(Aggregator $eventAggregator)
    {
        $this->eventsAggregator = $eventAggregator;
    }

    public function onFlush(OnFlushEventArgs $eventArgs) : void
    {
        $unitOfWork = $eventArgs->getEntityManager()
                                ->getUnitOfWork();

        $events = [];

        foreach (self::getEventAwareEntities($unitOfWork) as $entity) {
            $events += $entity->expelRecordedEvents();
        }

        ksort($events);

        Counter::reset();
        $this->eventsAggregator->aggregate(...$events);
    }

    /**
     * @return EventAware[]
     */
    private static function getEventAwareEntities(UnitOfWork $unitOfWork) : array
    {
        $entities = array_merge(
            $unitOfWork->getScheduledEntityInsertions(),
            $unitOfWork->getScheduledEntityUpdates(),
            $unitOfWork->getScheduledEntityDeletions()
        );

        return array_filter($entities, static function ($entity) {
            return $entity instanceof EventAware;
        });
    }
}