addresscollection.php

<?php
 
namespace Bitrix\Location\Entity\Address;
 
use Bitrix\Location\Entity\Address;
use Bitrix\Main\ArgumentTypeException;
 
final class AddressCollection extends \Bitrix\Location\Entity\Generic\Collection
{
    protected $items = [];
 
    public function addItem($address): int
    {
        if(!($address instanceof Address))
        {
            throw new ArgumentTypeException('address must be the instance of Address');
        }
 
        return parent::addItem($address);
    }
 
    public function getAddressById(int $addressId): ?Address
    {
        foreach ($this->items as $item)
        {
            if ($item->getId() === $addressId)
            {
                return $item;
            }
        }
 
        return null;
    }
}