mirror of
https://github.com/wallabag/wallabag.git
synced 2024-05-10 13:22:38 +00:00
58 lines
2 KiB
PHP
58 lines
2 KiB
PHP
<?php
|
|
|
|
namespace Wallabag\Command\Import;
|
|
|
|
use Simpleue\Worker\QueueWorker;
|
|
use Symfony\Component\Config\Definition\Exception\Exception;
|
|
use Symfony\Component\Console\Command\Command;
|
|
use Symfony\Component\Console\Input\InputArgument;
|
|
use Symfony\Component\Console\Input\InputInterface;
|
|
use Symfony\Component\Console\Input\InputOption;
|
|
use Symfony\Component\Console\Output\OutputInterface;
|
|
use Symfony\Component\DependencyInjection\ContainerInterface;
|
|
|
|
class RedisWorkerCommand extends Command
|
|
{
|
|
protected static $defaultName = 'wallabag:import:redis-worker';
|
|
protected static $defaultDescription = 'Launch Redis worker';
|
|
|
|
private $container;
|
|
|
|
public function __construct(ContainerInterface $container)
|
|
{
|
|
$this->container = $container;
|
|
|
|
parent::__construct();
|
|
}
|
|
|
|
protected function configure()
|
|
{
|
|
$this
|
|
->addArgument('serviceName', InputArgument::REQUIRED, 'Service to use: wallabag_v1, wallabag_v2, pocket, readability, pinboard, delicious, firefox, chrome or instapaper')
|
|
->addOption('maxIterations', '', InputOption::VALUE_OPTIONAL, 'Number of iterations before stopping', false)
|
|
;
|
|
}
|
|
|
|
protected function execute(InputInterface $input, OutputInterface $output)
|
|
{
|
|
$output->writeln('Worker started at: ' . (new \DateTime())->format('d-m-Y G:i:s'));
|
|
$output->writeln('Waiting for message ...');
|
|
|
|
$serviceName = $input->getArgument('serviceName');
|
|
|
|
if (!$this->container->has('wallabag.queue.redis.' . $serviceName) || !$this->container->has('wallabag.consumer.redis.' . $serviceName)) {
|
|
throw new Exception(sprintf('No queue or consumer found for service name: "%s"', $input->getArgument('serviceName')));
|
|
}
|
|
|
|
$worker = new QueueWorker(
|
|
$this->container->get('wallabag.queue.redis.' . $serviceName),
|
|
$this->container->get('wallabag.consumer.redis.' . $serviceName),
|
|
(int) $input->getOption('maxIterations')
|
|
);
|
|
|
|
$worker->start();
|
|
|
|
return 0;
|
|
}
|
|
}
|