1

Abstracts/BaseProcess.php

@@ -3,7 +3,7 @@
 namespace Server\Abstracts;
 
 
-use JetBrains\PhpStorm\Pure;
+use Kiri\Context;
 use Server\Contract\OnProcessInterface;
 use Swoole\Coroutine;
 use Swoole\Process;
@@ -17,7 +17,7 @@ abstract class BaseProcess implements OnProcessInterface
 	protected bool $isStop = false;
 
 
-	protected mixed $redirect_stdin_and_stdout = null;
+	protected bool $redirect_stdin_and_stdout = FALSE;
 
 
 	protected int $pipe_type = SOCK_DGRAM;
@@ -47,10 +47,10 @@ abstract class BaseProcess implements OnProcessInterface
 	}
 
 	/**
-	 * @return mixed
-	 */
-	public function getRedirectStdinAndStdout(): mixed
-	{
+	 * @return bool
+     */
+	public function getRedirectStdinAndStdout(): bool
+    {
 		return $this->redirect_stdin_and_stdout;
 	}
 
@@ -80,4 +80,32 @@ abstract class BaseProcess implements OnProcessInterface
 	}
 
 
+	/**
+	 *
+	 */
+	public function onSigterm(): static
+	{
+		if (!Context::inCoroutine()) {
+			Process::signal(SIGTERM, fn($data) => $this->onShutdown($data));
+		} else {
+			Coroutine::create(function () {
+				$data = Coroutine::waitSignal(SIGTERM, -1);
+				if ($data) {
+					$this->onShutdown($data);
+				}
+			});
+		}
+		return $this;
+	}
+
+
+	/**
+	 * @param $data
+	 */
+	protected function onShutdown($data): void
+	{
+		$this->isStop = true;
+	}
+
+
 }

Server.php

@@ -3,13 +3,15 @@
 
 namespace Server;
 
-use Note\Inject;
 use Exception;
 use Http\Handler\Abstracts\HttpService;
-use JetBrains\PhpStorm\Pure;
 use Kiri\Abstracts\Config;
 use Kiri\Events\EventDispatch;
 use Kiri\Exception\ConfigException;
+use Kiri\Kiri;
+use Note\Inject;
+use Psr\Container\ContainerExceptionInterface;
+use Psr\Container\NotFoundExceptionInterface;
 use Server\Events\OnShutdown;
 
 
@@ -26,19 +28,14 @@ class Server extends HttpService
 	];
 
 
-	/**
-	 * @Inject ServerManager
-	 * @var null|ServerManager
-	 */
-	#[Inject(ServerManager::class)]
-	public ?ServerManager $manager = null;
-
 	private mixed $daemon = 0;
 
 
