改名

http-server/Events/OnWorkerStart.php

@@ -0,0 +1,90 @@
+<?php
+
+
+namespace HttpServer\Events;
+
+
+use Exception;
+use HttpServer\Events\Abstracts\Callback;
+use HttpServer\Events\Http;
+use HttpServer\Events\WebSocket;
+use HttpServer\Route\Annotation\Websocket as AWebsocket;
+use Snowflake\Error\Logger;
+use Snowflake\Event;
+use Snowflake\Snowflake;
+use Swoole\Server;
+
+/**
+ * Class OnWorkerStart
+ * @package HttpServer\Events
+ */
+class OnWorkerStart extends Callback
+{
+
+
+	/**
+	 * @param Server $server
+	 * @param int $worker_id
+	 *
+	 * @return mixed|void
+	 * @throws Exception
+	 */
+	public function onHandler(Server $server, $worker_id)
+	{
+		Snowflake::setProcessId($server->worker_pid);
+
+		$get_name = $this->get_process_name($server, $worker_id);
+		if (!empty($get_name) && !Snowflake::isMac()) {
+			swoole_set_process_name($get_name);
+		}
+		if ($worker_id >= $server->setting['worker_num']) {
+			return;
+		}
+		$this->setWorkerAction($server, $worker_id);
+	}
+
+	/**
+	 * @param $worker_id
+	 * @param  $socket
+	 * @throws Exception
+	 */
+	private function setWorkerAction($socket, $worker_id)
+	{
+		try {
+			if ($socket instanceof Http) {
+				$router = Snowflake::get()->router;
+				$router->loadRouterSetting();
+			} else if ($socket instanceof WebSocket) {
+				$path = APP_PATH . 'app/Websocket';
+
+				/** @var AWebsocket $websocket */
+				$websocket = Snowflake::get()->annotation->register('websocket', AWebsocket::class);
+				$websocket->registration_notes($path, 'App\\Websocket');
+			}
+			$event = Snowflake::get()->event;
+			if (!$event->exists(Event::SERVER_WORKER_START)) {
+				return;
+			}
+			$event->trigger(Event::SERVER_WORKER_START);
+		} catch (\Throwable $exception) {
+			Snowflake::get()->getLogger()->write($exception->getMessage(), 'worker');
+		}
+	}
+
+	/**
+	 * @param $socket
+	 * @param $worker_id
+	 * @return string
+	 */
+	private function get_process_name($socket, $worker_id)
+	{
+		$prefix = 'system:';
+		if ($worker_id >= $socket->setting['worker_num']) {
+			return $prefix . ': Task: No.' . $worker_id;
+		} else {
+			return $prefix . ': worker: No.' . $worker_id;
+		}
+	}
+
+
+}