12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182 |
- <?php
- /**
- * Copyright © Magento, Inc. All rights reserved.
- * See COPYING.txt for license details.
- */
- declare(strict_types=1);
- namespace Magento\InventoryIndexer\Indexer;
- use Magento\Framework\App\ResourceConnection;
- use Magento\Framework\Indexer\SaveHandler\Batch;
- use Magento\InventoryMultiDimensionalIndexerApi\Model\IndexHandlerInterface;
- use Magento\InventoryMultiDimensionalIndexerApi\Model\IndexName;
- use Magento\InventoryMultiDimensionalIndexerApi\Model\IndexNameResolverInterface;
- /**
- * Index handler is responsible for index data manipulation
- */
- class IndexHandler implements IndexHandlerInterface
- {
- /**
- * @var IndexNameResolverInterface
- */
- private $indexNameResolver;
- /**
- * @var Batch
- */
- private $batch;
- /**
- * @var ResourceConnection
- */
- private $resourceConnection;
- /**
- * @var int
- */
- private $batchSize;
- /**
- * @param IndexNameResolverInterface $indexNameResolver
- * @param Batch $batch
- * @param ResourceConnection $resourceConnection
- * @param int $batchSize
- */
- public function __construct(
- IndexNameResolverInterface $indexNameResolver,
- Batch $batch,
- ResourceConnection $resourceConnection,
- $batchSize
- ) {
- $this->indexNameResolver = $indexNameResolver;
- $this->batch = $batch;
- $this->resourceConnection = $resourceConnection;
- $this->batchSize = $batchSize;
- }
- /**
- * @inheritdoc
- */
- public function saveIndex(IndexName $indexName, \Traversable $documents, string $connectionName): void
- {
- $connection = $this->resourceConnection->getConnection($connectionName);
- $tableName = $this->indexNameResolver->resolveName($indexName);
- $columns = [IndexStructure::SKU, IndexStructure::QUANTITY, IndexStructure::IS_SALABLE];
- foreach ($this->batch->getItems($documents, $this->batchSize) as $batchDocuments) {
- $connection->insertArray($tableName, $columns, $batchDocuments);
- }
- }
- /**
- * @inheritdoc
- */
- public function cleanIndex(IndexName $indexName, \Traversable $documents, string $connectionName): void
- {
- $connection = $this->resourceConnection->getConnection($connectionName);
- $tableName = $this->indexNameResolver->resolveName($indexName);
- $connection->delete($tableName, ['sku IN (?)' => iterator_to_array($documents)]);
- }
- }
|