159 lines
4.5 KiB
PHP
159 lines
4.5 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 ]
|
||
// +----------------------------------------------------------------------
|
||
|
||
namespace catchAdmin\monitor\command\process;
|
||
|
||
use catchAdmin\monitor\model\Crontab;
|
||
use catcher\facade\FileSystem;
|
||
use Cron\CronExpression;
|
||
use EasyWeChat\Kernel\Messages\News;
|
||
use think\cache\driver\Redis;
|
||
|
||
trait RegisterSignal
|
||
{
|
||
public function registerSignal()
|
||
{
|
||
// 退出信号
|
||
$this->exit();
|
||
// 等待子进程退出
|
||
$this->waitWorkersExit();
|
||
// 动态扩容
|
||
$this->workerChecked();
|
||
// 重启进程
|
||
$this->reload();
|
||
// 统计信息
|
||
$this->showStatus();
|
||
}
|
||
|
||
/**
|
||
* 进程退出
|
||
*
|
||
* @time 2020年07月21日
|
||
* @return void
|
||
*/
|
||
protected function exit()
|
||
{
|
||
Process::signal(SIGTERM, function ($signal) {
|
||
foreach ($this->workerIds as $pid => $v) {
|
||
if (isset($this->workerIds[$pid])) {
|
||
unset($this->workerIds[$pid]);
|
||
}
|
||
Process::kill($pid, SIGTERM);
|
||
}
|
||
Process::kill(self::getMasterPid(), SIGKILL);
|
||
});
|
||
|
||
Process::signal(SIGINT, function ($signal) {
|
||
foreach ($this->workerIds as $pid => $v) {
|
||
if (isset($this->workerIds[$pid])) {
|
||
unset($this->workerIds[$pid]);
|
||
}
|
||
Process::kill($pid, SIGKILL);
|
||
}
|
||
Process::kill(self::getMasterPid(), SIGKILL);
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 子进程退出
|
||
*
|
||
* @time 2020年07月21日
|
||
* @return void
|
||
*/
|
||
protected function waitWorkersExit()
|
||
{
|
||
Process::signal(SIGCHLD, function ($signal) {
|
||
|
||
});
|
||
}
|
||
|
||
|
||
/**
|
||
* 进程检测
|
||
*
|
||
* @time 2020年07月21日
|
||
* @return void
|
||
*/
|
||
protected function workerChecked()
|
||
{
|
||
Process::signal(SIGALRM, function ($signal) {
|
||
$process = new Process(function (Process $process) {
|
||
$crontabs = Crontab::where('status', Crontab::ENABLE)
|
||
->where('tactics', '<>', Crontab::EXECUTE_FORBIDDEN)
|
||
->select()->toArray();
|
||
|
||
// 任务
|
||
foreach ($crontabs as $crontab) {
|
||
$can = date('Y-m-d H:i', CronExpression::factory(trim($crontab['cron']))
|
||
->getNextRunDate(date('Y-m-d H:i:s'), 0 , true)
|
||
->getTimestamp()) == date('Y-m-d H:i', time());
|
||
|
||
if ($can) {
|
||
// 如果任务只执行一次 之后禁用该任务
|
||
if ($crontab['tactics'] === Crontab::EXECUTE_ONCE) {
|
||
Crontab::where('id', $crontab['id'])->update([
|
||
'status' => Crontab::DISABLE,
|
||
]);
|
||
}
|
||
|
||
$redis = $this->getRedisHandle();
|
||
|
||
$redis->lpush($this->crontabQueueName, json_encode([
|
||
'id' => $crontab['id'],
|
||
'task' => $crontab['task'],
|
||
]));
|
||
}
|
||
}
|
||
|
||
$process->exit();
|
||
});
|
||
|
||
$process->start();
|
||
|
||
Process::alarm($this->interval);
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 重启
|
||
*
|
||
* @time 2020年07月21日
|
||
* @return void
|
||
*/
|
||
protected function reload()
|
||
{
|
||
Process::signal(SIGUSR1, function ($signal) {
|
||
$this->worker_start_at = 0;
|
||
foreach ($this->workerIds as $pid => $v) {
|
||
Process::kill($pid, SIGTERM);
|
||
}
|
||
}, false);
|
||
}
|
||
|
||
/**
|
||
* 预留信号
|
||
*
|
||
* @time 2020年07月21日
|
||
* @return void
|
||
*/
|
||
protected function showStatus()
|
||
{
|
||
Process::signal(SIGUSR2, function ($signal) {
|
||
$this->setWorkerStatus($this->name . ' master');
|
||
foreach ($this->workerIds as $pid => $v) {
|
||
Process::kill($pid, SIGUSR2);
|
||
}
|
||
});
|
||
}
|
||
}
|
||
|
||
|