Source.php

Go to the documentation of this file.

<?php
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\InventoryApi\Model\SourceCarrierLinkManagementInterface;
use Magento\InventoryApi\Api\Data\SourceInterface;
use Magento\Framework\Model\ResourceModel\PredefinedId;

class Source extends AbstractDb
{
    use PredefinedId;

    const TABLE_NAME_SOURCE = 'inventory_source';
    protected $_isPkAutoIncrement = false;

    private $sourceCarrierLinkManagement;

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

    protected function _construct()
    {
        $this->_init(self::TABLE_NAME_SOURCE, SourceInterface::SOURCE_CODE);
    }

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

    public function save(AbstractModel $object)
    {
        $connection = $this->getConnection();
        $connection->beginTransaction();
        try {
            parent::save($object);
            $this->sourceCarrierLinkManagement->saveCarrierLinksBySource($object);
            $connection->commit();
        } catch (Exception $e) {
            $connection->rollBack();
            throw $e;
        }
        return $this;
    }
}