129 lines
3.9 KiB
PHP
129 lines
3.9 KiB
PHP
<?php
|
||
|
||
// +----------------------------------------------------------------------
|
||
// | CatchAdmin [Just Like ～ ]
|
||
// +----------------------------------------------------------------------
|
||
// | Copyright (c) 2017~{$year} http://catchadmin.com All rights reserved.
|
||
// +----------------------------------------------------------------------
|
||
// | Licensed ( https://github.com/yanwenwu/catch-admin/blob/master/LICENSE.txt )
|
||
// +----------------------------------------------------------------------
|
||
// | Author: JaguarJack [ njphper@gmail.com ]
|
||
// +----------------------------------------------------------------------
|
||
|
||
declare (strict_types=1);
|
||
|
||
namespace catchAdmin\monitor\command;
|
||
|
||
use catchAdmin\monitor\command\process\Process;
|
||
use catchAdmin\monitor\model\Crontab;
|
||
use catchAdmin\monitor\model\CrontabLog;
|
||
use Cron\CronExpression;
|
||
use think\console\Command;
|
||
use think\console\Input;
|
||
use think\console\Output;
|
||
use think\facade\Log;
|
||
|
||
class ScheduleCommand extends Command
|
||
{
|
||
protected $pid;
|
||
|
||
protected function configure()
|
||
{
|
||
// 指令配置
|
||
$this->setName('catch:schedule')
|
||
->setDescription('catch schedule');
|
||
}
|
||
|
||
protected function execute(Input $input, Output $output)
|
||
{
|
||
try {
|
||
foreach ($this->getExecuteAbleCommands() as $command) {
|
||
|
||
$process = new Process(function (Process $process) use ($command) {
|
||
$this->executeCommand($command);
|
||
$process->exit();
|
||
});
|
||
|
||
$process->start();
|
||
}
|
||
} catch (\Exception $e) {
|
||
Log::error('CatchSchedule Error:' . $e->getMessage());
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 执行 command
|
||
*
|
||
* @time 2021年01月18日
|
||
* @param $command
|
||
* @return void
|
||
*/
|
||
protected function executeCommand($command)
|
||
{
|
||
$start = time();
|
||
|
||
$errorMessage = '';
|
||
|
||
try {
|
||
$this->getConsole()->call($command->task);
|
||
} catch (\Exception $e) {
|
||
$errorMessage = $e->getMessage();
|
||
}
|
||
|
||
$end = time();
|
||
|
||
// 插入 crontab 执行日志
|
||
CrontabLog::insert([
|
||
'crontab_id' => $command->id,
|
||
'used_time' => $end - $start,
|
||
'status' => $errorMessage ? CrontabLog::FAILED : CrontabLog::SUCCESS,
|
||
'error_message' => $errorMessage,
|
||
'created_at' => time(),
|
||
'updated_at' => time(),
|
||
]);
|
||
}
|
||
|
||
/**
|
||
* 获取可执行任务
|
||
*
|
||
* @time 2021年01月17日
|
||
* @throws \think\db\exception\DataNotFoundException
|
||
* @throws \think\db\exception\DbException
|
||
* @throws \think\db\exception\ModelNotFoundException
|
||
* @return array
|
||
*/
|
||
protected function getExecuteAbleCommands()
|
||
{
|
||
$executeAbleCommands = [];
|
||
|
||
Crontab::where('status', Crontab::ENABLE)
|
||
->where('tactics', '<>', Crontab::EXECUTE_FORBIDDEN)
|
||
->select()
|
||
->each(function ($command) use (&$executeAbleCommands){
|
||
if ($command->tactics == Crontab::EXECUTE_IMMEDIATELY) {
|
||
$executeAbleCommands[] = $command;
|
||
return true;
|
||
}
|
||
|
||
$can = date('Y-m-d H:i',
|
||
(new CronExpression(trim($command->cron)))
|
||
->getNextRunDate(date('Y-m-d H:i:s'), 0, true)
|
||
->getTimestamp()) == date('Y-m-d H:i', time());
|
||
|
||
if ($can) {
|
||
// 如果任务只执行一次 之后禁用该任务
|
||
if ($command->tactics === Crontab::EXECUTE_ONCE) {
|
||
Crontab::where('id', $command->id)->update([
|
||
'status' => Crontab::DISABLE,
|
||
]);
|
||
}
|
||
|
||
$executeAbleCommands[] = $command;
|
||
}
|
||
return true;
|
||
});
|
||
|
||
return $executeAbleCommands;
|
||
}
|
||
}
|