kiri-http-server/Server.php

<?php


namespace Kiri\Server;

use Exception;
use Kiri;
use Kiri\Events\EventDispatch;
use Kiri\Router\Router;
use Kiri\Server\Events\OnShutdown;
use Kiri\Server\Events\OnTaskerStart;
use Kiri\Server\Events\OnWorkerStart;
use Kiri\Server\Events\OnWorkerStop;
use Kiri\Server\Abstracts\AsyncServer;
use Psr\Container\ContainerExceptionInterface;
use Psr\Container\NotFoundExceptionInterface;
use Swoole\Timer;


defined('PID_PATH') or define('PID_PATH', APP_PATH . 'storage/server.pid');


/**
 * Class Server
 * @package Http
 */
class Server
{


    /**
     * @var int
     */
    private int $daemon = 0;


    /**
     * @param AsyncServer $manager
     * @param State $state
     * @param EventDispatch $dispatch
     * @param Router $router
     */
    public function __construct(public AsyncServer   $manager,
                                public State         $state,
                                public EventDispatch $dispatch,
                                public Router        $router)
    {
    }


    /**
     * @return void
     */
    public function init(): void
    {
        on(OnWorkerStart::class, [$this, 'onWorkerName']);
        on(OnTaskerStart::class, [$this, 'onTaskerName']);
    }


    /**
     * @return void
     * @throws Exception
     */
    public function start(): void
    {
        on(OnWorkerStop::class, [Timer::class, 'clearAll'], 9999);
        if (\config('reload.hot', false) === true) {
            $this->manager->addProcess(HotReload::class);
        } else {
            $this->router->scan_build_route();
        }
        $this->manager->initCoreServers(\config('server', []), $this->daemon);
        $this->manager->start();
    }


    /**
     * @param OnWorkerStart $onWorkerStart
     */
    public function onWorkerName(OnWorkerStart $onWorkerStart): void
    {
        if (!property_exists($onWorkerStart->server, 'worker_pid')) {
            return;
        }
        $prefix = sprintf('Worker Process[%d].%d', $onWorkerStart->server->worker_pid, $onWorkerStart->workerId);
        set_env('environmental', Kiri::WORKER);

        Kiri::setProcessName($prefix);
    }


    /**
     * @param OnTaskerStart $onWorkerStart
     */
    public function onTaskerName(OnTaskerStart $onWorkerStart): void
    {
        if (!property_exists($onWorkerStart->server, 'worker_pid')) {
            return;
        }
        $prefix = sprintf('Tasker Process[%d].%d', $onWorkerStart->server->worker_pid, $onWorkerStart->workerId);
        set_env('environmental', Kiri::TASK);

        Kiri::setProcessName($prefix);
    }


    /**
     * @return void
     * @throws ContainerExceptionInterface
     * @throws NotFoundExceptionInterface
     * @throws Exception
     */
    public function shutdown(): void
    {
        $configs = \config('server', []);
        $instances = $this->manager->sortService($configs['ports'] ?? []);
        foreach ($instances as $config) {
            $this->state->exit($config->port);
        }
        $this->dispatch->dispatch(new OnShutdown());
    }


    /**
     * @return bool
     * @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;
    }
}