Your IP : 216.73.216.220
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
namespace Magento\CatalogInventory\Model\Stock;
use Magento\CatalogInventory\Api\Data\StockStatusCollectionInterfaceFactory;
use Magento\CatalogInventory\Api\Data\StockStatusInterface;
use Magento\CatalogInventory\Api\StockStatusRepositoryInterface;
use Magento\CatalogInventory\Model\ResourceModel\Stock\Status as StockStatusResource;
use Magento\CatalogInventory\Model\StockRegistryStorage;
use Magento\Framework\DB\MapperFactory;
use Magento\Framework\DB\QueryBuilderFactory;
use Magento\Framework\Exception\CouldNotDeleteException;
use Magento\Framework\Exception\CouldNotSaveException;
/**
* Class StockStatusRepository
* @SuppressWarnings(PHPMD.CouplingBetweenObjects)
*/
class StockStatusRepository implements StockStatusRepositoryInterface
{
/**
* @var StockStatusResource
*/
protected $resource;
/**
* @var StatusFactory
*/
protected $stockStatusFactory;
/**
* @var StockStatusCollectionInterfaceFactory
*/
protected $stockStatusCollectionFactory;
/**
* @var QueryBuilderFactory
*/
protected $queryBuilderFactory;
/**
* @var MapperFactory
*/
protected $mapperFactory;
/**
* @var StockRegistryStorage
*/
protected $stockRegistryStorage;
/**
* @param StockStatusResource $resource
* @param StatusFactory $stockStatusFactory
* @param StockStatusCollectionInterfaceFactory $collectionFactory
* @param QueryBuilderFactory $queryBuilderFactory
* @param MapperFactory $mapperFactory
*/
public function __construct(
StockStatusResource $resource,
StatusFactory $stockStatusFactory,
StockStatusCollectionInterfaceFactory $collectionFactory,
QueryBuilderFactory $queryBuilderFactory,
MapperFactory $mapperFactory
) {
$this->resource = $resource;
$this->stockStatusFactory = $stockStatusFactory;
$this->stockStatusCollectionFactory = $collectionFactory;
$this->queryBuilderFactory = $queryBuilderFactory;
$this->mapperFactory = $mapperFactory;
}
/**
* @param StockStatusInterface $stockStatus
* @return StockStatusInterface
* @throws CouldNotSaveException
*/
public function save(StockStatusInterface $stockStatus)
{
try {
$this->resource->save($stockStatus);
} catch (\Exception $exception) {
throw new CouldNotSaveException(
__('The stock status was unable to be saved. Please try again.'),
$exception
);
}
return $stockStatus;
}
/**
* @param string $stockStatusId
* @return StockStatusInterface|Status
*/
public function get($stockStatusId)
{
$stockStatus = $this->stockStatusFactory->create();
$this->resource->load($stockStatus, $stockStatusId);
return $stockStatus;
}
/**
* @param \Magento\CatalogInventory\Api\StockStatusCriteriaInterface $criteria
* @return \Magento\CatalogInventory\Api\Data\StockStatusCollectionInterface
*/
public function getList(\Magento\CatalogInventory\Api\StockStatusCriteriaInterface $criteria)
{
$queryBuilder = $this->queryBuilderFactory->create();
$queryBuilder->setCriteria($criteria);
$queryBuilder->setResource($this->resource);
$query = $queryBuilder->create();
$collection = $this->stockStatusCollectionFactory->create(['query' => $query]);
return $collection;
}
/**
* @param StockStatusInterface $stockStatus
* @return bool|true
* @throws CouldNotDeleteException
*/
public function delete(StockStatusInterface $stockStatus)
{
try {
$this->resource->delete($stockStatus);
$this->getStockRegistryStorage()->removeStockStatus($stockStatus->getProductId());
} catch (\Exception $exception) {
throw new CouldNotDeleteException(
__('Unable to remove Stock Status for product %1', $stockStatus->getProductId()),
$exception
);
}
return true;
}
/**
* @param int $id
* @return bool
* @throws CouldNotDeleteException
*/
public function deleteById($id)
{
try {
$stockStatus = $this->get($id);
$this->delete($stockStatus);
} catch (\Exception $exception) {
throw new CouldNotDeleteException(
__('Unable to remove Stock Status for product %1', $id),
$exception
);
}
return true;
}
/**
* @return StockRegistryStorage
*/
private function getStockRegistryStorage()
{
if (null === $this->stockRegistryStorage) {
$this->stockRegistryStorage = \Magento\Framework\App\ObjectManager::getInstance()
->get(\Magento\CatalogInventory\Model\StockRegistryStorage::class);
}
return $this->stockRegistryStorage;
}
}