doctrine / skeleton-mapper

lib/Doctrine/SkeletonMapper/ObjectIdentityMap.php

<?php

declare(strict_types=1);

namespace Doctrine\SkeletonMapper;

use Doctrine\SkeletonMapper\ObjectRepository\ObjectRepositoryFactoryInterface;
use function count;
use function get_class;
use function serialize;

/**
 * Class for maintaining an object identity map.
 */
class ObjectIdentityMap
{
    /** @var object[][] */
    private $identityMap = [];

    /** @var ObjectRepositoryFactoryInterface */
    private $objectRepositoryFactory;

    public function __construct(ObjectRepositoryFactoryInterface $objectRepositoryFactory)
    {
        $this->objectRepositoryFactory = $objectRepositoryFactory;
    }

    /**
     * @param object $object
     */
    public function contains($object) : bool
    {
        $className = get_class($object);

        $objectIdentifier = $this->getObjectIdentifier($object);

        $serialized = serialize($objectIdentifier);

        return isset($this->identityMap[$className][$serialized]);
    }

    /**
     * @param mixed[] $data
     *
     * @return object|null
     */
    public function tryGetById(string $className, array $data)
    {
        $serialized = serialize($this->extractIdentifierFromData($className, $data));

        if (isset($this->identityMap[$className][$serialized])) {
            return $this->identityMap[$className][$serialized];
        }

        return null;
    }

    /**
     * @param object  $object
     * @param mixed[] $data
     */
    public function addToIdentityMap($object, array $data) : void

The parameter $data is not used and could be removed.

    {
        $className = get_class($object);

        if (! isset($this->identityMap[$className])) {
            $this->identityMap[get_class($object)] = [];
        }

        $serialized = serialize($this->getObjectIdentifier($object));

        $this->identityMap[get_class($object)][$serialized] = $object;
    }

    public function clear(?string $objectName = null) : void
    {
        if ($objectName !== null) {
            unset($this->identityMap[$objectName]);
        } else {
            $this->identityMap = [];
        }
    }

    /**
     * @param object $object
     */
    public function detach($object) : void
    {
        $objectIdentifier = $this->getObjectIdentifier($object);

        $serialized = serialize($objectIdentifier);
        unset($this->identityMap[get_class($object)][$serialized]);
    }

    public function count() : int
    {
        return count($this->identityMap);
    }

    /**
     * @param object $object
     *
     * @return mixed[] $identifier
     */
    private function getObjectIdentifier($object) : array
    {
        return $this->objectRepositoryFactory
            ->getRepository(get_class($object))
            ->getObjectIdentifier($object);
    }

    /**
     * @param mixed[] $data
     *
     * @return mixed[] $identifier
     */
    private function extractIdentifierFromData(string $className, array $data) : array
    {
        return $this->objectRepositoryFactory
            ->getRepository($className)
            ->getObjectIdentifierFromData($data);
    }
}