Server : Apache System : Linux server2.corals.io 4.18.0-348.2.1.el8_5.x86_64 #1 SMP Mon Nov 15 09:17:08 EST 2021 x86_64 User : corals ( 1002) PHP Version : 7.4.33 Disable Function : exec,passthru,shell_exec,system Directory :  /home/corals/old/vendor/magento/module-inventory/Model/ResourceModel/

Upload File : current_dir [ Writeable ] document_root [ Writeable ]

<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\Inventory\Model\ResourceModel;

use Exception;
use Magento\Framework\Model\AbstractModel;
use Magento\Framework\Model\ResourceModel\Db\AbstractDb;
use Magento\Framework\Model\ResourceModel\Db\Context;
use Magento\Framework\Model\ResourceModel\PredefinedId;
use Magento\InventoryApi\Api\Data\SourceInterface;
use Magento\InventoryApi\Model\SourceCarrierLinkManagementInterface;

/**
 * Implementation of basic operations for Source entity for specific db layer
 */
class Source extends AbstractDb
{
    /**
     * Provides possibility of saving entity with predefined/pre-generated id
     */
    use PredefinedId;

    /**#@+
     * Constants related to specific db layer
     */
    const TABLE_NAME_SOURCE = 'inventory_source';
    /**#@-*/

    /**
     * Primary key auto increment flag
     *
     * @var bool
     */
    protected $_isPkAutoIncrement = false;

    /**
     * @var SourceCarrierLinkManagementInterface
     */
    private $sourceCarrierLinkManagement;

    /**
     * @param Context $context
     * @param SourceCarrierLinkManagementInterface $sourceCarrierLinkManagement
     * @param null $connectionName
     */
    public function __construct(
        Context $context,
        SourceCarrierLinkManagementInterface $sourceCarrierLinkManagement,
        $connectionName = null
    ) {
        parent::__construct($context, $connectionName);
        $this->sourceCarrierLinkManagement = $sourceCarrierLinkManagement;
    }

    /**
     * @inheritdoc
     */
    protected function _construct()
    {
        $this->_init(self::TABLE_NAME_SOURCE, SourceInterface::SOURCE_CODE);
        $this->addUniqueField(
            [
                'field' => SourceInterface::NAME,
                'title' => 'Name'
            ]
        );
    }

    /**
     * @inheritdoc
     */
    public function load(AbstractModel $object, $value, $field = null)
    {
        parent::load($object, $value, $field);
        /** @var SourceInterface $object */
        $this->sourceCarrierLinkManagement->loadCarrierLinksBySource($object);
        return $this;
    }

    /**
     * @inheritdoc
     */
    public function save(AbstractModel $object)
    {
        $connection = $this->getConnection();
        $connection->beginTransaction();
        try {
            $object->isObjectNew(!$this->isObjectNotNew($object));
            parent::save($object);
            /** @var SourceInterface $object */
            $this->sourceCarrierLinkManagement->saveCarrierLinksBySource($object);
            $connection->commit();
        } catch (Exception $e) {
            $connection->rollBack();
            throw $e;
        }
        return $this;
    }
}