treehouselabs / worker-bundle

src/TreeHouse/WorkerBundle/Command/ClearCommand.php

<?php

namespace TreeHouse\WorkerBundle\Command;

use Pheanstalk\Exception\ServerException;
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 TreeHouse\WorkerBundle\QueueManager;

class ClearCommand extends Command
{
    /**
     * @var QueueManager
     */
    protected $manager;

    /**
     * @param QueueManager $queueManager
     */
    public function __construct(QueueManager $queueManager)
    {
        $this->manager = $queueManager;

        parent::__construct();
    }

    /**
     * @inheritdoc
     */
    protected function configure()
    {
        $this
            ->setName('worker:clear')
            ->addArgument('action', InputArgument::OPTIONAL | InputArgument::IS_ARRAY, 'Selects which actions to clear. Defaults to all actions')
            ->addOption('state', null, InputOption::VALUE_OPTIONAL | InputOption::VALUE_IS_ARRAY, 'The state of jobs to peek for. Valid options are <comment>ready</comment>, <comment>delayed</comment> and <comment>buried</comment>.', ['ready'])
            ->setDescription('Clears jobs for action(s)')
        ;
    }

    /**
     * @inheritdoc
     */
    protected function execute(InputInterface $input, OutputInterface $output)
    {
        $actions = $input->getArgument('action');
        if (empty($actions)) {
            $actions = array_keys($this->manager->getExecutors());
        }

        $states = $input->getOption('state');

        $pheanstalk = $this->manager->getPheanstalk();

        foreach ($actions as $action) {
            try {
                $stats = $pheanstalk->statsTube($action);

                $amount = 0;
                foreach ($states as $state) {
                    $amount += $stats['current-jobs-' . $state];
                }

                $output->writeln(
                    sprintf(
                        'Clearing <info>%d %s jobs</info> with <info>%s</info> status',
                        $amount,
                        $action,
                        implode(', ', $states)
                    )
                );

                $this->manager->clear($action, $states);

                $output->writeln(['<info>Done!</info>', '']);

array('<info>Done!</info>', '') is of type array<integer,string,{"0":"string","1":"string"}>, but the function expects a string|object<Symfony\Component\Console\Output\iterable>.

            } catch (ServerException $e) {
                if (false === strpos($e->getMessage(), 'NOT_FOUND')) {
                    throw $e;
                }
            }
        }
    }
}