137 lines
3.9 KiB
PHP
137 lines
3.9 KiB
PHP
<?php
|
|
|
|
namespace Kiri\Server\Handler;
|
|
|
|
use Exception;
|
|
use Kiri;
|
|
use Kiri\Core\Help;
|
|
use Kiri\Events\EventDispatch;
|
|
use Kiri\Server\Events\OnAfterWorkerStart;
|
|
use Kiri\Server\Events\OnBeforeWorkerStart;
|
|
use Kiri\Server\Events\OnTaskerStart as OnTaskStart;
|
|
use Kiri\Server\Events\OnWorkerError;
|
|
use Kiri\Server\Events\OnWorkerExit;
|
|
use Kiri\Server\Events\OnWorkerStart;
|
|
use Kiri\Server\Events\OnWorkerStop;
|
|
use Psr\Container\ContainerExceptionInterface;
|
|
use Psr\Container\NotFoundExceptionInterface;
|
|
use ReflectionException;
|
|
use Swoole\Server;
|
|
use Swoole\Timer;
|
|
|
|
|
|
/**
|
|
* Class OnServerWorker
|
|
* @package Server\Worker
|
|
*/
|
|
class OnServerWorker extends \Kiri\Server\Abstracts\Server
|
|
{
|
|
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param int $workerId
|
|
* @return void
|
|
* @throws ContainerExceptionInterface
|
|
* @throws NotFoundExceptionInterface
|
|
* @throws Exception
|
|
*/
|
|
public function onWorkerStart(Server $server, int $workerId): void
|
|
{
|
|
$dispatch = \Kiri::getDi()->get(EventDispatch::class);
|
|
$dispatch->dispatch(new OnBeforeWorkerStart($workerId));
|
|
set_env('environmental_workerId', $workerId);
|
|
|
|
if ($workerId < $server->setting['worker_num']) {
|
|
$this->processName($server, 'Worker');
|
|
set_env('environmental', Kiri::WORKER);
|
|
$dispatch->dispatch(new OnWorkerStart($server, $workerId));
|
|
} else {
|
|
$this->processName($server, 'Tasker');
|
|
set_env('environmental', Kiri::TASK);;
|
|
$dispatch->dispatch(new OnTaskStart($server, $workerId));
|
|
}
|
|
|
|
$dispatch->dispatch(new OnAfterWorkerStart());
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param string $prefix
|
|
* @return void
|
|
*/
|
|
protected function processName(Server $server, string $prefix): void
|
|
{
|
|
$prefix = sprintf($prefix . ' Process[%d].%d', $server->worker_pid, $server->worker_id);
|
|
Kiri::setProcessName($prefix);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param int $workerId
|
|
* @throws ContainerExceptionInterface
|
|
* @throws NotFoundExceptionInterface|ReflectionException
|
|
*/
|
|
public function onWorkerStop(Server $server, int $workerId): void
|
|
{
|
|
event(new OnWorkerStop($server, $workerId));
|
|
Timer::clearAll();
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param int $workerId
|
|
* @throws ContainerExceptionInterface
|
|
* @throws NotFoundExceptionInterface|ReflectionException
|
|
*/
|
|
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 ContainerExceptionInterface
|
|
* @throws NotFoundExceptionInterface
|
|
* @throws Exception
|
|
*/
|
|
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));
|
|
|
|
$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(), 9)
|
|
);
|
|
|
|
error($message);
|
|
|
|
$this->system_mail($message);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $messageContent
|
|
* @throws Exception
|
|
*/
|
|
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']);
|
|
}
|
|
}
|
|
|
|
}
|