e

http-helper/Shutdown.php

@@ -0,0 +1,181 @@
+<?php
+
+
+namespace Http;
+
+
+use Exception;
+use Server\ServerManager;
+use Kiri\Abstracts\Component;
+
+
+/**
+ * Class Shutdown
+ * @package Http
+ */
+class Shutdown extends Component
+{
+
+
+	private string $taskDirectory;
+	private string $workerDirectory;
+	private string $managerDirectory;
+	private string $processDirectory;
+
+
+	private array $_pids = [];
+
+
+	/**
+	 * @throws Exception
+	 */
+	public function init()
+	{
+		$this->taskDirectory = storage(null, 'pid/task');
+		$this->workerDirectory = storage(null, 'pid/worker');
+		$this->managerDirectory = storage(null, 'pid/manager');
+		$this->processDirectory = storage(null, 'pid/process');
+	}
+
+
+	/**
+	 * @throws Exception
+	 */
+	public function shutdown(): void
+	{
+		clearstatcache(storage());
+		$output = shell_exec('[ -f /.dockerenv ] && echo yes || echo no');
+		if (trim($output) === 'yes') {
+			return;
+		}
+
+		$server = ServerManager::getContext()->getServer();
+		$master_pid = $server->setting['pid_file'] ?? PID_PATH;
+		if (file_exists($master_pid)) {
+			$this->close(file_get_contents($master_pid));
+		}
+		$this->closeOther();
+	}
+
+
+	/**
+	 * 关闭其他进程
+	 */
+	private function closeOther(): void
+	{
+		$this->directoryCheck($this->managerDirectory);
+		$this->directoryCheck($this->taskDirectory);
+		$this->directoryCheck($this->workerDirectory);
+		$this->directoryCheck($this->processDirectory);
+	}
+
+
+	/**
+	 * @return bool
+	 * @throws Exception
+	 * check server is running.
+	 */
+	public function isRunning(): bool
+	{
+		$server = ServerManager::getContext()->getServer();
+		$master_pid = $server->setting['pid_file'] ?? PID_PATH;
+
+		if (!file_exists($master_pid)) {
+			return false;
+		}
+
+		return $this->pidIsExists(file_get_contents($master_pid));
+	}
+
+
+	/**
+	 * @param $content
+	 * @return bool
+	 */
+	public function pidIsExists($content): bool
+	{
+		if (intval($content) < 1) {
+			return false;
+		}
+		exec('ps -eo pid', $output);
+		$output = array_filter($output, function ($value) {
+			return intval($value);
+		});
+		return in_array(intval($content), $output);
+	}
+
+
+	/**
+	 * @param string $path
+	 * @return bool
+	 */
+	public function directoryCheck(string $path): bool
+	{
+		$values = $this->getProcessPidS($path);
+		if (empty($values)) return false;
+
+		$diff = array_diff($values, $this->getPidS());
+		foreach ($diff as $value) {
+			$this->pidIsExists($value);
+		}
+		return false;
+	}
+
+
+	/**
+	 * @param $path
+	 * @return array|bool
+	 */
+	private function getProcessPidS($path): bool|array
+	{
+		$values = [];
+		$dir = new \DirectoryIterator($path);
+		if ($dir->getSize() < 1) {
+			return $values;
+		}
+		foreach ($dir as $value) {
+			if ($value->isDot()) continue;
+
+			if (!$value->valid()) continue;
+
+			$_value = file_get_contents($value->getRealPath());
+			if (empty($_value)) {
+				continue;
+			}
+			$values[] = intval($_value);
+
+			@unlink($value->getRealPath());
+		}
+		return $values;
+	}
+
+
+	/**
+	 * @return array
+	 */
+	private function getPidS(): array
+	{
+		exec('ps -eo pid', $output);
+		return array_filter($output, function ($value) {
+			return intval($value);
+		});
+	}
+
+
+	/**
+	 * @param string $value
+	 */
+	public function close(mixed $value)
+	{
+		while ($this->pidIsExists($value)) {
+			exec('kill -15 ' . $value);
+			usleep(100);
+		}
+		clearstatcache($value);
+		if (file_exists($value)) {
+			@unlink($value);
+		}
+	}
+
+
+}