DeleteRelation.php

Go to the documentation of this file.

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

use Magento\Customer\Api\Data\CustomerInterface;

class DeleteRelation
{
    public function deleteRelation(
        \Magento\Framework\Model\AbstractModel $address,
        \Magento\Customer\Model\Customer $customer
    ) {
        $toUpdate = $this->getDataToUpdate($address, $customer);

        if (!$address->getIsCustomerSaveTransaction() && !empty($toUpdate)) {
            $address->getResource()->getConnection()->update(
                $address->getResource()->getTable('customer_entity'),
                $toUpdate,
                $address->getResource()->getConnection()->quoteInto('entity_id = ?', $customer->getId())
            );
        }
    }

    private function getDataToUpdate(
        \Magento\Framework\Model\AbstractModel $address,
        \Magento\Customer\Model\Customer $customer
    ) {
        $toUpdate = [];
        if ($address->getId()) {
            if ($customer->getDefaultBilling() == $address->getId()) {
                $toUpdate[CustomerInterface::DEFAULT_BILLING] = null;
            }

            if ($customer->getDefaultShipping() == $address->getId()) {
                $toUpdate[CustomerInterface::DEFAULT_SHIPPING] = null;
            }
        }

        return $toUpdate;
    }
}