变更

Abstracts/TraitServer.php

@@ -3,8 +3,10 @@
 namespace Kiri\Server\Abstracts;
 
 use Exception;
+use Kiri\Server\CoroutineServer;
 use Psr\Container\ContainerExceptionInterface;
 use Psr\Container\NotFoundExceptionInterface;
+use Swoole\Coroutine;
 use Swoole\Http\Server as HServer;
 use Swoole\Server;
 use Kiri\Server\Constant;
@@ -36,8 +38,6 @@ trait TraitServer
 	}
 	
 	
-	
-	
 	/**
 	 * @param array $signal
 	 * @return void
@@ -47,7 +47,7 @@ trait TraitServer
 	 */
 	public function onSignal(array $signal): void
 	{
-		pcntl_signal(SIGINT, [$this, 'onSigint']);
+		$this->onPcntlSignal(SIGINT, [$this, 'onSigint']);
 		foreach ($signal as $sig => $value) {
 			if (is_array($value) && is_string($value[0])) {
 				$value[0] = $this->container->get($value[0]);
@@ -55,7 +55,27 @@ trait TraitServer
 			if (!is_callable($value, true)) {
 				throw new Exception('Register signal callback must can exec.');
 			}
-			pcntl_signal($sig, $value);
+			$this->onPcntlSignal($sig, $value);
+		}
+	}
+	
+	
+	/**
+	 * @param $signal
+	 * @param $callback
+	 * @return void
+	 */
+	private function onPcntlSignal($signal, $callback): void
+	{
+		if (get_called_class() == CoroutineServer::class) {
+			pcntl_signal(SIGINT, [$this, 'onSigint']);
+		} else {
+			Coroutine::create(function () use ($signal, $callback) {
+				$data = Coroutine::waitSignal($signal);
+				if ($data) {
+					$callback($signal, [true]);
+				}
+			});
 		}
 	}
 	

CoroutineServer.php

@@ -172,6 +172,8 @@ class CoroutineServer implements ServerInterface
 		Coroutine\run(function () {
 			$this->dispatch->dispatch(new OnServerBeforeStart());
 			
+			$this->onSignal(Config::get('signal', []));
+			
 			$this->onTasker();
 			foreach ($this->servers as $server) {
 				Coroutine::create(function () use ($server) {

Server.php

@@ -107,7 +107,6 @@ class Server extends HttpService
 	{
 		$this->onHotReload();
 		$this->manager->initCoreServers(Config::get('server', [], true), $this->daemon);
-		$this->manager->onSignal(Config::get('signal', []));
 		$this->manager->start();
 	}