<?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\InventoryApi\Model\SourceCarrierLinkManagementInterface; use Magento\InventoryApi\Api\Data\SourceInterface; use Magento\Framework\Model\ResourceModel\PredefinedId; /** * 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); } /** * @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 { parent::save($object); /** @var SourceInterface $object */ $this->sourceCarrierLinkManagement->saveCarrierLinksBySource($object); $connection->commit(); } catch (Exception $e) { $connection->rollBack(); throw $e; } return $this; } }