kiri-http-server/Handler/OnServerWorker.php

<?php

namespace Kiri\Server\Handler;

use Kiri;
use Kiri\Abstracts\CoordinatorManager;
use Kiri\Coordinator;
use Kiri\Core\Help;
use Kiri\Events\EventDispatch;
use Kiri\Server\Events\OnAfterWorkerStart;
use Kiri\Server\Events\OnBeforeWorkerStart;
use Kiri\Server\Events\OnTaskerStart as OnTaskerStart;
use Kiri\Server\Events\OnWorkerError;
use Kiri\Server\Events\OnWorkerExit;
use Kiri\Server\Events\OnWorkerStart;
use Kiri\Server\Events\OnWorkerStop;
use Psr\Http\Message\RequestInterface;
use Swoole\Server;
use Swoole\Timer;
use Throwable;
use function config;


/**
 * Class OnServerWorker
 * @package Server\Worker
 */
class OnServerWorker extends Kiri\Server\Abstracts\Server
{

    /**
     * @return void
     */
    public function init(): void
    {
        on(OnBeforeWorkerStart::class, [$this, 'onWorkerNameAlias']);
    }


    /**
     * @param OnBeforeWorkerStart $workerStart
     * @return void
     */
    public function onWorkerNameAlias(OnBeforeWorkerStart $workerStart): void
    {
        set_env('environmental_workerId', $workerStart->workerId);
        if ($workerStart->workerId < $workerStart->server->setting['worker_num']) {
            $this->processName($workerStart->server, 'Worker');
            set_env('environmental', Kiri::WORKER);
        } else {
            $this->processName($workerStart->server, 'Tasker');
            set_env('environmental', Kiri::TASK);
        }
    }


    /**
     * @param Server $server
     * @param int $workerId
     * @return void
     * @throws
     */
    public function onWorkerStart(Server $server, int $workerId): void
    {
        $dispatch = Kiri::getDi()->get(EventDispatch::class);
        $dispatch->dispatch(new OnBeforeWorkerStart($server, $workerId));
        if ($workerId < $server->setting['worker_num']) {
            CoordinatorManager::utility(Coordinator::WORKER_START)->waite();

            $dispatch->dispatch(new OnWorkerStart($server, $workerId));
        } else {
            $dispatch->dispatch(new OnTaskerStart($server, $workerId));
        }
        $dispatch->dispatch(new OnAfterWorkerStart($server, $workerId));
    }


    /**
     * @param Server $server
     * @param string $prefix
     * @return void
     */
    protected function processName(Server $server, string $prefix): void
    {
        Kiri::setProcessName(sprintf($prefix . '[%d]', $server->worker_pid));
    }


    /**
     * @param Server $server
     * @param int $workerId
     * @throws
     */
    public function onWorkerStop(Server $server, int $workerId): void
    {
        event(new OnWorkerStop($server, $workerId));
        Timer::clearAll();
    }


    /**
     * @param Server $server
     * @param int $workerId
     * @throws
     */
    public function onWorkerExit(Server $server, int $workerId): void
    {
        event(new OnWorkerExit($server, $workerId));
    }


    /**
     * @param Server $server
     * @param int $worker_id
     * @param int $worker_pid
     * @param int $exit_code
     * @param int $signal
     * @throws
     */
    public function onWorkerError(Server $server, int $worker_id, int $worker_pid, int $exit_code, int $signal): void
    {
        event(new OnWorkerError($server, $worker_id, $worker_pid, $exit_code, $signal));

        debug_print_backtrace(DEBUG_BACKTRACE_PROVIDE_OBJECT);

        /** @var RequestInterface $context */
        $context = Kiri\Di\Context::get(RequestInterface::class);
        if (is_null($context)) {
            $message = sprintf('Worker#%d::%d error stop. signal %d, exit_code %d, msg %s', $worker_id, $worker_pid, $signal, $exit_code, swoole_strerror(swoole_last_error(), $signal));
        } else {
            $message = sprintf('Worker#%d::%d error stop. signal %d, exit_code %d, msg %s, method: %s, path: %s, query: %s', $worker_id, $worker_pid, $signal, $exit_code, swoole_strerror(swoole_last_error(), $signal),
                $context->getMethod(), $context->getUri()->getPath(), $context->getUri()->getQuery());
        }
        error($message . PHP_EOL);

        $this->system_mail($message);
    }


    /**
     * @param $messageContent
     * @throws
     */
    protected function system_mail($messageContent): void
    {
        try {
            $email = config('email', ['enable' => false]);
            if (!empty($email) && ($email['enable'] ?? false)) {
                Help::sendEmail($email, 'Service Error', $messageContent);
            }
        } catch (Throwable $e) {
            error($e, ['email']);
        }
    }

}