Phraseanet/lib/classes/module/console/taskrun.php

<?php

/*
 * This file is part of Phraseanet
 *
 * (c) 2005-2012 Alchemy
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

use Alchemy\Phrasea\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 Monolog\Handler;
use Monolog\Logger;
use Monolog\Handler\StreamHandler;
use Monolog\Handler\RotatingFileHandler;

/**
 * @todo write tests
 *
 * @license     http://opensource.org/licenses/gpl-3.0 GPLv3
 * @link        www.phraseanet.com
 */
class module_console_taskrun extends Command
{
    private $task;
    private $shedulerPID;

    public function __construct($name = null)
    {
        parent::__construct($name);

        $this->task = NULL;
        $this->shedulerPID = NULL;

        $this->addArgument('task_id', InputArgument::REQUIRED, 'The task_id to run');
        $this->addOption(
            'runner'
            , 'r'
            , InputOption::VALUE_REQUIRED
            , 'The name of the runner (manual, scheduler...)'
            , task_abstract::RUNNER_MANUAL
        );
        $this->addOption(
            'ttyloglevel'
            , 't'
            , InputOption::VALUE_REQUIRED
            , 'threshold : (DEBUG|INFO|WARNING|ERROR|CRITICAL|ALERT)'
            , ''
        );
        $this->setDescription('Run task');

        return $this;
    }

    public function sig_handler($signo)
    {
        if ($this->task) {
            $this->task->log(sprintf("signal %s received", $signo));
            $this->task->setRunning(false);
        }
    }

    protected function doExecute(InputInterface $input, OutputInterface $output)
    {
        if (!$this->container['phraseanet.configuration-tester']->isInstalled()) {
            return self::EXITCODE_SETUP_ERROR;
        }

        $task_id = (int) $input->getArgument('task_id');
        if ($task_id <= 0 || strlen($task_id) !== strlen($input->getArgument('task_id'))) {
            throw new \RuntimeException('Argument must be an Id.');
        }

        $task_manager = $this->container['task-manager'];
        $logger = $task_manager->getLogger();

        if ($input->getOption('runner') === task_abstract::RUNNER_MANUAL) {
            $schedStatus = $task_manager->getSchedulerState();

            if ($schedStatus && $schedStatus['status'] == task_abstract::STATE_STARTED && $schedStatus['pid']) {
                $this->shedulerPID = $schedStatus['pid'];
            }
            $runner = task_abstract::RUNNER_MANUAL;
        } else {
            $runner = task_abstract::RUNNER_SCHEDULER;
            $schedStatus = $task_manager->getSchedulerState();
            if ($schedStatus && $schedStatus['status'] == task_abstract::STATE_STARTED && $schedStatus['pid']) {
                $this->shedulerPID = $schedStatus['pid'];
            }
        }


        if ($input->getOption('verbose')) {
            $handler = new StreamHandler(fopen('php://stdout', 'a'));
            $this->container['monolog']->pushHandler($handler);
        }

        $logfile = __DIR__ . '/../../../../logs/task_' . $task_id . '.log';
        $handler = new RotatingFileHandler($logfile, 10);
        $this->container['monolog']->pushHandler($handler);
        $this->task = $task_manager->getTask($task_id, $this->container['monolog']);

        $lib2v = array(
            'DEBUG'       => task_abstract::LOG_DEBUG,
            'INFO'        => task_abstract::LOG_INFO,
            'WARNING'     => task_abstract::LOG_WARNING,
            'ERROR'       => task_abstract::LOG_ERROR,
            'CRITICAL'    => task_abstract::LOG_CRITICAL,
            'ALERT'       => task_abstract::LOG_ALERT
        );

        $tmpTask = $task_manager->getTask($task_id, null);
        $taskname =  $tmpTask->getName();
        unset($tmpTask);


        // log to tty ?

        if(($ttyloglevel = strtoupper($input->getOption('ttyloglevel'))) != '') {
            if (!array_key_exists($ttyloglevel, $lib2v)) {
                throw(new Alchemy\Phrasea\Exception\RuntimeException(sprintf(
                        "Bad value '%s' for option loglevel\nuse DEBUG|INFO|WARNING|ERROR|CRITICAL|ALERT", $ttyloglevel))
                );
            }
            $handler = new Handler\StreamHandler(
                "php://stdout",
                $lib2v[$ttyloglevel],
                true
            );
            $logger->pushHandler($handler);
        }

        $logfile = __DIR__ . '/../../../../logs/task_' . $task_id . '.log';
        $handler = new RotatingFileHandler($logfile, 10);
        $logger->pushHandler($handler);

        $this->task = $task_manager->getTask($task_id, $logger);

        register_tick_function(array($this, 'tick_handler'), true);
        declare(ticks = 1);

        if (function_exists('pcntl_signal')) {
            pcntl_signal(SIGTERM, array($this, 'sig_handler'));
            pcntl_signal(SIGINT, array($this, 'sig_handler'));
            //   pcntl_signal(SIGKILL, array($this, 'sig_handler'));
        }

        try {
            $this->task->run($runner);
        } catch (Exception $e) {
            $this->task->log(sprintf("taskrun : exception from 'run()', %s \n", $e->getMessage()));

            return($e->getCode());
        }

        if ($input->getOption('runner') === task_abstract::RUNNER_MANUAL) {
            $runner = task_abstract::RUNNER_MANUAL;
        }
    }

    public function tick_handler()
    {
        static $start;

        if ($start === null) {
            $start = time();
        }

        if (time() - $start > 0) {
            if ($this->shedulerPID) {
                if (function_exists('posix_kill') && !posix_kill($this->shedulerPID, 0)) {
                    if (method_exists($this->task, 'signal')) {
                        $this->task->signal('SIGNAL_SCHEDULER_DIED');
                    } else {
                        $this->task->setState(task_abstract::STATE_TOSTOP);
                    }
                }

                $start = time();
            }
        }
    }
}