\addons\crontab\controllerAutotask

<?php

namespace addons\crontab\controller;

use app\common\model\Crontab;
use Cron\CronExpression;
use fast\Http;
use think\Controller;
use think\Db;
use think\Exception;
use think\Log;

/**
 * 定时任务接口
 *
 * 以Crontab方式每分钟定时执行,且只可以Cli方式运行
 * @internal
 */
class Autotask extends Controller
{

    /**
     * 初始化方法,最前且始终执行
     */
    public function _initialize()
    {
        // 只可以以cli方式执行
        if (!$this->request->isCli())
            $this->error('Autotask script only work at client!');

        parent::_initialize();

        // 清除错误
        error_reporting(0);

        // 设置永不超时
        set_time_limit(0);
    }

    /**
     * 执行定时任务
     */
    public function index()
    {
        $time = time();
        $logDir = LOG_PATH . 'crontab/';
        if (!is_dir($logDir)) {
            mkdir($logDir, 0755);
        }
        //筛选未过期且未完成的任务
        $crontabList = Crontab::where('status', '=', 'normal')->order('weigh desc,id desc')->select();
        $execTime = time();
        foreach ($crontabList as $crontab) {
            $update = [];
            $execute = FALSE;
            if ($time < $crontab['begintime']) {
                //任务未开始
                continue;
            }
            if ($crontab['maximums'] && $crontab['executes'] > $crontab['maximums']) {
                //任务已超过最大执行次数
                $update['status'] = 'completed';
            } else if ($crontab['endtime'] > 0 && $time > $crontab['endtime']) {
                //任务已过期
                $update['status'] = 'expired';
            } else {
                //重复执行
                //如果未到执行时间则继续循环
                $cron = CronExpression::factory($crontab['schedule']);
                if (!$cron->isDue(date("YmdHi", $execTime)) || date("YmdHi", $execTime) === date("YmdHi", $crontab['executetime']))
                    continue;
                $execute = TRUE;
            }

            // 如果允许执行
            if ($execute) {
                $update['executetime'] = $time;
                $update['executes'] = $crontab['executes'] + 1;
                $update['status'] = ($crontab['maximums'] > 0 && $update['executes'] >= $crontab['maximums']) ? 'completed' : 'normal';
            }

            // 如果需要更新状态
            if (!$update)
                continue;
            // 更新状态
            $crontab->save($update);

            // 将执行放在后面是为了避免超时导致多次执行
            if (!$execute)
                continue;
            try {
                if ($crontab['type'] == 'url') {
                    if (substr($crontab['content'], 0, 1) == "/") {
                        // 本地项目URL
                        exec('nohup php ' . ROOT_PATH . 'public/index.php ' . $crontab['content'] . ' >> ' . $logDir . date("Y-m-d") . '.log 2>&1 &');
                    } else {
                        // 远程异步调用URL
                        Http::sendAsyncRequest($crontab['content']);
                    }
                } else if ($crontab['type'] == 'sql') {
                    //这里需要强制重连数据库,使用已有的连接会报2014错误
                    $connect = Db::connect([], true);
                    $connect->execute("select 1");
                    // 执行SQL
                    $connect->getPdo()->exec($crontab['content']);
                } else if ($crontab['type'] == 'shell') {
                    // 执行Shell
                    exec($crontab['content'] . ' >> ' . $logDir . date("Y-m-d") . '.log 2>&1 &');
                }
            } catch (Exception $e) {
                Log::record($e->getMessage());
            }
        }
        return 'Execute completed!';
    }

}