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 Exception;
use Kiri\Abstracts\Component; use Kiri\Abstracts\Component;
use Kiri\Config\ConfigProvider;
use Kiri\Router\DataGrip; use Kiri\Router\DataGrip;
use Kiri\Router\Handler; use Kiri\Router\Handler;
use Kiri\Router\RouterCollector; use Kiri\Router\RouterCollector;
use Kiri\Server\Constant;
use Kiri\Server\Contract\OnCloseInterface; use Kiri\Server\Contract\OnCloseInterface;
use Kiri\Server\Contract\OnDisconnectInterface; use Kiri\Server\Contract\OnDisconnectInterface;
use Kiri\Server\Contract\OnHandshakeInterface; use Kiri\Server\Contract\OnHandshakeInterface;
@@ -22,117 +24,131 @@ class WebSocket extends Component implements OnHandshakeInterface, OnMessageInte
{ {
public Handler $handler; public Handler $handler;
/** /**
* @var RouterCollector * @var RouterCollector
*/ */
public RouterCollector $collector; public RouterCollector $collector;
/** /**
* @return void * @return void
* @throws ReflectionException * @throws ReflectionException
*/ */
public function init(): void public function init(): void
{ {
$this->collector = \Kiri::getDi()->get(DataGrip::class)->get('wss'); $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 Request $request
* @param Response $response * @param Response $response
* @return void * @return void
* @throws Exception * @throws Exception
*/ */
public function onHandshake(Request $request, Response $response): void public function onHandshake(Request $request, Response $response): void
{ {
// TODO: Implement onHandshake() method. // TODO: Implement onHandshake() method.
$secWebSocketKey = $request->header['sec-websocket-key']; $secWebSocketKey = $request->header['sec-websocket-key'];
$patten = '#^[+/0-9A-Za-z]{21}[AQgw]==$#'; $patten = '#^[+/0-9A-Za-z]{21}[AQgw]==$#';
if (0 === preg_match($patten, $secWebSocketKey) || 16 !== strlen(base64_decode($secWebSocketKey))) { if (0 === preg_match($patten, $secWebSocketKey) || 16 !== strlen(base64_decode($secWebSocketKey))) {
throw new Exception('protocol error.', 500); throw new Exception('protocol error.', 500);
} }
$key = base64_encode(sha1($request->header['sec-websocket-key'] . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', TRUE)); $key = base64_encode(sha1($request->header['sec-websocket-key'] . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', TRUE));
$headers = [ $headers = [
'Upgrade' => 'websocket', 'Upgrade' => 'websocket',
'Connection' => 'Upgrade', 'Connection' => 'Upgrade',
'Sec-websocket-Accept' => $key, 'Sec-websocket-Accept' => $key,
'Sec-websocket-Version' => '13', 'Sec-websocket-Version' => '13',
]; ];
if (isset($request->header['sec-websocket-protocol'])) { if (isset($request->header['sec-websocket-protocol'])) {
$explode = explode(',', $request->header['sec-websocket-protocol']); $explode = explode(',', $request->header['sec-websocket-protocol']);
$headers['Sec-websocket-Protocol'] = $explode[0]; $headers['Sec-websocket-Protocol'] = $explode[0];
} }
foreach ($headers as $key => $val) { foreach ($headers as $key => $val) {
$response->setHeader($key, $val); $response->setHeader($key, $val);
} }
if ($this->handler->implement(OnHandshakeInterface::class)) { if ($this->handler->implement(OnHandshakeInterface::class)) {
$handler = $this->handler->getClass(); $handler = $this->handler->getClass();
call_user_func([$handler, 'onHandshake'], $request, $response); call_user_func([$handler, 'onHandshake'], $request, $response);
} else { } else {
$response->setStatusCode(101); $response->setStatusCode(101);
$response->end(); $response->end();
} }
} }
/** /**
* @param Server $server * @param Server $server
* @param Frame $frame * @param Frame $frame
* @return void * @return void
*/ */
public function onMessage(Server $server, Frame $frame): void public function onMessage(Server $server, Frame $frame): void
{ {
// TODO: Implement onMessage() method. // TODO: Implement onMessage() method.
if (!$this->handler->implement(OnMessageInterface::class)) { if (!$this->handler->implement(OnMessageInterface::class)) {
return; 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 \Swoole\WebSocket\Server $server
* @param int $fd * @param int $fd
* @return void * @return void
*/ */
public function onDisconnect(\Swoole\WebSocket\Server $server, int $fd): void public function onDisconnect(\Swoole\WebSocket\Server $server, int $fd): void
{ {
// TODO: Implement onDisconnect() method. // TODO: Implement onDisconnect() method.
if (!$this->handler->implement(OnDisconnectInterface::class)) { if (!$this->handler->implement(OnDisconnectInterface::class)) {
return; 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 Server $server
* @param int $fd * @param int $fd
* @return void * @return void
*/ */
public function onClose(Server $server, int $fd): void public function onClose(Server $server, int $fd): void
{ {
// TODO: Implement onDisconnect() method. // TODO: Implement onDisconnect() method.
if (!$this->handler->implement(OnCloseInterface::class)) { if (!$this->handler->implement(OnCloseInterface::class)) {
return; return;
} }
$handler = $this->handler->getClass(); $handler = $this->handler->getClass();
call_user_func([$handler, 'onClose'], $server, $fd); call_user_func([$handler, 'onClose'], $server, $fd);
} }
} }