as2252258@163.com
132 lines
3.1 KiB
PHP
132 lines
3.1 KiB
PHP
<?php
|
|
declare(strict_types=1);
|
|
|
|
namespace HttpServer\Events;
|
|
|
|
use Annotation\Annotation;
|
|
use Exception;
|
|
use HttpServer\Abstracts\Callback;
|
|
use Snowflake\Event;
|
|
use Snowflake\Exception\ConfigException;
|
|
use Snowflake\Snowflake;
|
|
use Swoole\Server;
|
|
|
|
/**
|
|
* Class OnWorkerStart
|
|
* @package HttpServer\Events
|
|
*/
|
|
class OnWorkerStart extends Callback
|
|
{
|
|
|
|
private Server $server;
|
|
|
|
|
|
/**
|
|
* @param $server
|
|
* @param $worker_id
|
|
* @throws ConfigException
|
|
*/
|
|
public function actionBefore($server, $worker_id)
|
|
{
|
|
}
|
|
|
|
|
|
/**
|
|
* @throws Exception
|
|
*/
|
|
public function injectLoader($isWorker = false)
|
|
{
|
|
$runtime = file_get_contents(storage('runtime.php'));
|
|
$annotation = Snowflake::app()->getAnnotation();
|
|
$annotation->setLoader(unserialize($runtime));
|
|
|
|
putenv('state=start');
|
|
putenv('worker=' . $this->server->worker_id);
|
|
|
|
$pipeLine = new Pipeline();
|
|
$pipeLine->if($isWorker, function ($annotation, $server) {
|
|
$annotation->runtime(CONTROLLER_PATH);
|
|
$annotation->runtime(APP_PATH, CONTROLLER_PATH);
|
|
|
|
name($server->worker_pid, 'Worker.' . $server->worker_id);
|
|
})
|
|
->else(function ($annotation, $server) {
|
|
$annotation->runtime(MODEL_PATH);
|
|
|
|
name($server->worker_pid, 'Task.' . $server->worker_id);
|
|
})
|
|
->catch(function (\Throwable $throwable) {
|
|
$this->addError($throwable->getMessage());
|
|
})
|
|
->exec($annotation, $this->server);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param int $worker_id
|
|
*
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
public function onHandler(Server $server, int $worker_id): void
|
|
{
|
|
$this->server = $server;
|
|
|
|
$this->injectLoader($this->isWorker($worker_id));
|
|
|
|
if ($worker_id < $server->setting['worker_num']) {
|
|
$this->onWorker($server, $worker_id);
|
|
} else {
|
|
$this->onTask($server, $worker_id);
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param int $worker_id
|
|
* @return bool
|
|
*/
|
|
private function isWorker(int $worker_id): bool
|
|
{
|
|
return $worker_id < $this->server->setting['worker_num'];
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param int $worker_id
|
|
* @throws Exception
|
|
*/
|
|
public function onTask(Server $server, int $worker_id)
|
|
{
|
|
putenv('environmental=' . Snowflake::TASK);
|
|
|
|
Snowflake::setTaskId($server->worker_pid);
|
|
|
|
fire(Event::SERVER_TASK_START);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param int $worker_id
|
|
* @throws Exception
|
|
*/
|
|
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, 'throwable');
|
|
write($exception->getMessage(), 'worker');
|
|
}
|
|
}
|
|
|
|
|
|
}
|