155 lines
3.2 KiB
PHP
155 lines
3.2 KiB
PHP
<?php
|
|
|
|
namespace Rpc;
|
|
|
|
|
|
use Exception;
|
|
use HttpServer\Http\Context;
|
|
use HttpServer\Http\Request;
|
|
use HttpServer\Route\Router;
|
|
use Server\Constant;
|
|
use Snowflake\Core\Json;
|
|
use Snowflake\Event;
|
|
use Snowflake\Snowflake;
|
|
use Swoole\Server;
|
|
use function Swoole\Coroutine\defer;
|
|
|
|
|
|
/**
|
|
* Class Service
|
|
* @package Rpc
|
|
*/
|
|
class Service extends \Server\Abstracts\Server
|
|
{
|
|
|
|
const RPC_CONNECT = 'RPC::CONNECT';
|
|
const RPC_CLOSE = 'RPC::CLOSE';
|
|
|
|
private Router $router;
|
|
|
|
|
|
/**
|
|
* @throws Exception
|
|
*/
|
|
public function init()
|
|
{
|
|
$this->router = Snowflake::getApp('router');
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param int $fd
|
|
* @param int $reactorId
|
|
* @throws Exception
|
|
*/
|
|
public function onConnect(Server $server, int $fd, int $reactorId)
|
|
{
|
|
defer(fn() => fire(Event::SYSTEM_RESOURCE_RELEASES));
|
|
|
|
$this->runEvent(Constant::CONNECT, null, [$server, $fd, $reactorId]);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param int $fd
|
|
* on tcp client close
|
|
* @throws Exception
|
|
*/
|
|
public function onClose(Server $server, int $fd)
|
|
{
|
|
defer(fn() => fire(Event::SYSTEM_RESOURCE_RELEASES));
|
|
|
|
$this->runEvent(Constant::CLOSE, null, [$server, $fd]);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param int $fd
|
|
* on tcp client close
|
|
* @throws Exception
|
|
*/
|
|
public function onDisconnect(Server $server, int $fd)
|
|
{
|
|
defer(fn() => fire(Event::SYSTEM_RESOURCE_RELEASES));
|
|
|
|
$this->runEvent(Constant::DISCONNECT, null, [$server, $fd]);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param int $fd
|
|
* @param int $reID
|
|
* @param string $data
|
|
* @throws Exception
|
|
*/
|
|
public function onReceive(Server $server, int $fd, int $reID, string $data)
|
|
{
|
|
defer(fn() => fire(Event::SYSTEM_RESOURCE_RELEASES));
|
|
try {
|
|
$client = $server->getClientInfo($fd, $reID);
|
|
|
|
$request = $this->requestSpl((int)$client['server_port'], $data);
|
|
|
|
$result = $this->router->find_path($request)?->dispatch();
|
|
|
|
$server->send($fd, $result);
|
|
} catch (\Throwable $exception) {
|
|
$server->send($fd, $exception->getMessage());
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Server $server
|
|
* @param string $data
|
|
* @param array $client
|
|
* @throws Exception
|
|
*/
|
|
public function onPacket(Server $server, string $data, array $client)
|
|
{
|
|
defer(fn() => fire(Event::SYSTEM_RESOURCE_RELEASES));
|
|
try {
|
|
$request = $this->requestSpl((int)$client['server_port'], $data);
|
|
|
|
$result = $this->router->find_path($request)?->dispatch();
|
|
|
|
$server->sendto($client['address'], $client['port'], $result);
|
|
} catch (\Throwable $exception) {
|
|
$server->sendto($client['address'], $client['port'], $exception->getMessage());
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* @param int $server_port
|
|
* @param string $data
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
public function requestSpl(int $server_port, string $data): mixed
|
|
{
|
|
$sRequest = new Request();
|
|
|
|
[$cmd, $repeat, $body] = explode("\n", $data);
|
|
if (is_null($body) || is_null($cmd) || !empty($repeat)) {
|
|
throw new Exception('Protocol format error.');
|
|
}
|
|
|
|
if (is_string($body) && is_null($data = Json::decode($body))) {
|
|
throw new Exception('Protocol format error.');
|
|
}
|
|
|
|
$sRequest->params->setPosts($data);
|
|
$sRequest->headers->setRequestUri('rpc/p' . $server_port . '/' . ltrim($cmd, '/'));
|
|
$sRequest->headers->setRequestMethod('rpc');
|
|
|
|
return Context::setContext(Request::class, $sRequest);
|
|
}
|
|
|
|
|
|
}
|