kiri-core/HttpServer/Events/OnWorkerStart.php

<?php
declare(strict_types=1);

namespace HttpServer\Events;

use Annotation\Target;
use Exception;
use HttpServer\Abstracts\Callback;
use Snowflake\Abstracts\Config;
use Snowflake\Event;
use Snowflake\Exception\ConfigException;
use Snowflake\Snowflake;
use Swoole\Coroutine;
use Swoole\Process;
use Swoole\Server;

/**
 * Class OnWorkerStart
 * @package HttpServer\Events
 */
#[Target]
class OnWorkerStart extends Callback
{

    /** @var int|string 重启信号 */
    private int $signal = SIGUSR1 | SIGUSR2;


    /** @var bool 是否打印 */
    private bool $isPrint = false;

    /**
     * @param Server $server
     * @param int $worker_id
     *
     * @return mixed
     * @throws ConfigException
     * @throws Exception
     */
    public function onHandler(Server $server, int $worker_id): void
    {
        Coroutine::set(['enable_deadlock_check' => false]);
        Snowflake::app()->stateInit();

        $this->isPrint = false;

        $this->debug(sprintf('System#%d start.', $worker_id));
        if ($worker_id >= $server->setting['worker_num']) {
            $this->onTask($server, $worker_id);
        } else {
            $this->onWorker($server, $worker_id);
        }

        Coroutine::create([$this, 'onSignal'], $server, $worker_id);
    }


    /**
     * @param Server $server
     * @param int $worker_id
     * @throws \Snowflake\Exception\ComponentException
     * OnTask Worker
     */
    public function onTask(Server $server, int $worker_id)
    {
        putenv('environmental=' . Snowflake::TASK);

        fire(Event::SERVER_TASK_START);

        $this->set_process_name($server, $worker_id);
    }


    /**
     * @param Server $server
     * @param int $worker_id
     * @throws Exception
     * onWorker
     */
    public function onWorker(Server $server, int $worker_id)
    {
        Snowflake::setWorkerId($server->worker_pid);

        putenv('environmental=' . Snowflake::WORKER);
        try {
            fire(Event::SERVER_WORKER_START, [$worker_id]);
        } catch (\Throwable $exception) {
            $this->addError($exception);
            write($exception->getMessage(), 'worker');
        }
        $this->set_process_name($server, $worker_id);
    }


    /**
     * @param $server
     * @param $worker_id
     */
    public function onSignal($server, $worker_id)
    {
        $env = ucfirst(Snowflake::getEnvironmental());
        while (Coroutine::waitSignal($this->signal, -1)) {
            if ($this->isPrint === false) {
                $this->warning(sprintf('Receive %s#%d stop event.', $env, $worker_id));
                $this->isPrint = true;
            }
            if (!Snowflake::app()->isRun()) {
                break;
            }
            sleep(1);
        }
        return $server->stop($worker_id);
    }


    /**
     * @param $socket
     * @param $worker_id
     * @return string
     * @throws ConfigException
     */
    private function set_process_name($socket, $worker_id): mixed
    {
        $prefix = Config::get('id', false, 'system');
        if ($worker_id >= $socket->setting['worker_num']) {
            $name = $prefix . ' Task: No.' . $worker_id;
        } else {
            $name = $prefix . ' worker: No.' . $worker_id;
        }
        return swoole_set_process_name($name);
    }


}