kiri-http-server/Server.php

<?php


namespace Kiri\Server;

use Exception;
use Kiri\Message\Handler\Abstracts\HttpService;
use Kiri\Message\Handler\Router;
use Kiri\Abstracts\Config;
use Kiri\Events\EventDispatch;
use Kiri\Exception\ConfigException;
use Kiri\Kiri;
use Kiri\Annotation\Inject;
use Psr\Container\ContainerExceptionInterface;
use Psr\Container\NotFoundExceptionInterface;
use Kiri\Server\Events\OnShutdown;
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 function init()
	{
	}


	/**
	 * @param $process
	 */
	public function addProcess($process)
	{
		$this->process[] = $process;
	}


	/**
	 * @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);

		$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', []));
		foreach ($processes as $process) {
			$this->manager()->addProcess($process);
		}

		return $this->manager()->getServer()->start();
	}

    /**
     * @throws ConfigException
     */
    private function configure_set()
    {
        $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;
            }
        ]);
    }


    /**
     * @return void
     * @throws ConfigException
     * @throws ContainerExceptionInterface
     * @throws NotFoundExceptionInterface
     * @throws \ReflectionException
     * @throws \Exception
     */
    public function runtime_start(): void
    {
        $this->configure_set();

        $this->container->get(Router::class)->read_files();

        $this->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->container->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
	 * @throws \ReflectionException
	 */
	public function getServer(): \Swoole\Http\Server|\Swoole\Server|\Swoole\WebSocket\Server|null
	{
		return $this->manager()->getServer();
	}


	/**
	 * @return ServerManager
	 */
	private function manager(): ServerManager
	{
		return Kiri::getDi()->get(ServerManager::class);
	}

}