This commit is contained in:
2021-12-11 05:33:00 +08:00
parent ebb7ac9673
commit 8870a7ca27
3 changed files with 286 additions and 281 deletions
+166 -163
View File
@@ -25,212 +25,215 @@ class HotReload extends Command
{ {
public bool $isReloading = false; public bool $isReloading = FALSE;
public bool $isReloadingOut = false; public bool $isReloadingOut = FALSE;
public ?array $dirs = []; public ?array $dirs = [];
public int $events; public int $events;
public int $int = -1; public int $int = -1;
private ?Process $process = null; private ?Process $process = NULL;
public Inotify|Scaner $driver; public Inotify|Scaner $driver;
#[Inject(Logger::class)] #[Inject(Logger::class)]
public Logger $logger; public Logger $logger;
protected mixed $source = null; protected mixed $source = NULL;
protected mixed $pipes = []; protected mixed $pipes = [];
protected ?Coroutine\Channel $channel = null; protected ?Coroutine\Channel $channel = NULL;
/** /**
*/ */
protected function configure() protected function configure()
{ {
$this->setName('sw:wather')->setDescription('server start') $this->setName('sw:wather')->setDescription('server start')
->addOption('daemon', null, InputOption::VALUE_OPTIONAL, "是否后台运行", "--front"); ->addOption('daemon', NULL, InputOption::VALUE_OPTIONAL, "是否后台运行", "--front");
} }
/** /**
* @throws ConfigException * @throws ConfigException
* @throws \ReflectionException * @throws \ReflectionException
* @throws Exception * @throws Exception
*/ */
protected function initCore() protected function initCore()
{ {
$this->dirs = Config::get('inotify', [APP_PATH . 'app']); $this->dirs = Config::get('inotify', [APP_PATH . 'app']);
if (!extension_loaded('inotify')) { if (!extension_loaded('inotify')) {
$this->driver = Kiri::getDi()->make(Scaner::class, [$this->dirs, $this]); $this->driver = Kiri::getDi()->make(Scaner::class, [$this->dirs, $this]);
} else { } else {
$this->driver = Kiri::getDi()->make(Inotify::class, [$this->dirs, $this]); $this->driver = Kiri::getDi()->make(Inotify::class, [$this->dirs, $this]);
} }
$this->clearOtherService(); $this->clearOtherService();
$this->setProcessName(); $this->setProcessName();
} }
/** /**
* @throws ConfigException * @throws ConfigException
*/ */
public function setProcessName() public function setProcessName()
{ {
swoole_async_set(['enable_coroutine' => false]); swoole_async_set(['enable_coroutine' => FALSE]);
set_error_handler([$this, 'errorHandler']); set_error_handler([$this, 'errorHandler']);
if (Kiri::getPlatform()->isLinux()) { if (Kiri::getPlatform()->isLinux()) {
swoole_set_process_name('[' . Config::get('id', 'sw service.') . '].sw:wather'); swoole_set_process_name('[' . Config::get('id', 'sw service.') . '].sw:wather');
} }
} }
/** /**
* @throws Exception * @throws Exception
*/ */
public function clearOtherService() public function clearOtherService()
{ {
if (file_exists(storage('.manager.pid'))) { if (file_exists(storage('.manager.pid'))) {
$pid = (int)file_get_contents(storage('.manager.pid')); $pid = (int)file_get_contents(storage('.manager.pid'));
if ($pid > 0 && Process::kill($pid, 0)) { if ($pid > 0 && Process::kill($pid, 0)) {
Process::kill($pid, 15) && Process::wait(true); Process::kill($pid, 15) && Process::wait(TRUE);
} }
} }
file_put_contents(storage('.manager.pid'), getmypid()); file_put_contents(storage('.manager.pid'), getmypid());
} }
/** /**
* @throws Exception * @throws Exception
*/ */
public function errorHandler() public function errorHandler()
{ {
$error = func_get_args(); $error = func_get_args();
$path = ['file' => $error[2], 'line' => $error[3]]; $path = ['file' => $error[2], 'line' => $error[3]];
if ($error[0] === 0) { if ($error[0] === 0) {
$error[0] = 500; $error[0] = 500;
} }
$data = Json::to(500, $error[1], $path); $data = Json::to(500, $error[1], $path);
$this->logger->error($data, 'error'); $this->logger->error($data, 'error');
} }
/** /**
* @param InputInterface $input * @param InputInterface $input
* @param OutputInterface $output * @param OutputInterface $output
* @return int * @return int
* @throws ConfigException * @throws ConfigException
* @throws Exception * @throws Exception
*/ */
public function execute(InputInterface $input, OutputInterface $output): int public function execute(InputInterface $input, OutputInterface $output): int
{ {
$this->initCore(); $this->initCore();
$this->trigger_reload(); $this->trigger_reload();
if ($input->getOption('') == '--daemon') { if ($input->getOption('') == '--daemon') {
Process::daemon(true, true); Process::daemon(TRUE, TRUE);
} }
Timer::tick(1000, fn() => $this->healthCheck()); Timer::tick(1000, fn() => $this->healthCheck());
Process::signal(SIGTERM, [$this, 'onSignal']); Process::signal(SIGTERM, [$this, 'onSignal']);
Process::signal(SIGKILL, [$this, 'onSignal']); Process::signal(SIGKILL, [$this, 'onSignal']);
$this->driver->start(); $this->driver->start();
return 0; return 0;
} }
/** /**
* @throws Exception * @throws Exception
*/ */
public function healthCheck() public function healthCheck()
{ {
$pid = (int)file_get_contents(storage('.swoole.pid')); $pid = (int)file_get_contents(storage('.swoole.pid'));
if (empty($pid)) { if ($this->driver->isReloading) {
$this->logger->warning('service is shutdown you need reload.'); return;
$this->trigger_reload(); }
} else if (!Process::kill($pid, 0)) { if (empty($pid)) {
$this->logger->warning('service is shutdown you need reload.'); $this->logger->warning('service is shutdown you need reload.');
$this->trigger_reload(); $this->trigger_reload();
} } else if (!Process::kill($pid, 0)) {
} $this->logger->warning('service is shutdown you need reload.');
$this->trigger_reload();
}
}
/** /**
* @param $data * @param $data
* @throws Exception * @throws Exception
*/ */
public function onSignal($data) public function onSignal($data)
{ {
if (!$data) { if (!$data) {
return; return;
} }
Timer::clearAll(); Timer::clearAll();
$this->driver->clear(); $this->driver->clear();
$this->stopServer(); $this->stopServer();
$this->stopManager(); $this->stopManager();
while ($ret = Process::wait(true)) { while ($ret = Process::wait(TRUE)) {
echo "PID={$ret['pid']}\n"; echo "PID={$ret['pid']}\n";
sleep(1); sleep(1);
} }
} }
/** /**
* @throws Exception * @throws Exception
*/ */
protected function stopServer() protected function stopServer()
{ {
$pid = file_get_contents(storage('.swoole.pid')); $pid = file_get_contents(storage('.swoole.pid'));
if (!empty($pid) && Process::kill($pid, 0)) { if (!empty($pid) && Process::kill($pid, 0)) {
Process::kill($pid, SIGTERM); Process::kill($pid, SIGTERM);
} }
} }
/** /**
* *
*/ */
protected function stopManager() protected function stopManager()
{ {
if ($this->process && Process::kill($this->process->pid, 0)) { if ($this->process && Process::kill($this->process->pid, 0)) {
Process::kill($this->process->pid) && Process::wait(true); Process::kill($this->process->pid) && Process::wait(TRUE);
} }
} }
/** /**
* 重启 * 重启
* *
* @throws Exception * @throws Exception
*/ */
public function trigger_reload() public function trigger_reload()
{ {
if ($this->int == 1) { if ($this->int == 1) {
return; return;
} }
$this->int = 1; $this->int = 1;
$this->logger->warning('change reload'); $this->logger->warning('change reload');
$this->stopServer(); $this->stopServer();
$this->stopManager(); $this->stopManager();
$this->process = new Process(function (Process $process) { $this->process = new Process(function (Process $process) {
$process->exec(PHP_BINARY, [APP_PATH . "kiri.php", "sw:server", "restart"]); $process->exec(PHP_BINARY, [APP_PATH . "kiri.php", "sw:server", "restart"]);
}); });
$this->process->start(); $this->process->start();
$this->int = -1; $this->int = -1;
} }
} }
+1 -1
View File
@@ -15,7 +15,7 @@ class Inotify
private array $watchFiles = []; private array $watchFiles = [];
protected bool $isReloading = FALSE; public bool $isReloading = FALSE;
protected int $cid; protected int $cid;
+119 -117
View File
@@ -3,145 +3,147 @@
namespace Kiri\FileListen; namespace Kiri\FileListen;
use Exception; use Exception;
use Swoole\Timer;
class Scaner class Scaner
{ {
private array $md5Map = []; private array $md5Map = [];
/** public bool $isReloading = FALSE;
* @param array $dirs
* @param HotReload $process
*/
public function __construct(protected array $dirs, public HotReload $process)
{
}
/** /**
* @throws Exception * @param array $dirs
*/ * @param HotReload $process
public function start(): void */
{ public function __construct(protected array $dirs, public HotReload $process)
$this->loadDirs(); {
$this->tick(); }
}
/** /**
* @param bool $isReload * @throws Exception
* @throws Exception */
*/ public function start(): void
private function loadDirs(bool $isReload = false) {
{ $this->loadDirs();
foreach ($this->dirs as $value) { $this->tick();
if (is_bool($path = realpath($value))) { }
continue;
}
if (!is_dir($path)) continue;
$this->loadByDir($path, $isReload);
}
}
/** /**
* @param $path * @param bool $isReload
* @param bool $isReload * @throws Exception
* @return void */
* @throws Exception private function loadDirs(bool $isReload = FALSE)
*/ {
private function loadByDir($path, bool $isReload = false): void foreach ($this->dirs as $value) {
{ if (is_bool($path = realpath($value))) {
if (!is_string($path)) { continue;
return; }
}
$path = rtrim($path, '/'); if (!is_dir($path)) continue;
foreach (glob(realpath($path) . '/*') as $value) {
if (is_dir($value)) { $this->loadByDir($path, $isReload);
$this->loadByDir($value, $isReload); }
} }
if (is_file($value)) {
if ($this->checkFile($value, $isReload)) {
$this->timerReload();
break;
}
}
}
}
/** /**
* @param $value * @param $path
* @param $isReload * @param bool $isReload
* @return bool * @return void
*/ * @throws Exception
private function checkFile($value, $isReload): bool */
{ private function loadByDir($path, bool $isReload = FALSE): void
$md5 = md5($value); {
$mTime = filectime($value); if (!is_string($path)) {
if (!isset($this->md5Map[$md5])) { return;
if ($isReload) { }
return true; $path = rtrim($path, '/');
} foreach (glob(realpath($path) . '/*') as $value) {
$this->md5Map[$md5] = $mTime; if (is_dir($value)) {
} else { $this->loadByDir($value, $isReload);
if ($this->md5Map[$md5] != $mTime) { }
if ($isReload) { if (is_file($value)) {
return true; if ($this->checkFile($value, $isReload)) {
} Timer::after(2000, fn() => $this->timerReload());
$this->md5Map[$md5] = $mTime; $this->isReloading = TRUE;
} break;
} }
return false; }
} }
}
/** /**
* @throws Exception * @param $value
*/ * @param $isReload
public function timerReload() * @return bool
{ */
if ($this->process->isReloading) { private function checkFile($value, $isReload): bool
return; {
} $md5 = md5($value);
$this->process->isReloading = true; $mTime = filectime($value);
$this->process->trigger_reload(); if (!isset($this->md5Map[$md5])) {
if ($isReload) {
$this->process->int = -1; return TRUE;
}
$this->loadDirs(); $this->md5Map[$md5] = $mTime;
} else {
$this->process->isReloading = FALSE; if ($this->md5Map[$md5] != $mTime) {
$this->process->isReloadingOut = FALSE; if ($isReload) {
return TRUE;
$this->tick(); }
} $this->md5Map[$md5] = $mTime;
}
}
return FALSE;
}
private bool $isStop = false; /**
* @throws Exception
*/
public function timerReload()
{
$this->isReloading = TRUE;
$this->process->trigger_reload();
public function clear() $this->process->int = -1;
{
$this->isStop = true; $this->loadDirs();
}
$this->isReloading = FALSE;
$this->process->isReloadingOut = FALSE;
$this->tick();
}
/** private bool $isStop = FALSE;
* @throws Exception
*/
public function tick()
{
if ($this->process->isReloading || $this->isStop) {
return;
}
$this->loadDirs(true); public function clear()
{
$this->isStop = TRUE;
}
sleep(2);
$this->tick(); /**
} * @throws Exception
*/
public function tick()
{
if ($this->isReloading || $this->isStop) {
return;
}
$this->loadDirs(TRUE);
sleep(2);
$this->tick();
}
} }