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\Di\Inject\Container;
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
{


	/**
	 * @var EventDispatch
	 */
	#[Container(EventDispatch::class)]
	public EventDispatch $dispatch;


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


	/**
	 * @param OnBeforeWorkerStart $workerStart
	 * @return void
	 */
	public function onWorkerNameAlias(OnBeforeWorkerStart $workerStart): void
	{
		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);
		}
		set_env('environmental_worker_id', $workerStart->workerId);
	}


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

				$this->dispatch->dispatch(new OnWorkerStart($server, $workerId));
			} else {
				$this->dispatch->dispatch(new OnTaskerStart($server, $workerId));
			}
		} catch (Throwable $exception) {
			\Kiri::getLogger()->json_log($exception);
		} finally {
			$this->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());
		}
		$this->getLogger()->println($message);
		$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) {
			\Kiri::getLogger()->json_log($e);
		}
	}

}