eee
This commit is contained in:
+156
-86
@@ -12,149 +12,219 @@ use Throwable;
|
||||
|
||||
abstract class Websocket
|
||||
{
|
||||
|
||||
#[Container(Transport::class)]
|
||||
public Transport $transport;
|
||||
|
||||
public Config $config;
|
||||
|
||||
/**
|
||||
* @var Config
|
||||
*/
|
||||
public Config $config {
|
||||
get {
|
||||
return $this->config;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param Config $config
|
||||
* @return void
|
||||
*/
|
||||
public function start(Config $config): void
|
||||
{
|
||||
$this->config = $config;
|
||||
$this->configureRuntime();
|
||||
$this->onStart();
|
||||
|
||||
$server = new Server($config->host, $config->port, false);
|
||||
$server->handle('/websocket', fn(Request $request, Response $ws) => $this->handler($request, $ws));
|
||||
$server->handle('/online/lists', fn(Request $request, Response $ws) => $this->getLists($request, $ws));
|
||||
$this->configureServer($server);
|
||||
|
||||
$server->handle($config->normalizePath($config->websocketPath), fn(Request $request, Response $ws) => $this->handler($request, $ws));
|
||||
if ($config->exposeOnlineLists) {
|
||||
$server->handle($config->normalizePath($config->onlineListPath), fn(Request $request, Response $ws) => $this->getLists($request, $ws));
|
||||
}
|
||||
|
||||
echo 'websocket server start at ' . $config->host . ':' . $config->port . PHP_EOL;
|
||||
$server->start();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param Request $request
|
||||
* @param Response $ws
|
||||
* @return void
|
||||
*/
|
||||
public function getLists(Request $request, Response $ws): void
|
||||
{
|
||||
$data = json_encode($this->transport->getLists(), JSON_UNESCAPED_UNICODE);
|
||||
$ws->end($data);
|
||||
try {
|
||||
if (!$this->config->exposeOnlineLists) {
|
||||
throw new \RuntimeException('Not found.', 404);
|
||||
}
|
||||
|
||||
$psr7Request = ConstrictRequest::builder($request);
|
||||
if (!$this->authorizeOnlineListRequest($psr7Request)) {
|
||||
throw new \RuntimeException('Unauthorized.', 401);
|
||||
}
|
||||
|
||||
$payload = json_encode($this->transport->getLists(), JSON_UNESCAPED_UNICODE | JSON_UNESCAPED_SLASHES);
|
||||
if ($payload === false) {
|
||||
$payload = '[]';
|
||||
}
|
||||
|
||||
@$ws->header('Content-Type', 'application/json; charset=utf-8');
|
||||
$ws->end($payload);
|
||||
} catch (Throwable $throwable) {
|
||||
$this->throwable($ws, $throwable, false);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
/**
|
||||
* @param Request $request
|
||||
* @param Response $ws
|
||||
* @return void
|
||||
*/
|
||||
public function handler(Request $request, Response $ws): void
|
||||
{
|
||||
$upgraded = false;
|
||||
$userId = null;
|
||||
$fd = null;
|
||||
|
||||
try {
|
||||
$psr7Request = ConstrictRequest::builder($request);
|
||||
if (!$psr7Request->hasQuery($this->config->authKey)) {
|
||||
throw new \Exception('Params required.');
|
||||
}
|
||||
|
||||
if (!$this->onConnected($psr7Request) || $psr7Request->getAuthority() == null) {
|
||||
throw new \Exception('Token error, unable to obtain user.');
|
||||
}
|
||||
|
||||
if ($this->transport->has($psr7Request->getAuthority()->getUniqueId())) {
|
||||
$this->transport->remove($psr7Request->getAuthority()->getUniqueId());
|
||||
}
|
||||
$psr7Request = $this->authorizeConnection($request);
|
||||
|
||||
if (!$ws->upgrade()) {
|
||||
throw new \Exception('Connection upgrade to websocket failed.');
|
||||
throw new \RuntimeException('Connection upgrade to websocket failed.', 500);
|
||||
}
|
||||
|
||||
$upgraded = true;
|
||||
$userId = $psr7Request->getAuthority()->getUniqueId();
|
||||
$ws->push($userId);
|
||||
defer(function () use ($userId) {
|
||||
$this->transport->remove($userId);
|
||||
|
||||
$connection = new Struct($psr7Request->getAuthority(), $request, $ws);
|
||||
$fd = $connection->fd;
|
||||
|
||||
$this->transport->add($userId, $connection);
|
||||
defer(function () use ($userId, $fd) {
|
||||
$this->transport->remove($userId, $fd);
|
||||
});
|
||||
|
||||
$this->transport->add($userId, new Struct($psr7Request->getAuthority(), $request, $ws));
|
||||
@$ws->push((string)$userId);
|
||||
|
||||
$this->onBeforeMessageLoop();
|
||||
|
||||
while (true) {
|
||||
$frame = $ws->recv();
|
||||
if ($frame === '' || $frame === false || $frame->data == 'close' || get_class($frame) === CloseFrame::class) {
|
||||
$ws->close();
|
||||
if ($frame === '' || $frame === false || $frame === null) {
|
||||
break;
|
||||
}
|
||||
|
||||
$this->onMessage($psr7Request, $frame->data);
|
||||
if ($frame instanceof CloseFrame) {
|
||||
break;
|
||||
}
|
||||
|
||||
$payload = (string)($frame->data ?? '');
|
||||
if ($payload === 'close') {
|
||||
break;
|
||||
}
|
||||
|
||||
$this->assertPayloadSize($payload);
|
||||
$this->transport->touch($userId, $fd);
|
||||
$this->onMessage($psr7Request, $payload);
|
||||
}
|
||||
} catch (Throwable $throwable) {
|
||||
$this->throwable($ws, $throwable);
|
||||
$this->throwable($ws, $throwable, $upgraded);
|
||||
} finally {
|
||||
if (isset($userId)) {
|
||||
if ($userId !== null) {
|
||||
$this->onDisconnect($userId);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
abstract public function onBeforeMessageLoop(): void;
|
||||
|
||||
|
||||
/**
|
||||
* @param $ws
|
||||
* @param Throwable $throwable
|
||||
* @return void
|
||||
*/
|
||||
private function throwable($ws, Throwable $throwable): void
|
||||
protected function authorizeConnection(Request $request): ConstrictRequest
|
||||
{
|
||||
$error = sprintf("Message: %s \n
|
||||
File: %s \n
|
||||
Line: %d \n", $throwable->getMessage(), $throwable->getFile(), $throwable->getLine());
|
||||
$psr7Request = ConstrictRequest::builder($request);
|
||||
if (!$psr7Request->hasQuery($this->config->authKey)) {
|
||||
throw new \RuntimeException('Missing auth parameter.', 400);
|
||||
}
|
||||
|
||||
$ws->setStatusCode(500);
|
||||
$ws->end($throwable->getMessage());
|
||||
$ws->close();
|
||||
if (!$this->onConnected($psr7Request) || $psr7Request->getAuthority() === null) {
|
||||
throw new \RuntimeException('Unauthorized.', 401);
|
||||
}
|
||||
|
||||
echo $error;
|
||||
return $psr7Request;
|
||||
}
|
||||
|
||||
protected function authorizeOnlineListRequest(ConstrictRequest $request): bool
|
||||
{
|
||||
if (!$request->hasQuery($this->config->authKey)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
return $this->canViewOnlineLists($request) && $request->getAuthority() !== null;
|
||||
}
|
||||
|
||||
protected function canViewOnlineLists(ConstrictRequest $request): bool
|
||||
{
|
||||
return $this->onConnected($request);
|
||||
}
|
||||
|
||||
private function configureRuntime(): void
|
||||
{
|
||||
if ($this->config->maxCoroutine > 0 && class_exists(\Swoole\Coroutine::class)) {
|
||||
\Swoole\Coroutine::set([
|
||||
'max_coroutine' => $this->config->maxCoroutine,
|
||||
]);
|
||||
}
|
||||
}
|
||||
|
||||
private function configureServer(Server $server): void
|
||||
{
|
||||
if (!method_exists($server, 'set')) {
|
||||
return;
|
||||
}
|
||||
|
||||
$settings = [];
|
||||
if ($this->config->maxPackageSize > 0) {
|
||||
$settings['package_max_length'] = $this->config->maxPackageSize;
|
||||
}
|
||||
if ($this->config->maxFrameSize > 0) {
|
||||
$settings['websocket_max_frame_size'] = $this->config->maxFrameSize;
|
||||
}
|
||||
|
||||
if ($settings !== []) {
|
||||
$server->set($settings);
|
||||
}
|
||||
}
|
||||
|
||||
private function assertPayloadSize(string $payload): void
|
||||
{
|
||||
$size = strlen($payload);
|
||||
|
||||
if ($this->config->maxPackageSize > 0 && $size > $this->config->maxPackageSize) {
|
||||
throw new \RuntimeException('Payload too large.', 413);
|
||||
}
|
||||
|
||||
if ($this->config->maxFrameSize > 0 && $size > $this->config->maxFrameSize) {
|
||||
throw new \RuntimeException('Frame too large.', 413);
|
||||
}
|
||||
}
|
||||
|
||||
private function throwable(Response $ws, Throwable $throwable, bool $upgraded): void
|
||||
{
|
||||
$status = $throwable->getCode();
|
||||
if (!in_array($status, [400, 401, 403, 404, 409, 413, 500], true)) {
|
||||
$status = 500;
|
||||
}
|
||||
|
||||
$message = match ($status) {
|
||||
400 => 'Bad Request',
|
||||
401 => 'Unauthorized',
|
||||
403 => 'Forbidden',
|
||||
404 => 'Not Found',
|
||||
409 => 'Conflict',
|
||||
413 => 'Payload Too Large',
|
||||
default => 'Internal Server Error',
|
||||
};
|
||||
|
||||
error_log(sprintf(
|
||||
'[%s] %s in %s:%d',
|
||||
static::class,
|
||||
$throwable->getMessage(),
|
||||
$throwable->getFile(),
|
||||
$throwable->getLine(),
|
||||
));
|
||||
|
||||
if ($upgraded) {
|
||||
@$ws->close();
|
||||
return;
|
||||
}
|
||||
|
||||
@$ws->setStatusCode($status);
|
||||
$ws->end($message);
|
||||
}
|
||||
|
||||
/**
|
||||
* @param ConstrictRequest $request
|
||||
* @return bool
|
||||
*/
|
||||
abstract public function onConnected(ConstrictRequest $request): bool;
|
||||
|
||||
|
||||
/**
|
||||
* @return void
|
||||
*/
|
||||
abstract public function onStart(): void;
|
||||
|
||||
|
||||
/**
|
||||
* @param ConstrictRequest $psr7Request
|
||||
* @param string $frame
|
||||
* @return void
|
||||
*/
|
||||
abstract public function onMessage(ConstrictRequest $psr7Request, string $frame): void;
|
||||
|
||||
/**
|
||||
* @param int $userId
|
||||
* @return void
|
||||
*/
|
||||
abstract public function onDisconnect(int $userId): void;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user