Behavioral\UnitOfWork

Complexity

Total Complexity

12

Size/Duplication

Total Lines	62
Duplicated Lines	0 %

Test Coverage

Coverage

38.71%

Importance

Changes	1
Bugs	0	Features	0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	registerDataMapper()	0	3	1
A	getDataMapper()	0	8	2
A	__construct()	0	6	1
A	registerNew()	0	3	1
A	registerRemoved()	0	11	3
A	registerDirty()	0	7	4

<?php

declare(strict_types = 1);

namespace Behavioral;

use SplObjectStorage;
use DataSource\Connection;
use DataSource\DataMapper\AbstractMapper;
use BasePatterns\LayerSupertype\DomainObject;

class UnitOfWork
{
    private SplObjectStorage $newObjects;

    private SplObjectStorage $dirtyObjects;

    private SplObjectStorage $removedObjects;

    private array $mappers = [];

    private Connection $connection;

    public function __construct(Connection $connection)
    {
        $this->newObjects = new SplObjectStorage();
        $this->dirtyObjects = new SplObjectStorage();
        $this->removedObjects = new SplObjectStorage();
        $this->connection = $connection;
    }

    public function registerNew(DomainObject $object): void
    {
        $this->newObjects->attach($object);
    }

    public function registerDirty(DomainObject $object): void
    {
        if (!$this->removedObjects->contains($object)
            && !$this->dirtyObjects->contains($object)
            && !$this->newObjects->contains($object)
        ) {
            $this->dirtyObjects->attach($object);
        }
    }

    public function registerRemoved(DomainObject $object): void
    {
        if ($this->newObjects->contains($object)) {
            $this->newObjects->detach($object);

            return;
        }

        $this->dirtyObjects->detach($object);
        if (!$this->removedObjects->contains($object)) {
            $this->removedObjects->attach($object);
        }
    }

    public function registerDataMapper(string $className, string $mapperClassName): void
    {
        $this->mappers[$className] = new $mapperClassName($this->connection);
    }

    public function getDataMapper(string $className): AbstractMapper
    {
        if (\array_key_exists($className, $this->mappers)) {
            return $this->mappers[$className];
        }

        throw new \DomainException(
            \sprintf("Unable to find mapper for class `%s`", $className)
        );
    }
}