Phraseanet/lib/classes/task/manager.class.php

<?php

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

/**
 *
 * @package
 * @license     http://opensource.org/licenses/gpl-3.0 GPLv3
 * @link        www.phraseanet.com
 */
class task_manager
{
    const STATUS_SCHED_STOPPED = 'stopped';
    const STATUS_SCHED_STOPPING = 'stopping';
    const STATUS_SCHED_STARTED = 'started';
    const STATUS_SCHED_TOSTOP = 'tostop';

    protected $appbox;
    protected $tasks;

    public function __construct(appbox &$appbox)
    {
        $this->appbox = $appbox;

        return $this;
    }

    public function old_get_tasks($refresh = false)
    {
        if ($this->tasks && ! $refresh) {
            return $this->tasks;
        }

        $sql = "SELECT task2.* FROM task2 ORDER BY task_id ASC";
        $stmt = $this->appbox->get_connection()->prepare($sql);
        $stmt->execute();
        $rs = $stmt->fetchAll(PDO::FETCH_ASSOC);
        $stmt->closeCursor();

        $tasks = array();

        foreach ($rs as $row) {
            $classname = $row['class'];
            if ( ! class_exists($classname))
                continue;
            try {
                $tasks[$row['task_id']] = new $classname($row['task_id']);
            } catch (Exception $e) {

            }
        }

        $this->tasks = $tasks;

        return $this->tasks;
    }

    public function get_tasks($refresh = false)
    {
        if ($this->tasks && ! $refresh) {
            return $this->tasks;
        }

        $sql = "SELECT task2.* FROM task2 ORDER BY task_id ASC";
        $stmt = $this->appbox->get_connection()->prepare($sql);
        $stmt->execute();
        $rs = $stmt->fetchAll(PDO::FETCH_ASSOC);
        $stmt->closeCursor();

        $tasks = array();

        $appbox = \appbox::get_instance(\bootstrap::getCore());
        $lockdir = $appbox->get_registry()->get('GV_RootPath') . 'tmp/locks/';

        foreach ($rs as $row) {
            $row['pid'] = NULL;

            $classname = $row['class'];
            if ( ! class_exists($classname))
                continue;
            try {
//        if( ($lock = fopen( $lockdir . 'task.'.$row['task_id'].'.lock', 'a+')) )
//        {
//          if (flock($lock, LOCK_SH | LOCK_NB) === FALSE)
//          {
//            // already locked : running !
//            $row['pid'] = fgets($lock, 512);
//          }
//          else
//          {
//            // can lock : not running
//            flock($lock, LOCK_UN);
//          }
//          fclose($lock);
//        }
                $tasks[$row['task_id']] = new $classname($row['task_id']);
            } catch (Exception $e) {

            }
        }

        $this->tasks = $tasks;

        return $this->tasks;
    }

    /**
     *
     * @param <type> $task_id
     * @return task_abstract
     */
    public function get_task($task_id)
    {
        $tasks = $this->get_tasks();

        if ( ! isset($tasks[$task_id]))
            throw new Exception('Unknown task_id');

        return $tasks[$task_id];
    }

    public function set_sched_status($status)
    {
        $av_status = array(
            self::STATUS_SCHED_STARTED
            , self::STATUS_SCHED_STOPPED
            , self::STATUS_SCHED_STOPPING
            , self::STATUS_SCHED_TOSTOP
        );

        if ( ! in_array($status, $av_status))
            throw new Exception(sprintf('unknown status `%s` ', $status));

        $sql = "UPDATE sitepreff SET schedstatus = :schedstatus, schedqtime=NOW()";
        $stmt = $this->appbox->get_connection()->prepare($sql);
        $stmt->execute(array(':schedstatus' => $status));
        $stmt->closeCursor();

        return $this;
    }

    public function get_scheduler_state()
    {
        $pid = NULL;
        $appbox = appbox::get_instance(\bootstrap::getCore());
        $lockdir = $appbox->get_registry()->get('GV_RootPath') . 'tmp/locks/';
        if (($schedlock = fopen($lockdir . 'scheduler.lock', 'a+'))) {
            if (flock($schedlock, LOCK_SH | LOCK_NB) === FALSE) {
                // already locked : running !
                $pid = trim(fgets($schedlock, 512));
            } else {
                // can lock : not running
                flock($schedlock, LOCK_UN);
            }
            fclose($schedlock);
        }

        $sql = "SELECT UNIX_TIMESTAMP()-UNIX_TIMESTAMP(schedqtime) AS qdelay
            , schedstatus AS status FROM sitepreff";
        $stmt = $this->appbox->get_connection()->prepare($sql);
        $stmt->execute();
        $ret = $stmt->fetch(PDO::FETCH_ASSOC);
        $stmt->closeCursor();

        if ($pid === NULL && $ret['status'] !== 'stopped') {
            // auto fix
            $this->appbox->get_connection()->exec('UPDATE sitepreff SET schedstatus=\'stopped\'');
            $ret['status'] = 'stopped';
        }
        $ret['pid'] = $pid;

        return $ret;
    }

    public static function getAvailableTasks()
    {
        $registry = registry::get_instance();
        $taskdir = array($registry->get('GV_RootPath') . "lib/classes/task/period/"
            , $registry->get('GV_RootPath') . "config/classes/task/period/"
        );

        $tasks = array();
        foreach ($taskdir as $path) {
            if (($hdir = @opendir($path))) {
                $tskin = array();
                $max = 9999;
                while (($max -- > 0) && (($file = readdir($hdir)) !== false)) {
                    if ( ! is_file($path . '/' . $file) || substr($file, 0, 1) == "." || substr($file, -10) != ".class.php")
                        continue;

                    $classname = 'task_period_' . substr($file, 0, strlen($file) - 10);

                    try {
                        //      $testclass = new $classname(null);
                        if ($classname::interfaceAvailable()) {
                            $tasks[] = array("class" => $classname, "name"  => $classname::getName(), "err"   => null);
                        }
                    } catch (Exception $e) {

                    }
                }
                closedir($hdir);
            }
        }

        return $tasks;
    }
}