* @copyright 2013 Akeneo SAS (http://www.akeneo.com)
* @license http://opensource.org/licenses/osl-3.0.php Open Software License (OSL 3.0)
*/
class CalculateCompletenessCommand extends Command
{
use LockableTrait;
private const DEFAULT_BATCH_SIZE = 1000;
protected static $defaultName = 'pim:completeness:calculate';
public function __construct(
private readonly ProductAndAncestorsIndexer $productAndAncestorsIndexer,
private readonly ComputeAndPersistProductCompletenesses $computeAndPersistProductCompleteness,
private readonly Connection $connection
) {
parent::__construct();
}
/**
* {@inheritdoc}
*/
protected function configure(): void
{
$this
->setDescription('Launch the product completeness calculation')
->addOption(
'batch-size',
null,
InputArgument::OPTIONAL,
'The number of product completeness calculated in one cycle.',
self::DEFAULT_BATCH_SIZE
)
;
}
/**
* {@inheritdoc}
*/
protected function execute(InputInterface $input, OutputInterface $output): int
{
if (!$this->lock()) {
$output->writeln(sprintf('The command "%s" is still running in another process.', self::$defaultName));
return 0;
}
$batchSize = (int) $input->getOption('batch-size') ?: self::DEFAULT_BATCH_SIZE;
$progressBar = new ProgressBar($output, $this->getTotalNumberOfProducts());
$output->writeln('Computing product completenesses...');
$progressBar->start();
foreach ($this->getProductUuids($batchSize) as $productUuids) {
$this->computeAndPersistProductCompleteness->fromProductUuids($productUuids);
$this->productAndAncestorsIndexer->indexFromProductUuids($productUuids);
$progressBar->advance(count($productUuids));
}
$progressBar->finish();
$output->writeln('');
$output->writeln('Completeness successfully computed.');
return 0;
}
private function getTotalNumberOfProducts(): int
{
return $this->connection->executeQuery('SELECT COUNT(0) FROM pim_catalog_product')->fetchOne();
}
/**
* @return iterable
*/
private function getProductUuids(int $batchSize): iterable
{
$lastUuidAsBytes = '';
$sql = << :lastUuid
ORDER BY uuid ASC
LIMIT :limit
SQL;
while (true) {
$rows = $this->connection->fetchFirstColumn(
$sql,
[
'lastUuid' => $lastUuidAsBytes,
'limit' => $batchSize,
],
[
'lastUuid' => \PDO::PARAM_STR,
'limit' => \PDO::PARAM_INT,
]
);
if (empty($rows)) {
return;
}
$lastUuidAsBytes = end($rows);
yield array_map(fn (string $uuidAsBytes): UuidInterface => Uuid::fromBytes($uuidAsBytes), $rows);
}
}
}