156 lines
3.3 KiB
PHP
156 lines
3.3 KiB
PHP
<?php
|
|
|
|
|
|
namespace Kiri\Server;
|
|
|
|
use Exception;
|
|
use JetBrains\PhpStorm\Pure;
|
|
use Kiri;
|
|
use Kiri\Abstracts\Config;
|
|
use Kiri\Annotation\Inject;
|
|
use Kiri\Events\EventDispatch;
|
|
use Kiri\Exception\ConfigException;
|
|
use Kiri\Message\Handler\Abstracts\HttpService;
|
|
use Kiri\Server\Events\OnServerBeforeStart;
|
|
use Kiri\Server\Events\OnShutdown;
|
|
use Psr\Container\ContainerExceptionInterface;
|
|
use Psr\Container\NotFoundExceptionInterface;
|
|
use ReflectionException;
|
|
use Swoole\Coroutine;
|
|
|
|
|
|
defined('PID_PATH') or define('PID_PATH', APP_PATH . 'storage/server.pid');
|
|
|
|
/**
|
|
* Class Server
|
|
* @package Http
|
|
*/
|
|
class Server extends HttpService
|
|
{
|
|
|
|
private array $process = [
|
|
];
|
|
|
|
|
|
private mixed $daemon = 0;
|
|
|
|
|
|
/**
|
|
* @var State
|
|
*/
|
|
#[Inject(State::class)]
|
|
public State $state;
|
|
|
|
|
|
public ServerManager $manager;
|
|
|
|
|
|
/**
|
|
*
|
|
*/
|
|
public function init()
|
|
{
|
|
$this->manager = Kiri::getContainer()->get(Server::class);
|
|
$enable_coroutine = Config::get('servers.settings.enable_coroutine', false);
|
|
Config::set('servers.settings.enable_coroutine', true);
|
|
if ($enable_coroutine != true) {
|
|
return;
|
|
}
|
|
Coroutine::set([
|
|
'hook_flags' => SWOOLE_HOOK_ALL ^ SWOOLE_HOOK_BLOCKING_FUNCTION,
|
|
'enable_deadlock_check' => FALSE,
|
|
'exit_condition' => function () {
|
|
return Coroutine::stats()['coroutine_num'] === 0;
|
|
}
|
|
]);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $process
|
|
*/
|
|
public function addProcess($process)
|
|
{
|
|
$this->process[] = $process;
|
|
}
|
|
|
|
|
|
/**
|
|
* @return void
|
|
* @throws ConfigException
|
|
* @throws ContainerExceptionInterface
|
|
* @throws NotFoundExceptionInterface
|
|
* @throws ReflectionException
|
|
*/
|
|
public function start(): void
|
|
{
|
|
$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'], $rpcService['mode'], $rpcService);
|
|
}
|
|
|
|
$processes = array_merge($this->process, Config::get('processes', []));
|
|
|
|
$this->getContainer()->get(ProcessManager::class)->batch($processes);
|
|
|
|
$this->getEventDispatch()->dispatch(new OnServerBeforeStart());
|
|
|
|
$this->manager->start();
|
|
}
|
|
|
|
|
|
/**
|
|
* @return void
|
|
* @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->state->exit($config['port']);
|
|
}
|
|
$this->getContainer()->get(EventDispatch::class)->dispatch(new OnShutdown());
|
|
}
|
|
|
|
|
|
/**
|
|
* @return bool
|
|
* @throws ConfigException
|
|
* @throws Exception
|
|
*/
|
|
public function isRunner(): bool
|
|
{
|
|
return $this->state->isRunner();
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $daemon
|
|
* @return Server
|
|
*/
|
|
public function setDaemon($daemon): static
|
|
{
|
|
if (!in_array($daemon, [0, 1])) {
|
|
return $this;
|
|
}
|
|
$this->daemon = $daemon;
|
|
return $this;
|
|
}
|
|
|
|
|
|
/**
|
|
* @return \Swoole\Http\Server|\Swoole\Server|\Swoole\WebSocket\Server|null
|
|
*/
|
|
#[Pure] public function getServer(): \Swoole\Http\Server|\Swoole\Server|\Swoole\WebSocket\Server|null
|
|
{
|
|
return $this->manager->getServer();
|
|
}
|
|
|
|
}
|