as2252258/kiri-http-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

qqq

Browse Source
This commit is contained in:
xl
2023-05-26 18:26:45 +08:00
parent a42daff851
commit 29357fbfb9
1 changed files with 108 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Abstracts/WebSocket.php
+108 -92
View File
@@ -4,9 +4,11 @@ namespace Kiri\Server\Abstracts;
use Exception;
use Kiri\Abstracts\Component;
use Kiri\Config\ConfigProvider;
use Kiri\Router\DataGrip;
use Kiri\Router\Handler;
use Kiri\Router\RouterCollector;
use Kiri\Server\Constant;
use Kiri\Server\Contract\OnCloseInterface;
use Kiri\Server\Contract\OnDisconnectInterface;
use Kiri\Server\Contract\OnHandshakeInterface;
@@ -22,117 +24,131 @@ class WebSocket extends Component implements OnHandshakeInterface, OnMessageInte
{
public Handler $handler;
public Handler $handler;
/**
* @var RouterCollector
*/
public RouterCollector $collector;
/**
* @var RouterCollector
*/
public RouterCollector $collector;
/**
* @return void
* @throws ReflectionException
*/
public function init(): void
{
$this->collector = \Kiri::getDi()->get(DataGrip::class)->get('wss');
/**
* @return void
* @throws ReflectionException
*/
public function init(): void
{
$this->collector = \Kiri::getDi()->get(DataGrip::class)->get('wss');
$this->handler = $this->collector->query('/', 'GET');
}
$this->handler = $this->collector->query('/', 'GET');
$config = \Kiri::getDi()->get(ConfigProvider::class);
if ($this->handler->implement(OnHandshakeInterface::class)) {
$config->modify('server.ports.events.' . Constant::HANDSHAKE, [$this->handler, 'onHandshake']);
}
if ($this->handler->implement(OnMessageInterface::class)) {
$config->modify('server.ports.events.' . Constant::MESSAGE, [$this->handler, 'onMessage']);
}
if ($this->handler->implement(OnDisconnectInterface::class)) {
$config->modify('server.ports.events.' . Constant::DISCONNECT, [$this->handler, 'onDisconnect']);
}
if ($this->handler->implement(OnCloseInterface::class)) {
$config->modify('server.ports.events.' . Constant::CLOSE, [$this->handler, 'onClose']);
}
}
/**
* @param Request $request
* @param Response $response
* @return void
* @throws Exception
*/
public function onHandshake(Request $request, Response $response): void
{
// TODO: Implement onHandshake() method.
$secWebSocketKey = $request->header['sec-websocket-key'];
$patten = '#^[+/0-9A-Za-z]{21}[AQgw]==$#';
if (0 === preg_match($patten, $secWebSocketKey) || 16 !== strlen(base64_decode($secWebSocketKey))) {
throw new Exception('protocol error.', 500);
}
$key = base64_encode(sha1($request->header['sec-websocket-key'] . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', TRUE));
$headers = [
'Upgrade' => 'websocket',
'Connection' => 'Upgrade',
'Sec-websocket-Accept' => $key,
'Sec-websocket-Version' => '13',
];
if (isset($request->header['sec-websocket-protocol'])) {
$explode = explode(',', $request->header['sec-websocket-protocol']);
$headers['Sec-websocket-Protocol'] = $explode[0];
}
foreach ($headers as $key => $val) {
$response->setHeader($key, $val);
}
if ($this->handler->implement(OnHandshakeInterface::class)) {
$handler = $this->handler->getClass();
/**
* @param Request $request
* @param Response $response
* @return void
* @throws Exception
*/
public function onHandshake(Request $request, Response $response): void
{
// TODO: Implement onHandshake() method.
$secWebSocketKey = $request->header['sec-websocket-key'];
$patten = '#^[+/0-9A-Za-z]{21}[AQgw]==$#';
if (0 === preg_match($patten, $secWebSocketKey) || 16 !== strlen(base64_decode($secWebSocketKey))) {
throw new Exception('protocol error.', 500);
}
$key = base64_encode(sha1($request->header['sec-websocket-key'] . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', TRUE));
$headers = [
'Upgrade' => 'websocket',
'Connection' => 'Upgrade',
'Sec-websocket-Accept' => $key,
'Sec-websocket-Version' => '13',
];
if (isset($request->header['sec-websocket-protocol'])) {
$explode = explode(',', $request->header['sec-websocket-protocol']);
$headers['Sec-websocket-Protocol'] = $explode[0];
}
foreach ($headers as $key => $val) {
$response->setHeader($key, $val);
}
if ($this->handler->implement(OnHandshakeInterface::class)) {
$handler = $this->handler->getClass();
call_user_func([$handler, 'onHandshake'], $request, $response);
} else {
$response->setStatusCode(101);
$response->end();
}
}
call_user_func([$handler, 'onHandshake'], $request, $response);
} else {
$response->setStatusCode(101);
$response->end();
}
}
/**
* @param Server $server
* @param Frame $frame
* @return void
*/
public function onMessage(Server $server, Frame $frame): void
{
// TODO: Implement onMessage() method.
if (!$this->handler->implement(OnMessageInterface::class)) {
return;
}
/**
* @param Server $server
* @param Frame $frame
* @return void
*/
public function onMessage(Server $server, Frame $frame): void
{
// TODO: Implement onMessage() method.
if (!$this->handler->implement(OnMessageInterface::class)) {
return;
}
$handler = $this->handler->getClass();
$handler = $this->handler->getClass();
call_user_func([$handler, 'onMessage'], $server, $frame);
}
call_user_func([$handler, 'onMessage'], $server, $frame);
}
/**
* @param \Swoole\WebSocket\Server $server
* @param int $fd
* @return void
*/
public function onDisconnect(\Swoole\WebSocket\Server $server, int $fd): void
{
// TODO: Implement onDisconnect() method.
if (!$this->handler->implement(OnDisconnectInterface::class)) {
return;
}
/**
* @param \Swoole\WebSocket\Server $server
* @param int $fd
* @return void
*/
public function onDisconnect(\Swoole\WebSocket\Server $server, int $fd): void
{
// TODO: Implement onDisconnect() method.
if (!$this->handler->implement(OnDisconnectInterface::class)) {
return;
}
$handler = $this->handler->getClass();
$handler = $this->handler->getClass();
call_user_func([$handler, 'onDisconnect'], $server, $fd);
}
call_user_func([$handler, 'onDisconnect'], $server, $fd);
}
/**
* @param Server $server
* @param int $fd
* @return void
*/
public function onClose(Server $server, int $fd): void
{
// TODO: Implement onDisconnect() method.
if (!$this->handler->implement(OnCloseInterface::class)) {
return;
}
/**
* @param Server $server
* @param int $fd
* @return void
*/
public function onClose(Server $server, int $fd): void
{
// TODO: Implement onDisconnect() method.
if (!$this->handler->implement(OnCloseInterface::class)) {
return;
}
$handler = $this->handler->getClass();
$handler = $this->handler->getClass();
call_user_func([$handler, 'onClose'], $server, $fd);
}
call_user_func([$handler, 'onClose'], $server, $fd);
}
}