identitymap.php

<?php
namespace Bitrix\Main\ORM\Objectify;
 
use Bitrix\Main\ORM\Entity;
 
class IdentityMap
{
    protected $collections;
 
    public function get($class, $primary)
    {
        $collection = $this->getCollectionByClass($class);
        return $collection->getByPrimary($primary);
    }
 
    public function put($object)
    {
        $collection = $this->getCollectionByClass(get_class($object));
        $collection[] = $object;
 
        return $collection;
    }
 
    public function getCollectionByClass($class)
    {
        if (empty($this->collections[$class]))
        {
            $normalizedClass = Entity::normalizeName($class);
 
            if (!empty($this->collections[$normalizedClass]))
            {
                $this->collections[$class] = $this->collections[$normalizedClass];
            }
            else
            {
                if (Entity::has($normalizedClass))
                {
                    $entity = Entity::get($normalizedClass);
                }
                else
                {
                    $entity = Entity::getInstance($normalizedClass::$dataClass);
                }
 
                $collection = $entity->createCollection();
 
                $this->collections[$class] = $collection;
                $this->collections[$normalizedClass] = $collection;
            }
        }
 
        return $this->collections[$class];
    }
}