as2252258/kiri-core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

modify

Browse Source
This commit is contained in:
as2252258@163.com
2021-07-26 02:11:12 +08:00
parent 6d49bffcfe
commit 633cd1b140
1 changed files with 245 additions and 254 deletions
Download Patch File Download Diff File Expand all files Collapse all files
System/Process/ServerInotify.php
+245 -254
View File
@@ -24,302 +24,293 @@ use Swoole\Timer;
*/ */
class ServerInotify implements CustomProcess class ServerInotify implements CustomProcess
{ {
private mixed $inotify; private mixed $inotify;
private bool $isReloading = false; private bool $isReloading = false;
private bool $isReloadingOut = false; private bool $isReloadingOut = false;
private array $watchFiles = []; private array $watchFiles = [];
private ?array $dirs = []; private ?array $dirs = [];
private int $events; private int $events;
private int $int = -1; private int $int = -1;
/** /**
* @param \Swoole\Process $process * @param \Swoole\Process $process
* @return string * @return string
*/ */
public function getProcessName(\Swoole\Process $process): string public function getProcessName(\Swoole\Process $process): string
{ {
return 'file change process'; return 'file change process';
} }
/** /**
* @param \Swoole\Process $process * @param \Swoole\Process $process
* @throws Exception * @throws Exception
*/ */
public function onHandler(\Swoole\Process $process): void public function onHandler(\Swoole\Process $process): void
{ {
// TODO: Implement before() method. // TODO: Implement before() method.
set_error_handler([$this, 'onErrorHandler']); set_error_handler([$this, 'onErrorHandler']);
$this->dirs = Config::get('inotify', [APP_PATH]); $this->dirs = Config::get('inotify', [APP_PATH]);
if (extension_loaded('inotify')) { if (extension_loaded('inotify')) {
$this->inotify = inotify_init(); $this->inotify = inotify_init();
$this->events = IN_MODIFY | IN_DELETE | IN_CREATE | IN_MOVE; $this->events = IN_MODIFY | IN_DELETE | IN_CREATE | IN_MOVE;
foreach ($this->dirs as $dir) { foreach ($this->dirs as $dir) {
if (!is_dir($dir)) continue; if (!is_dir($dir)) continue;
$this->watch($dir); $this->watch($dir);
} }
Event::add($this->inotify, [$this, 'check']); Event::add($this->inotify, [$this, 'check']);
Event::wait(); Event::wait();
} else { } else {
$this->loadDirs(); $this->loadDirs();
$this->tick(); $this->tick();
} }
} }
/** /**
* @param bool $isReload * @param bool $isReload
* @throws Exception * @throws Exception
*/ */
private function loadDirs(bool $isReload = false) private function loadDirs(bool $isReload = false)
{ {
foreach ($this->dirs as $value) { foreach ($this->dirs as $value) {
if (is_bool($path = realpath($value))) { if (is_bool($path = realpath($value))) {
continue; continue;
} }
if (!is_dir($path)) continue; if (!is_dir($path)) continue;
$this->loadByDir($path, $isReload); $this->loadByDir($path, $isReload);
} }
} }
private array $md5Map = []; private array $md5Map = [];
/** /**
* @throws Exception * @throws Exception
*/ */
public function tick() public function tick()
{ {
if ($this->isReloading) { if ($this->isReloading) {
return; return;
} }
$this->loadDirs(true); $this->loadDirs(true);
Timer::after(2000, [$this, 'tick']); Timer::after(2000, [$this, 'tick']);
} }
/** /**
* @param $path * @param $path
* @param bool $isReload * @param bool $isReload
* @return void * @return void
* @throws Exception * @throws Exception
*/ */
private function loadByDir($path, bool $isReload = false): void private function loadByDir($path, bool $isReload = false): void
{ {
if (!is_string($path)) { if (!is_string($path)) {
return; return;
} }
$path = rtrim($path, '/'); $path = rtrim($path, '/');
foreach (glob(realpath($path) . '/*') as $value) { foreach (glob(realpath($path) . '/*') as $value) {
if (is_dir($value)) { if (is_dir($value)) {
$this->loadByDir($value, $isReload); $this->loadByDir($value, $isReload);
} }
if (is_file($value)) { if (is_file($value)) {
if ($this->checkFile($value, $isReload)) { if ($this->checkFile($value, $isReload)) {
$this->timerReload(); $this->timerReload();
break; break;
} }
} }
} }
} }
/** /**
* @param $value * @param $value
* @param $isReload * @param $isReload
* @return bool * @return bool
*/ */
private function checkFile($value, $isReload): bool private function checkFile($value, $isReload): bool
{ {
$md5 = md5($value); $md5 = md5($value);
$mTime = filectime($value); $mTime = filectime($value);
if (!isset($this->md5Map[$md5])) { if (!isset($this->md5Map[$md5])) {
if ($isReload) { if ($isReload) {
return true; return true;
} }
$this->md5Map[$md5] = $mTime; $this->md5Map[$md5] = $mTime;
} else { } else {
if ($this->md5Map[$md5] != $mTime) { if ($this->md5Map[$md5] != $mTime) {
if ($isReload) { if ($isReload) {
return true; return true;
} }
$this->md5Map[$md5] = $mTime; $this->md5Map[$md5] = $mTime;
} }
} }
return false; return false;
} }
/** /**
* 开始监听 * 开始监听
*/ */
public function check() public function check()
{ {
if (!($events = inotify_read($this->inotify))) { if (!($events = inotify_read($this->inotify))) {
return; return;
} }
if ($this->isReloading) { if ($this->isReloading) {
if (!$this->isReloadingOut) { if (!$this->isReloadingOut) {
$this->isReloadingOut = true; $this->isReloadingOut = true;
} }
return; return;
} }
$eventList = [IN_CREATE, IN_DELETE, IN_MODIFY, IN_MOVED_TO, IN_MOVED_FROM]; foreach ($events as $ev) {
foreach ($events as $ev) { if (!in_array($ev['mask'], [IN_CREATE, IN_DELETE, IN_MODIFY, IN_MOVED_TO, IN_MOVED_FROM])) {
if (empty($ev['name'])) { continue;
continue; }
} //非重启类型
if ($ev['mask'] == IN_IGNORED) { if (str_ends_with($ev['name'], '.php')) {
continue; if ($this->int !== -1) {
} return;
if (!in_array($ev['mask'], $eventList)) { }
continue; $this->int = @swoole_timer_after(2000, [$this, 'reload']);
}
$fileType = strstr($ev['name'], '.');
//非重启类型
if ($fileType !== '.php') {
continue;
}
if ($this->int !== -1) {
return;
}
$this->int = @swoole_timer_after(2000, [$this, 'reload']);
$this->isReloading = true; $this->isReloading = true;
} }
} }
}
/** /**
* @throws Exception * @throws Exception
*/ */
public function reload() public function reload()
{ {
$this->isReloading = true; $this->isReloading = true;
$this->trigger_reload(); $this->trigger_reload();
$this->clearWatch(); $this->clearWatch();
foreach ($this->dirs as $root) { foreach ($this->dirs as $root) {
$this->watch($root); $this->watch($root);
} }
$this->int = -1; $this->int = -1;
$this->isReloading = FALSE; $this->isReloading = FALSE;
$this->isReloadingOut = FALSE; $this->isReloadingOut = FALSE;
$this->md5Map = []; $this->md5Map = [];
} }
/** /**
* @throws Exception * @throws Exception
*/ */
public function timerReload() public function timerReload()
{ {
$this->isReloading = true; $this->isReloading = true;
$this->trigger_reload(); $this->trigger_reload();
$this->int = -1; $this->int = -1;
$this->loadDirs(); $this->loadDirs();
$this->isReloading = FALSE; $this->isReloading = FALSE;
$this->isReloadingOut = FALSE; $this->isReloadingOut = FALSE;
$this->tick(); $this->tick();
} }
/** /**
* 重启 * 重启
* @throws Exception * @throws Exception
*/ */
public function trigger_reload() public function trigger_reload()
{ {
exec(PHP_BINARY . ' ' . APP_PATH . 'snowflake runtime:builder', $output); exec(PHP_BINARY . ' ' . APP_PATH . 'snowflake runtime:builder', $output);
var_dump(implode(PHP_EOL, $output)); var_dump(implode(PHP_EOL, $output));
Snowflake::reload(); Snowflake::reload();
} }
/** /**
* @throws Exception * @throws Exception
*/ */
public function clearWatch() public function clearWatch()
{ {
foreach ($this->watchFiles as $wd) { foreach ($this->watchFiles as $wd) {
try { try {
inotify_rm_watch($this->inotify, $wd); inotify_rm_watch($this->inotify, $wd);
} catch (\Throwable $exception) { } catch (\Throwable $exception) {
logger()->addError($exception, 'throwable'); logger()->addError($exception, 'throwable');
} }
} }
$this->watchFiles = []; $this->watchFiles = [];
} }
/** /**
* @param $code * @param $code
* @param $message * @param $message
* @param $file * @param $file
* @param $line * @param $line
* @throws Exception * @throws Exception
*/ */
protected function onErrorHandler($code, $message, $file, $line) protected function onErrorHandler($code, $message, $file, $line)
{ {
if (str_contains($message, 'The file descriptor is not an inotify instance')) { if (str_contains($message, 'The file descriptor is not an inotify instance')) {
return; return;
} }
debug('Error:' . $message . ' at ' . $file . ':' . $line); debug('Error:' . $message . ' at ' . $file . ':' . $line);
} }
/** /**
* @param $dir * @param $dir
* @return bool * @return bool
* @throws Exception * @throws Exception
*/ */
public function watch($dir): bool public function watch($dir): bool
{ {
//目录不存在 //目录不存在
if (!is_dir($dir)) { if (!is_dir($dir)) {
return logger()->addError("[$dir] is not a directory."); return logger()->addError("[$dir] is not a directory.");
} }
//避免重复监听 //避免重复监听
if (isset($this->watchFiles[$dir])) { if (isset($this->watchFiles[$dir])) {
return FALSE; return FALSE;
} }
if (in_array($dir, [APP_PATH . 'commands', APP_PATH . '.git', APP_PATH . '.gitee'])) { if (in_array($dir, [APP_PATH . 'commands', APP_PATH . '.git', APP_PATH . '.gitee'])) {
return FALSE; return FALSE;
} }
$wd = @inotify_add_watch($this->inotify, $dir, $this->events); $wd = @inotify_add_watch($this->inotify, $dir, $this->events);
$this->watchFiles[$dir] = $wd; $this->watchFiles[$dir] = $wd;
$files = scandir($dir); $files = scandir($dir);
foreach ($files as $f) { 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)) { if (!is_dir($f) && !str_ends_with($f, '.php')) {
continue; continue;
} }
$path = $dir . '/' . $f; $path = $dir . '/' . $f;
//递归目录 //递归目录
if (is_dir($path)) { if (is_dir($path)) {
$this->watch($path); $this->watch($path);
} }
//检测文件类型 //检测文件类型
if (strstr($f, '.') == '.php') { if (strstr($f, '.') == '.php') {
$wd = @inotify_add_watch($this->inotify, $path, $this->events); $wd = @inotify_add_watch($this->inotify, $path, $this->events);
$this->watchFiles[$path] = $wd; $this->watchFiles[$path] = $wd;
} }
} }
return TRUE; return TRUE;
} }
} }