改名

HttpServer/Action.php

@@ -0,0 +1,152 @@
+<?php
+
+
+namespace HttpServer;
+
+
+use Exception;
+use Snowflake\Snowflake;
+use Swoole\WebSocket\Server;
+
+/**
+ * Class Action
+ * @package HttpServer
+ */
+trait Action
+{
+
+	/**
+	 * @param $argv
+	 * @return Server
+	 * @throws
+	 */
+	public function getSwooleServer($argv)
+	{
+		/** @var \HttpServer\Server $socket */
+		$socket = Snowflake::app()->get('server');
+//		if (isset($argv[2])) {
+//			$this->modify($argv, $socket);
+//		}
+
+		if (!isset($argv[1])) $argv[1] = 'start';
+
+		return $this->checkAction($argv, $socket);
+	}
+
+	/**
+	 * @param $argv
+	 * @param $socket
+	 * @return Server
+	 * @throws Exception
+	 */
+	private function checkAction($argv, $socket)
+	{
+		if (!in_array($argv[1], ['stop', 'start', 'restart'])) {
+			exit($this->error('action not exists.'));
+		}
+		return $this->{$argv[1]}($socket);
+	}
+
+	/**
+	 * @param \HttpServer\Server $socket
+	 * @return mixed
+	 * @throws Exception
+	 */
+	public function restart($socket)
+	{
+		$this->_shutdown($socket);
+
+		return $this->start();
+	}
+
+
+	/**
+	 * @param \HttpServer\Server $socket
+	 * @throws Exception
+	 */
+	public function stop($socket)
+	{
+		$this->_shutdown($socket);
+	}
+
+	/**
+	 * @param $server
+	 * @return void
+	 * @throws Exception
+	 */
+	private function _shutdown($server)
+	{
+		$socket = storage('socket.sock');
+		if (!file_exists($socket)) {
+			$this->close($server);
+		} else {
+			$pathId = file_get_contents($socket);
+			@unlink($socket);
+
+			if (empty($pathId)) {
+				$this->close($server);
+			} else {
+				exec("kill -TERM $pathId");
+				$this->close($server);
+			}
+		}
+	}
+
+	/**
+	 * @param \HttpServer\Server $server
+	 * @return void
+	 * @throws Exception
+	 */
+	private function close($server)
+	{
+		echo 'waite.';
+		while ($server->isRunner()) {
+			echo '.';
+			$pods = glob(storage('workerIds') . '/*');
+			if (count($pods) < 1) {
+				break;
+			}
+			foreach ($pods as $pid) {
+				if (!file_exists($pid)) {
+					continue;
+				}
+				$content = file_get_contents($pid);
+				exec("ps -ax | awk '{ print $1 }' | grep -e '^{$content}$'", $output);
+				if (count($output) > 0) {
+					$this->closeByPid($content);
+				} else {
+					file_exists($pid) && @unlink($pid);
+				}
+			}
+			usleep(100);
+		}
+		echo PHP_EOL;
+	}
+
+
+	/**
+	 * @param $port
+	 * @return bool|array
+	 */
+	private function isUse($port)
+	{
+		if (empty($port)) {
+			return false;
+		}
+		exec('netstat -tunlp tcp | grep ' . $port, $output);
+		if (empty($output)) {
+			return false;
+		}
+		return $output;
+	}
+
+	/**
+	 * @param $pid
+	 */
+	private function closeByPid($pid)
+	{
+		shell_exec('kill -TERM ' . $pid);
+	}
+
+
+}