Your IP : 216.73.216.97
<?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;
}
}