This commit is contained in:
2021-03-18 14:34:46 +08:00
parent fe1ffb960b
commit 65d6cd43fa
3 changed files with 38 additions and 30 deletions
+31 -26
View File
@@ -86,9 +86,22 @@ class OnHandshake extends Callback
*/ */
public function onHandler(SRequest $request, SResponse $response): void public function onHandler(SRequest $request, SResponse $response): void
{ {
$this->execute($request, $response); try {
fire(Event::SYSTEM_RESOURCE_RELEASES); $this->execute($request, $response);
logger()->insert();
$clientInfo = Snowflake::getWebSocket()->getClientInfo($request->fd);
$event = Snowflake::app()->getEvent();
$eventName = 'listen ' . $clientInfo['server_port'] . ' ' . Event::SERVER_HANDSHAKE;
$event->trigger($eventName, [$request, $response]);
} catch (\Throwable $exception) {
$this->addError($exception);
$this->disconnect($response, 500);
} finally {
fire(Event::SYSTEM_RESOURCE_CLEAN);
logger()->insert();
}
} }
@@ -100,30 +113,22 @@ class OnHandshake extends Callback
*/ */
private function execute(SRequest $request, SResponse $response): mixed private function execute(SRequest $request, SResponse $response): mixed
{ {
try { $this->resolveParse($request, $response);
$this->resolveParse($request, $response); if (isset($request->get['debug']) && $request->get['debug'] == 'test') {
if (isset($request->get['debug']) && $request->get['debug'] == 'test') { return $this->connect($response, 101);
return $this->connect($response, 101);
}
$router = Snowflake::app()->getRouter();
$sRequest = Request::create($request);
$sRequest->uri = '/' . Socket::HANDSHAKE . '::event';
$sRequest->headers->replace('request_method', 'sw::socket');
$sRequest->headers->replace('request_uri', $sRequest->uri);
$sRequest->parseUri();
if (($node = $router->find_path($sRequest)) === null) {
return $this->disconnect($response, 502);
}
return $node->dispatch($request, $response);
} catch (\Throwable $exception) {
$this->addError($exception);
return $this->disconnect($response, 500);
} finally {
fire(Event::SYSTEM_RESOURCE_CLEAN);
logger()->insert();
} }
$router = Snowflake::app()->getRouter();
$sRequest = Request::create($request);
$sRequest->uri = '/' . Socket::HANDSHAKE . '::event';
$sRequest->headers->replace('request_method', 'sw::socket');
$sRequest->headers->replace('request_uri', $sRequest->uri);
$sRequest->parseUri();
if (($node = $router->find_path($sRequest)) === null) {
return $this->disconnect($response, 502);
}
return $node->dispatch($request, $response);
} }
+6 -3
View File
@@ -62,10 +62,13 @@ class OnMessage extends Callback
* @return mixed * @return mixed
* @throws Exception * @throws Exception
*/ */
private function resolve($event, Frame $frame, $server): mixed private function resolve($event, Frame $frame,Server $server): mixed
{ {
if ($event->exists(Event::SERVER_MESSAGE)) { $clientInfo = $server->getClientInfo($frame->fd);
$event->trigger(Event::SERVER_MESSAGE, [$server, $frame]);
$eventName = 'listen ' . $clientInfo['server_port'] . ' ' . Event::SERVER_MESSAGE;
if ($event->exists($eventName)) {
$event->trigger($eventName, [$server, $frame]);
} else { } else {
$frame->data = json_decode($frame->data, true); $frame->data = json_decode($frame->data, true);
} }
+1 -1
View File
@@ -59,7 +59,7 @@ class Redis extends Pool
{ {
$this->printClients($config['host'], $name, true); $this->printClients($config['host'], $name, true);
$redis = new SRedis(); $redis = new SRedis();
if (!$redis->connect($config['host'], (int)$config['port'], $config['timeout'])) { if (!$redis->pconnect($config['host'], (int)$config['port'], $config['timeout'])) {
throw new RedisConnectException(sprintf('The Redis Connect %s::%d Fail.', $config['host'], $config['port'])); throw new RedisConnectException(sprintf('The Redis Connect %s::%d Fail.', $config['host'], $config['port']));
} }
if (empty($config['auth']) || !$redis->auth($config['auth'])) { if (empty($config['auth']) || !$redis->auth($config['auth'])) {