qqq
This commit is contained in:
+289
-322
@@ -5,371 +5,338 @@ namespace Kiri\Server;
|
||||
|
||||
use Exception;
|
||||
use Kiri\Di\Context;
|
||||
use Kiri\Server\Abstracts\BaseProcess;
|
||||
use Swoole\Coroutine;
|
||||
use Swoole\Event;
|
||||
use Swoole\Process;
|
||||
use Symfony\Component\Console\Command\Command;
|
||||
use Symfony\Component\Console\Input\InputInterface;
|
||||
use Symfony\Component\Console\Output\OutputInterface;
|
||||
|
||||
class HotReload extends Command
|
||||
class HotReload extends BaseProcess
|
||||
{
|
||||
|
||||
public string $name = 'hot:reload';
|
||||
/**
|
||||
* @var array|mixed
|
||||
*/
|
||||
private array $watchFiles = [];
|
||||
|
||||
private array $md5Map = [];
|
||||
/**
|
||||
* @var array|string[]
|
||||
*/
|
||||
private array $dirs = [APP_PATH . 'app/', APP_PATH . 'config/', APP_PATH . 'routes/'];
|
||||
|
||||
|
||||
/**
|
||||
* @var Process
|
||||
*/
|
||||
private mixed $process;
|
||||
/**
|
||||
* @var array|mixed
|
||||
*/
|
||||
private array $watchFiles = [];
|
||||
/**
|
||||
* @return string
|
||||
*/
|
||||
public function getName(): string
|
||||
{
|
||||
return 'hot.load'; // TODO: Change the autogenerated stub
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @return void
|
||||
*/
|
||||
protected function configure()
|
||||
{
|
||||
$this->setName('hot:load');
|
||||
parent::configure(); // TODO: Change the autogenerated stub
|
||||
}
|
||||
/**
|
||||
* @return $this
|
||||
*/
|
||||
public function onSigterm(): static
|
||||
{
|
||||
// TODO: Implement onSigterm() method.
|
||||
if (Context::inCoroutine()) {
|
||||
Coroutine::create(fn() => $this->onShutdown(Coroutine::waitSignal(SIGTERM | SIGINT)));
|
||||
} else {
|
||||
Process::signal(SIGTERM | SIGINT, fn($data) => $this->onShutdown($data));
|
||||
}
|
||||
return $this;
|
||||
}
|
||||
|
||||
|
||||
private array $dirs = [APP_PATH . 'app/', APP_PATH . 'config/', APP_PATH . 'routes/'];
|
||||
/**
|
||||
* @param Process|null $process
|
||||
* @return void
|
||||
* @throws Exception
|
||||
*/
|
||||
public function process(?Process $process): void
|
||||
{
|
||||
// TODO: Implement process() method.
|
||||
if (extension_loaded('inotify')) {
|
||||
$this->onInotifyReload();
|
||||
} else {
|
||||
$this->onCrontabReload();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param InputInterface $input
|
||||
* @param OutputInterface $output
|
||||
* @return int
|
||||
* @throws Exception
|
||||
*/
|
||||
protected function execute(InputInterface $input, OutputInterface $output): int
|
||||
{
|
||||
$this->startProcess();
|
||||
Process::signal(SIGINT, fn() => $this->exit());
|
||||
Process::signal(SIGTERM, fn() => $this->exit());
|
||||
sleep(3);
|
||||
// if (extension_loaded('inotify')) {
|
||||
// $this->onInotifyReload();
|
||||
// } else {
|
||||
$this->onCrontabReload();
|
||||
// }
|
||||
return 1;
|
||||
}
|
||||
/**
|
||||
* @return void
|
||||
* @throws Exception
|
||||
*/
|
||||
private function onCrontabReload(): void
|
||||
{
|
||||
$this->loadDirs();
|
||||
$this->tick();
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @return void
|
||||
* @throws Exception
|
||||
*/
|
||||
public function exit(): void
|
||||
{
|
||||
$this->stopProcess();
|
||||
}
|
||||
/**
|
||||
* @return void
|
||||
* @throws Exception
|
||||
*/
|
||||
private function onInotifyReload(): void
|
||||
{
|
||||
$init = inotify_init();
|
||||
foreach ($this->dirs as $dir) {
|
||||
if (!is_dir($dir)) {
|
||||
continue;
|
||||
}
|
||||
$this->watch($init, $dir);
|
||||
}
|
||||
Event::add($init, fn() => $this->check($init));
|
||||
Event::cycle(function () use ($init) {
|
||||
$pid = (int)file_get_contents(storage('.swoole.pid'));
|
||||
if ($pid <= 0 || !Process::kill($pid, 0) || $this->isStop()) {
|
||||
Event::del($init);
|
||||
}
|
||||
}, true);
|
||||
Event::wait();
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @return void
|
||||
*/
|
||||
private function startProcess(): void
|
||||
{
|
||||
$this->process = proc_open([PHP_BINARY, APP_PATH . 'kiri.php', 'sw:server', 'restart'], [], $pipes);
|
||||
}
|
||||
/**
|
||||
* @param bool $isReload
|
||||
* @throws Exception
|
||||
*/
|
||||
private function loadDirs(bool $isReload = false): void
|
||||
{
|
||||
foreach ($this->dirs as $value) {
|
||||
if (is_bool($path = realpath($value))) {
|
||||
continue;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return void
|
||||
* @throws Exception
|
||||
*/
|
||||
private function stopProcess(): void
|
||||
{
|
||||
if (!is_resource($this->process)) {
|
||||
return;
|
||||
}
|
||||
$pid = (int)file_get_contents(storage('.swoole.pid'));
|
||||
if (posix_kill($pid, 0)) {
|
||||
posix_kill($pid, SIGTERM);
|
||||
}
|
||||
proc_close($this->process);
|
||||
$this->process = null;
|
||||
}
|
||||
if (!is_dir($path)) continue;
|
||||
|
||||
$this->loadByDir($path, $isReload);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @return void
|
||||
* @throws Exception
|
||||
*/
|
||||
private function onCrontabReload(): void
|
||||
{
|
||||
$this->loadDirs();
|
||||
$this->tick();
|
||||
}
|
||||
/**
|
||||
* @throws Exception
|
||||
*/
|
||||
public function tick(): void
|
||||
{
|
||||
$isReloading = Context::get('isReloading', false);
|
||||
if ($isReloading) {
|
||||
return;
|
||||
}
|
||||
|
||||
$pid = (int)file_get_contents(storage('.swoole.pid'));
|
||||
if ($pid <= 0 || !Process::kill($pid, 0) || $this->isStop()) {
|
||||
return;
|
||||
}
|
||||
|
||||
$this->loadDirs(true);
|
||||
|
||||
sleep(2);
|
||||
|
||||
$this->tick();
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @return void
|
||||
* @throws Exception
|
||||
*/
|
||||
private function onInotifyReload(): void
|
||||
{
|
||||
$init = inotify_init();
|
||||
foreach ($this->dirs as $dir) {
|
||||
if (!is_dir($dir)) {
|
||||
continue;
|
||||
}
|
||||
$this->watch($init, $dir);
|
||||
}
|
||||
Event::add($init, fn() => $this->check($init));
|
||||
Event::cycle(function () use ($init) {
|
||||
$pid = (int)file_get_contents(storage('.swoole.pid'));
|
||||
if ($pid <= 0 || !Process::kill($pid, 0)) {
|
||||
Event::del($init);
|
||||
}
|
||||
}, true);
|
||||
Event::wait();
|
||||
}
|
||||
/**
|
||||
* @param $path
|
||||
* @param bool $isReload
|
||||
* @return void
|
||||
* @throws Exception
|
||||
*/
|
||||
private function loadByDir($path, bool $isReload = false): void
|
||||
{
|
||||
if (!is_string($path)) {
|
||||
return;
|
||||
}
|
||||
$path = rtrim($path, '/');
|
||||
foreach (glob(realpath($path) . '/*') as $value) {
|
||||
if (is_dir($value)) {
|
||||
$this->loadByDir($value, $isReload);
|
||||
}
|
||||
if (is_file($value)) {
|
||||
if ($this->checkFile($value, $isReload)) {
|
||||
$this->timerReload();
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param bool $isReload
|
||||
* @throws Exception
|
||||
*/
|
||||
private function loadDirs(bool $isReload = false): void
|
||||
{
|
||||
foreach ($this->dirs as $value) {
|
||||
if (is_bool($path = realpath($value))) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (!is_dir($path)) continue;
|
||||
|
||||
$this->loadByDir($path, $isReload);
|
||||
}
|
||||
}
|
||||
/**
|
||||
* @param $value
|
||||
* @param $isReload
|
||||
* @return bool
|
||||
*/
|
||||
private function checkFile($value, $isReload): bool
|
||||
{
|
||||
$md5 = md5($value);
|
||||
$mTime = filectime($value);
|
||||
if (!isset($this->md5Map[$md5])) {
|
||||
if ($isReload) {
|
||||
return true;
|
||||
}
|
||||
$this->md5Map[$md5] = $mTime;
|
||||
} else {
|
||||
if ($this->md5Map[$md5] != $mTime) {
|
||||
if ($isReload) {
|
||||
return true;
|
||||
}
|
||||
$this->md5Map[$md5] = $mTime;
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @throws Exception
|
||||
*/
|
||||
public function tick(): void
|
||||
{
|
||||
$isReloading = Context::get('isReloading', false);
|
||||
if ($isReloading) {
|
||||
return;
|
||||
}
|
||||
/**
|
||||
* 开始监听
|
||||
*/
|
||||
public function check($inotify): void
|
||||
{
|
||||
if (!($events = inotify_read($inotify))) {
|
||||
return;
|
||||
}
|
||||
$isReloading = Context::get('isReloading', false);
|
||||
if ($isReloading) {
|
||||
return;
|
||||
}
|
||||
|
||||
$pid = (int)file_get_contents(storage('.swoole.pid'));
|
||||
if ($pid <= 0 || !Process::kill($pid, 0)) {
|
||||
return;
|
||||
}
|
||||
$eventList = [IN_CREATE, IN_DELETE, IN_MODIFY, IN_MOVED_TO, IN_MOVED_FROM];
|
||||
foreach ($events as $ev) {
|
||||
if (empty($ev['name'])) {
|
||||
continue;
|
||||
}
|
||||
if ($ev['mask'] == IN_IGNORED) {
|
||||
continue;
|
||||
}
|
||||
if (!in_array($ev['mask'], $eventList)) {
|
||||
continue;
|
||||
}
|
||||
$fileType = strstr($ev['name'], '.');
|
||||
//非重启类型
|
||||
if ($fileType !== '.php') {
|
||||
continue;
|
||||
}
|
||||
if (Context::get('swoole_timer_after') !== -1) {
|
||||
return;
|
||||
}
|
||||
$int = @swoole_timer_after(2000, fn() => $this->reload($inotify));
|
||||
Context::set('swoole_timer_after', $int);
|
||||
Context::set('isReloading', true);
|
||||
}
|
||||
}
|
||||
|
||||
$this->loadDirs(true);
|
||||
/**
|
||||
* @throws Exception
|
||||
*/
|
||||
public function reload($inotify): void
|
||||
{
|
||||
Context::set('isReloading', true);
|
||||
$this->trigger_reload();
|
||||
|
||||
sleep(2);
|
||||
$this->clearWatch($inotify);
|
||||
foreach ($this->dirs as $root) {
|
||||
$this->watch($inotify, $root);
|
||||
}
|
||||
Context::set('swoole_timer_after', -1);
|
||||
Context::set('isReloading', false);
|
||||
$this->md5Map = [];
|
||||
}
|
||||
|
||||
$this->tick();
|
||||
}
|
||||
/**
|
||||
* @throws Exception
|
||||
*/
|
||||
public function timerReload(): void
|
||||
{
|
||||
Context::set('isReloading', true);
|
||||
$this->trigger_reload();
|
||||
|
||||
Context::set('swoole_timer_after', -1);
|
||||
|
||||
$this->loadDirs();
|
||||
|
||||
Context::set('isReloading', false);
|
||||
|
||||
$this->tick();
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param $path
|
||||
* @param bool $isReload
|
||||
* @return void
|
||||
* @throws Exception
|
||||
*/
|
||||
private function loadByDir($path, bool $isReload = false): void
|
||||
{
|
||||
if (!is_string($path)) {
|
||||
return;
|
||||
}
|
||||
$path = rtrim($path, '/');
|
||||
foreach (glob(realpath($path) . '/*') as $value) {
|
||||
if (is_dir($value)) {
|
||||
$this->loadByDir($value, $isReload);
|
||||
}
|
||||
if (is_file($value)) {
|
||||
if ($this->checkFile($value, $isReload)) {
|
||||
$this->timerReload();
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
/**
|
||||
* 重启
|
||||
* @throws Exception
|
||||
*/
|
||||
public function trigger_reload(): void
|
||||
{
|
||||
echo 'tigger server Reload' . PHP_EOL;
|
||||
di(ServerInterface::class)->reload(false);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param $value
|
||||
* @param $isReload
|
||||
* @return bool
|
||||
*/
|
||||
private function checkFile($value, $isReload): bool
|
||||
{
|
||||
$md5 = md5($value);
|
||||
$mTime = filectime($value);
|
||||
if (!isset($this->md5Map[$md5])) {
|
||||
if ($isReload) {
|
||||
return true;
|
||||
}
|
||||
$this->md5Map[$md5] = $mTime;
|
||||
} else {
|
||||
if ($this->md5Map[$md5] != $mTime) {
|
||||
if ($isReload) {
|
||||
return true;
|
||||
}
|
||||
$this->md5Map[$md5] = $mTime;
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
/**
|
||||
* @throws Exception
|
||||
*/
|
||||
public function clearWatch($inotify): void
|
||||
{
|
||||
foreach ($this->watchFiles as $wd) {
|
||||
try {
|
||||
inotify_rm_watch($inotify, $wd);
|
||||
} catch (\Throwable $exception) {
|
||||
logger()->addError($exception, 'throwable');
|
||||
}
|
||||
}
|
||||
$this->watchFiles = [];
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 开始监听
|
||||
*/
|
||||
public function check($inotify): void
|
||||
{
|
||||
if (!($events = inotify_read($inotify))) {
|
||||
return;
|
||||
}
|
||||
$isReloading = Context::get('isReloading', false);
|
||||
if ($isReloading) {
|
||||
return;
|
||||
}
|
||||
/**
|
||||
* @param $inotify
|
||||
* @param $dir
|
||||
* @return bool
|
||||
* @throws Exception
|
||||
*/
|
||||
public function watch($inotify, $dir): bool
|
||||
{
|
||||
//目录不存在
|
||||
if (!is_dir($dir)) {
|
||||
return logger()->addError("[$dir] is not a directory.");
|
||||
}
|
||||
//避免重复监听
|
||||
if (isset($this->watchFiles[$dir])) {
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
$eventList = [IN_CREATE, IN_DELETE, IN_MODIFY, IN_MOVED_TO, IN_MOVED_FROM];
|
||||
foreach ($events as $ev) {
|
||||
if (empty($ev['name'])) {
|
||||
continue;
|
||||
}
|
||||
if ($ev['mask'] == IN_IGNORED) {
|
||||
continue;
|
||||
}
|
||||
if (!in_array($ev['mask'], $eventList)) {
|
||||
continue;
|
||||
}
|
||||
$fileType = strstr($ev['name'], '.');
|
||||
//非重启类型
|
||||
if ($fileType !== '.php') {
|
||||
continue;
|
||||
}
|
||||
if (Context::get('swoole_timer_after') !== -1) {
|
||||
return;
|
||||
}
|
||||
$int = @swoole_timer_after(2000, fn() => $this->reload($inotify));
|
||||
Context::set('swoole_timer_after', $int);
|
||||
Context::set('isReloading', true);
|
||||
}
|
||||
}
|
||||
if (in_array($dir, [APP_PATH . 'commands', APP_PATH . '.git', APP_PATH . '.gitee'])) {
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
/**
|
||||
* @throws Exception
|
||||
*/
|
||||
public function reload($inotify): void
|
||||
{
|
||||
Context::set('isReloading', true);
|
||||
$this->trigger_reload();
|
||||
$wd = @inotify_add_watch($inotify, $dir, IN_MODIFY | IN_DELETE | IN_CREATE | IN_MOVE);
|
||||
$this->watchFiles[$dir] = $wd;
|
||||
|
||||
$this->clearWatch($inotify);
|
||||
foreach ($this->dirs as $root) {
|
||||
$this->watch($inotify, $root);
|
||||
}
|
||||
Context::set('swoole_timer_after', -1);
|
||||
Context::set('isReloading', false);
|
||||
$this->md5Map = [];
|
||||
}
|
||||
$files = scandir($dir);
|
||||
foreach ($files as $f) {
|
||||
if ($f == '.' or $f == '..' or $f == 'runtime' or preg_match('/\.txt/', $f) or preg_match('/\.sql/', $f) or preg_match('/\.log/', $f)) {
|
||||
continue;
|
||||
}
|
||||
$path = $dir . '/' . $f;
|
||||
//递归目录
|
||||
if (is_dir($path)) {
|
||||
$this->watch($inotify, $path);
|
||||
}
|
||||
|
||||
/**
|
||||
* @throws Exception
|
||||
*/
|
||||
public function timerReload(): void
|
||||
{
|
||||
Context::set('isReloading', true);
|
||||
$this->trigger_reload();
|
||||
|
||||
Context::set('swoole_timer_after', -1);
|
||||
|
||||
$this->loadDirs();
|
||||
|
||||
Context::set('isReloading', false);
|
||||
|
||||
$this->tick();
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 重启
|
||||
* @throws Exception
|
||||
*/
|
||||
public function trigger_reload(): void
|
||||
{
|
||||
echo 'tigger server Reload' . PHP_EOL;
|
||||
$this->stopProcess();
|
||||
$this->startProcess();
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @throws Exception
|
||||
*/
|
||||
public function clearWatch($inotify): void
|
||||
{
|
||||
foreach ($this->watchFiles as $wd) {
|
||||
try {
|
||||
inotify_rm_watch($inotify, $wd);
|
||||
} catch (\Throwable $exception) {
|
||||
logger()->addError($exception, 'throwable');
|
||||
}
|
||||
}
|
||||
$this->watchFiles = [];
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param $inotify
|
||||
* @param $dir
|
||||
* @return bool
|
||||
* @throws Exception
|
||||
*/
|
||||
public function watch($inotify, $dir): bool
|
||||
{
|
||||
//目录不存在
|
||||
if (!is_dir($dir)) {
|
||||
return logger()->addError("[$dir] is not a directory.");
|
||||
}
|
||||
//避免重复监听
|
||||
if (isset($this->watchFiles[$dir])) {
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
if (in_array($dir, [APP_PATH . 'commands', APP_PATH . '.git', APP_PATH . '.gitee'])) {
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
$wd = @inotify_add_watch($inotify, $dir, IN_MODIFY | IN_DELETE | IN_CREATE | IN_MOVE);
|
||||
$this->watchFiles[$dir] = $wd;
|
||||
|
||||
$files = scandir($dir);
|
||||
foreach ($files as $f) {
|
||||
if ($f == '.' or $f == '..' or $f == 'runtime' or preg_match('/\.txt/', $f) or preg_match('/\.sql/', $f) or preg_match('/\.log/', $f)) {
|
||||
continue;
|
||||
}
|
||||
$path = $dir . '/' . $f;
|
||||
//递归目录
|
||||
if (is_dir($path)) {
|
||||
$this->watch($inotify, $path);
|
||||
}
|
||||
|
||||
//检测文件类型
|
||||
if (strstr($f, '.') == '.php') {
|
||||
$wd = @inotify_add_watch($inotify, $path, IN_MODIFY | IN_DELETE | IN_CREATE | IN_MOVE);
|
||||
$this->watchFiles[$path] = $wd;
|
||||
}
|
||||
}
|
||||
return TRUE;
|
||||
}
|
||||
//检测文件类型
|
||||
if (strstr($f, '.') == '.php') {
|
||||
$wd = @inotify_add_watch($inotify, $path, IN_MODIFY | IN_DELETE | IN_CREATE | IN_MOVE);
|
||||
$this->watchFiles[$path] = $wd;
|
||||
}
|
||||
}
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user