-	/** @var EventDispatch */
-	#[Inject(EventDispatch::class)]
-	public EventDispatch $eventDispatch;
+	/**
+	 * @var State
+	 */
+	#[Inject(State::class)]
+	public State $state;
 
 
 	/**
@@ -59,54 +56,58 @@ class Server extends HttpService
 
 
 	/**
-	 * @return string start server
-	 *
-	 * start server
+	 * @return string
 	 * @throws ConfigException
+	 * @throws ContainerExceptionInterface
+	 * @throws NotFoundExceptionInterface
+	 * @throws \ReflectionException
 	 * @throws Exception
 	 */
 	public function start(): string
 	{
-		$this->manager->initBaseServer(Config::get('server', [], true), $this->daemon);
+		$this->manager()->initBaseServer(Config::get('server', [], true), $this->daemon);
 
 		$rpcService = Config::get('rpc', []);
 		if (!empty($rpcService)) {
-			$this->manager->addListener($rpcService['type'], $rpcService['host'], $rpcService['port'],
+			$this->manager()->addListener($rpcService['type'], $rpcService['host'], $rpcService['port'],
 				$rpcService['mode'], $rpcService);
 		}
 
 		$processes = array_merge($this->process, Config::get('processes', []));
 		foreach ($processes as $process) {
-			$this->manager->addProcess($process);
+			$this->manager()->addProcess($process);
 		}
 
-		return $this->manager->getServer()->start();
+		return $this->manager()->getServer()->start();
 	}
 
 
 	/**
 	 * @return void
-	 *
-	 * start server
+	 * @throws ConfigException
+	 * @throws ContainerExceptionInterface
+	 * @throws NotFoundExceptionInterface
+	 * @throws \ReflectionException
 	 * @throws Exception
 	 */
 	public function shutdown()
 	{
 		$configs = Config::get('server', [], true);
-		foreach ($this->manager->sortService($configs['ports'] ?? []) as $config) {
-			$this->manager->stopServer($config['port']);
+		foreach ($this->manager()->sortService($configs['ports'] ?? []) as $config) {
+			$this->state->exit($config['port']);
 		}
-		$this->eventDispatch->dispatch(new OnShutdown());
+		$this->container->get(EventDispatch::class)->dispatch(new OnShutdown());
 	}
 
 
 	/**
 	 * @return bool
 	 * @throws ConfigException
+	 * @throws Exception
 	 */
 	public function isRunner(): bool
 	{
-		return $this->manager->isRunner();
+		return $this->state->isRunner();
 	}
 
 
@@ -126,10 +127,21 @@ class Server extends HttpService
 
 	/**
 	 * @return \Swoole\Http\Server|\Swoole\Server|\Swoole\WebSocket\Server|null
+	 * @throws \ReflectionException
 	 */
-	#[Pure] public function getServer(): \Swoole\Http\Server|\Swoole\Server|\Swoole\WebSocket\Server|null
+	public function getServer(): \Swoole\Http\Server|\Swoole\Server|\Swoole\WebSocket\Server|null
 	{
-		return $this->manager->getServer();
+		return $this->manager()->getServer();
+	}
+
+
+	/**
+	 * @return ServerManager
+	 * @throws \ReflectionException
+	 */
+	private function manager(): ServerManager
+	{
+		return Kiri::getDi()->get(ServerManager::class);
 	}
 
 }

ServerCommand.php

@@ -106,8 +106,6 @@ class ServerCommand extends Command
 
 		Kiri::app()->getRouter()->read_files();
 
-		$this->eventProvider->dispatch(new OnServerBeforeStart());
-
 		$manager->start();
 
 		return 1;

ServerManager.php

@@ -2,14 +2,15 @@
 
 namespace Server;
 
-use Note\Inject;
 use Exception;
+use Kiri\Abstracts\Component;
 use Kiri\Abstracts\Config;
-use Kiri\Di\Container;
-use Psr\Container\ContainerInterface;
+use Kiri\Di\NoteManager;
 use Kiri\Error\Logger;
+use Kiri\Events\EventDispatch;
 use Kiri\Exception\ConfigException;
 use Kiri\Kiri;
+use Note\Inject;
 use Psr\Container\ContainerExceptionInterface;
 use Psr\Container\NotFoundExceptionInterface;
 use ReflectionException;
@@ -22,13 +23,13 @@ use Server\Contract\OnMessageInterface;
 use Server\Contract\OnPacketInterface;
 use Server\Contract\OnProcessInterface;
 use Server\Contract\OnReceiveInterface;
-use Server\Contract\OnTaskInterface;
+use Server\Events\OnServerBeforeStart;
 use Server\Handler\OnPipeMessage;
 use Server\Handler\OnServer;
 use Server\Handler\OnServerManager;
 use Server\Handler\OnServerReload;
-use Server\Tasker\OnServerTask;
 use Server\Handler\OnServerWorker;
+use Server\Tasker\OnServerTask;
 use Swoole\Http\Server as HServer;
 use Swoole\Process;
 use Swoole\Server;
@@ -40,7 +41,7 @@ use Swoole\WebSocket\Server as WServer;
  * Class OnServerManager
  * @package Http\Service
  */
-class ServerManager
+class ServerManager extends Component
 {
 
 
@@ -52,10 +53,12 @@ class ServerManager
 	public int $port = 0;
 
 
+	/**
+	 * @var Logger
+	 */
 	#[Inject(Logger::class)]
 	public Logger $logger;
 
-
 	/** @var array<string,Port> */
 	public array $ports = [];
 
@@ -65,11 +68,7 @@ class ServerManager
 	private Server|null $server = null;
 
 
-	/**
-	 * @var Container
-	 */
-	#[Inject(ContainerInterface::class)]
-	public ContainerInterface $container;
+	protected array $initProcesses = [];
 
 
 	const DEFAULT_EVENT = [
@@ -100,9 +99,11 @@ class ServerManager
 
 	/**
 	 * @return Server|WServer|HServer|null
+	 * @throws ReflectionException
 	 */
 	public function getServer(): Server|WServer|HServer|null
 	{
+		di(EventDispatch::class)->dispatch(new OnServerBeforeStart());
 		return $this->server;
 	}
 
@@ -113,9 +114,10 @@ class ServerManager
 	 * @param int $port
 	 * @param int $mode
 	 * @param array $settings
-	 * @throws ReflectionException
 	 * @throws ConfigException
-	 * @throws Exception
+	 * @throws ContainerExceptionInterface
+	 * @throws NotFoundExceptionInterface
+	 * @throws ReflectionException
 	 */
 	public function addListener(string $type, string $host, int $port, int $mode, array $settings = [])
 	{
@@ -131,8 +133,12 @@ class ServerManager
 
 
 	/**
-	 * @throws ReflectionException
+	 * @param $configs
+	 * @param int $daemon
 	 * @throws ConfigException
+	 * @throws ContainerExceptionInterface
+	 * @throws NotFoundExceptionInterface
+	 * @throws ReflectionException
 	 */
 	public function initBaseServer($configs, int $daemon = 0): void
 	{
@@ -144,23 +150,6 @@ class ServerManager
 	}
 
 
-	/**
-	 * @return bool
-	 * @throws ConfigException
-	 * @throws Exception
-	 */
-	public function isRunner(): bool
-	{
-		$configs = Config::get('server', [], true);
-		foreach ($this->sortService($configs['ports']) as $config) {
-			if (checkPortIsAlready($config['port'])) {
-				return true;
-			}
-		}
-		return false;
-	}
-
-
 	/**
 	 * @param string|OnProcessInterface|BaseProcess $customProcess
 	 * @throws Exception
@@ -175,14 +164,23 @@ class ServerManager
 			if (Kiri::getPlatform()->isLinux()) {
 				$process->name($system . '(' . $customProcess->getName() . ')');
 			}
-			$customProcess->process($process);
+			$customProcess->onSigterm()->process($process);
 		}, $customProcess->getRedirectStdinAndStdout(), $customProcess->getPipeType(), $customProcess->isEnableCoroutine());
 		$this->logger->debug($system . ' ' . $customProcess->getName() . ' start.');
-		$this->container->setBindings($customProcess->getName(), $process);
+		$this->initProcesses[$customProcess->getName()] = $process;
 		$this->server->addProcess($process);
 	}
 
 
+	/**
+	 * @return array<string,Process>
+	 */
+	public function getProcesses(): array
+	{
+		return $this->initProcesses;
+	}
+
+
 	/**
 	 * @return array
 	 */
@@ -192,21 +190,13 @@ class ServerManager
 	}
 
 
-	/**
-	 * @param string $key
-	 * @param string|int $value
-	 */
-	public static function setEnv(string $key, string|int $value): void
-	{
-		putenv(sprintf('%s=%s', $key, (string)$value));
-	}
-
-
 	/**
 	 * @param ServerManager $context
 	 * @param array $config
 	 * @param int $daemon
 	 * @throws ConfigException
+	 * @throws ContainerExceptionInterface
+	 * @throws NotFoundExceptionInterface
 	 * @throws ReflectionException
 	 * @throws Exception
 	 */
@@ -256,7 +246,7 @@ class ServerManager
 	{
 		$id = Config::get('id', 'system-service');
 
-		$this->logger->debug(sprintf('[%s]' . $type . ' service %s::%d start', $id, $host, $port));
+		$this->logger->debug(sprintf('[%s].' . $type . ' service %s::%d start', $id, $host, $port));
 
 		/** @var Server\Port $service */
 		$this->ports[$port] = $this->server->addlistener($host, $port, $mode);
@@ -302,28 +292,12 @@ class ServerManager
 
 		$id = Config::get('id', 'system-service');
 
-		$this->logger->debug(sprintf('[%s]' . $type . ' service %s::%d start', $id, $host, $port));
+		$this->logger->debug(sprintf('[%s].' . $type . ' service %s::%d start', $id, $host, $port));
 
 		$this->addDefaultListener($settings);
 	}
 
 
-	/**
-	 * @param int $port
-	 * @throws Exception
-	 */
-	public function stopServer(int $port)
-	{
-		if (!($pid = checkPortIsAlready($port))) {
-			return;
-		}
-		while (checkPortIsAlready($port)) {
-			Process::kill($pid, SIGTERM);
-			usleep(300);
-		}
-	}
-
-
 	/**
 	 * @param array $settings
 	 * @throws ContainerExceptionInterface
@@ -370,40 +344,6 @@ class ServerManager
 	}
 
 
-	/**
-	 * @param string $class
-	 * @return object
-	 */
-	private function getNewInstance(string $class): object
-	{
-		return $this->container->create($class);
-	}
-
-
-	/**
-	 * @param OnTaskInterface|string $handler
-	 * @param array $params
-	 * @param int|null $workerId
-	 * @throws ReflectionException
-	 * @throws Exception
-	 */
-	public function task(OnTaskInterface|string $handler, array $params = [], int $workerId = null)
-	{
-		if ($workerId === null || $workerId <= $this->server->setting['worker_num']) {
-			$workerId = random_int($this->server->setting['worker_num'] + 1,
-				$this->server->setting['worker_num'] + 1 + $this->server->setting['task_worker_num']);
-		}
-		if (is_string($handler)) {
-			$implements = $this->container->getReflect($handler);
-			if (!in_array(OnTaskInterface::class, $implements->getInterfaceNames())) {
-				throw new Exception('Task must instance ' . OnTaskInterface::class);
-			}
-			$handler = $implements->newInstanceArgs($params);
-		}
-		$this->server->task(serialize($handler), $workerId);
-	}
-
-
 	/**
 	 * @param mixed $message
 	 * @param int $workerId
@@ -417,12 +357,15 @@ class ServerManager
 
 	/**
 	 * @param array $events
+	 * @return void
+	 * @throws ContainerExceptionInterface
+	 * @throws NotFoundExceptionInterface
 	 * @throws ReflectionException
 	 */
 	private function addTaskListener(array $events = []): void
 	{
 		$task_use_object = $this->server->setting['task_object'] ?? $this->server->setting['task_use_object'] ?? false;
-		$reflect = $this->container->getReflect(OnServerTask::class)?->newInstance();
+		$reflect = $this->container->get(OnServerTask::class);
 		$this->server->on('finish', $events[Constant::FINISH] ?? [$reflect, 'onFinish']);
 		if ($task_use_object || $this->server->setting['task_enable_coroutine']) {
 			$this->server->on('task', $events[Constant::TASK] ?? [$reflect, 'onCoroutineTask']);
@@ -434,6 +377,7 @@ class ServerManager
 
 	/**
 	 * @param array|null $settings
+	 * @return void
 	 * @throws ContainerExceptionInterface
 	 * @throws NotFoundExceptionInterface
 	 * @throws ReflectionException

State.php

@@ -0,0 +1,56 @@
+<?php
+
+namespace Server;
+
+use Exception;
+use Kiri\Abstracts\Component;
+use Kiri\Abstracts\Config;
+use Swoole\Process;
+
+class State extends Component
+{
+
+	use TraitServer;
+
+
+	public array $servers = [];
+
+
+	public function init()
+	{
+		$this->servers = Config::get('server.ports');
+	}
+
+
+	/**
+	 * @return bool
+	 * @throws Exception
+	 */
+	public function isRunner(): bool
+	{
+		$ports = $this->sortService($this->servers);
+		foreach ($ports as $config) {
+			if (checkPortIsAlready($config['port'])) {
+				return true;
+			}
+		}
+		return false;
+	}
+
+
+	/**
+	 * @param $port
+	 * @throws Exception
+	 */
+	public function exit($port)
+	{
+		if (!($pid = checkPortIsAlready($port))) {
+			return;
+		}
+		while (checkPortIsAlready($port)) {
+			Process::kill($pid, SIGTERM);
+			usleep(300);
+		}
+	}
+
+}

Tasker/AsyncTaskExecute.php

@@ -2,12 +2,10 @@
 
 namespace Server\Tasker;
 
-use Note\Inject;
 use Exception;
-use Kiri\Abstracts\BaseObject;
+use Kiri\Abstracts\Component;
 use Kiri\Core\HashMap;
-use Kiri\Di\Container;
-use Psr\Container\ContainerInterface;
+use Kiri\Kiri;
 use ReflectionException;
 use Server\Contract\OnTaskInterface;
 use Server\SwooleServerInterface;
@@ -16,22 +14,14 @@ use Server\SwooleServerInterface;
 /**
  *
  */
-class AsyncTaskExecute extends BaseObject
+class AsyncTaskExecute extends Component
 {
 
 
 	/**
-	 * @var SwooleServerInterface
+	 * @var SwooleServerInterface|null
 	 */
-	#[Inject(SwooleServerInterface::class)]
-	public SwooleServerInterface $server;
-
-
-	/**
-	 * @var Container
-	 */
-	#[Inject(ContainerInterface::class)]
-	public ContainerInterface $container;
+	public ?SwooleServerInterface $server = null;
 
 
 	private HashMap $hashMap;
@@ -64,8 +54,13 @@ class AsyncTaskExecute extends BaseObject
 	 */
 	public function execute(OnTaskInterface|string $handler, array $params = [], int $workerId = null)
 	{
+		if (!$this->server) {
+			$this->server = Kiri::getDi()->get(SwooleServerInterface::class);
+		}
 		if ($workerId === null || $workerId <= $this->server->setting['worker_num']) {
-			$workerId = random_int($this->server->setting['worker_num'] + 1, $this->server->setting['task_worker_num']);
+			$workerNum = $this->server->setting['worker_num'];
+			$taskerNum = $workerNum + $this->server->setting['task_worker_num'];
+			$workerId = random_int($workerNum, $taskerNum - 1);
 		}
 		if (is_string($handler)) {
 			$handler = $this->handle($handler, $params);