BatchProvider.php

Go to the documentation of this file.

<?php
namespace Magento\Framework\Indexer;

use \Magento\Framework\DB\Adapter\AdapterInterface;

class BatchProvider implements BatchProviderInterface
{
    public function getBatches(AdapterInterface $adapter, $tableName, $linkField, $batchSize)
    {
        $maxLinkFieldValue = $adapter->fetchOne(
            $adapter->select()->from(
                ['entity' => $tableName],
                [
                    'max_value' => new \Zend_Db_Expr('MAX(entity.' . $linkField . ')')
                ]
            )
        );

        $truncatedBatchSize = $maxLinkFieldValue % $batchSize;
        $fullBatchCount = ($maxLinkFieldValue - $truncatedBatchSize) / $batchSize;

        for ($batchIndex = 0; $batchIndex < $fullBatchCount; $batchIndex ++) {
            yield ['from' => $batchIndex * $batchSize + 1, 'to' => ($batchIndex + 1) * $batchSize];
        }
        // return the last batch if it has smaller size
        if ($truncatedBatchSize > 0) {
            yield ['from' => $fullBatchCount * $batchSize + 1, 'to' => $maxLinkFieldValue];
        }
    }

    public function getBatchIds(
        \Magento\Framework\DB\Adapter\AdapterInterface $connection,
        \Magento\Framework\DB\Select $select,
        array $batch
    ) {
        $betweenCondition = sprintf(
            '(%s BETWEEN %s AND %s)',
            'entity_id',
            $connection->quote($batch['from']),
            $connection->quote($batch['to'])
        );

        $ids = $connection->fetchCol($select->where($betweenCondition));
        return array_map('intval', $ids);
    }
}