kiri-core/HttpServer/Events/OnMessage.php

<?php
declare(strict_types=1);

namespace HttpServer\Events;


use Annotation\Route\Socket;
use Exception;
use HttpServer\Abstracts\Callback;
use HttpServer\Http\Request;
use HttpServer\Route\Annotation\Websocket as AWebsocket;
use Snowflake\Event;
use Snowflake\Snowflake;
use Swoole\Coroutine;
use Swoole\WebSocket\Frame;
use Swoole\WebSocket\Server;

/**
 * Class OnMessage
 * @package HttpServer\Events
 */
class OnMessage extends Callback
{

	/**
	 * @param Server $server
	 * @param Frame $frame
	 * @throws
	 */
	public function onHandler(Server $server, Frame $frame)
	{
		try {
			if ($frame->opcode != 0x08) {
				$event = Snowflake::app()->getEvent();
				Coroutine::defer(function () use ($event) {
					$event->trigger(Event::EVENT_AFTER_REQUEST);
				});
				$content = $this->resolve($event, $frame, $server);
				$server->send($frame->fd, $content);
			}
		} catch (\Throwable $exception) {
			$this->addError($exception->getMessage(), 'websocket');
			$server->send($frame->fd, $exception->getMessage());
		} finally {
			logger()->insert();
		}
	}

	/**
	 * @param $event
	 * @param $frame
	 * @param $server
	 * @return mixed
	 * @throws Exception
	 */
	private function resolve($event, $frame, $server): mixed
	{
		if ($event->exists(Event::SERVER_MESSAGE)) {
			$event->trigger(Event::SERVER_MESSAGE, [$server, $frame]);
		} else {
			$frame->data = json_decode($frame->data, true);
		}
		if (empty($route = $frame->data['route'] ?? null)) {
			throw new Exception('Format error.');
		}
		$router = Snowflake::app()->getRouter();
		$node = $router->search(Socket::MESSAGE . '::' . $route, 'sw::socket');
		if ($node === null) {
			throw new Exception('Page not found.');
		}
		return $node->dispatch();
	}

}