Collection.php

Go to the documentation of this file.

<?php
namespace Magento\Customer\Model\ResourceModel\Address;

class Collection extends \Magento\Eav\Model\Entity\Collection\VersionControl\AbstractCollection
{
    protected function _construct()
    {
        $this->_init(\Magento\Customer\Model\Address::class, \Magento\Customer\Model\ResourceModel\Address::class);
    }

    public function setCustomerFilter($customer)
    {
        if (is_array($customer)) {
            $this->addAttributeToFilter('parent_id', ['in' => $customer]);
        } elseif ($customer->getId()) {
            $this->addAttributeToFilter('parent_id', $customer->getId());
        } else {
            $this->addAttributeToFilter('parent_id', '-1');
        }
        return $this;
    }
}