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

modify

Browse Source
This commit is contained in:
as2252258@163.com
2021-08-04 02:17:20 +08:00
parent 54cff0fb5d
commit 9e65b23120
3 changed files with 316 additions and 199 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Server/HTTPServerListener.php
+91 -82
View File
@@ -4,6 +4,7 @@ namespace Server;
use Annotation\Inject;
use Exception;
use HttpServer\Exception\RequestException;
use HttpServer\Http\Context;
use HttpServer\Http\Request as HSRequest;
use HttpServer\Route\Node;
@@ -27,107 +28,115 @@ use Throwable;
class HTTPServerListener extends Abstracts\Server
{
protected static bool|Port $_http;
protected static bool|Port $_http;
use ListenerHelper;
use ListenerHelper;
/** @var Router|mixed */
#[Inject('router')]
public Router $router;
/** @var Router|mixed */
#[Inject('router')]
public Router $router;
/** @var \HttpServer\Http\Response|mixed */
#[Inject(\HttpServer\Http\Response::class)]
public \HttpServer\Http\Response $response;
/** @var \HttpServer\Http\Response|mixed */
#[Inject(\HttpServer\Http\Response::class)]
public \HttpServer\Http\Response $response;
/** @var EventDispatch */
#[Inject(EventDispatch::class)]
public EventDispatch $eventDispatch;
/** @var EventDispatch */
#[Inject(EventDispatch::class)]
public EventDispatch $eventDispatch;
/**
* UDPServerListener constructor.
* @param Server|Port $server
* @param array|null $settings
* @return Server|Port
* @throws Exception
*/
public function bindCallback(Server|Port $server, ?array $settings = []): Server|Port
{
$this->setEvents(Constant::CONNECT, $settings['events'][Constant::CONNECT] ?? null);
/**
* UDPServerListener constructor.
* @param Server|Port $server
* @param array|null $settings
* @return Server|Port
* @throws Exception
*/
public function bindCallback(Server|Port $server, ?array $settings = []): Server|Port
{
$this->setEvents(Constant::CONNECT, $settings['events'][Constant::CONNECT] ?? null);
$server->set(array_merge($settings['settings'] ?? [], ['enable_unsafe_event' => false]));
if (isset($settings['events'][Constant::REQUEST])) {
$event = $settings['events'][Constant::REQUEST];
if (is_array($event) && is_string($event[0])) {
$event[0] = di($event[0]);
}
$server->on('request', $event);
} else {
$server->on('request', [$this, 'onRequest']);
}
$server->on('connect', [$this, 'onConnect']);
$server->on('close', [$this, 'onClose']);
return $server;
}
$server->set(array_merge($settings['settings'] ?? [], ['enable_unsafe_event' => false]));
if (isset($settings['events'][Constant::REQUEST])) {
$event = $settings['events'][Constant::REQUEST];
if (is_array($event) && is_string($event[0])) {
$event[0] = di($event[0]);
}
$server->on('request', $event);
} else {
$server->on('request', [$this, 'onRequest']);
}
$server->on('connect', [$this, 'onConnect']);
$server->on('close', [$this, 'onClose']);
return $server;
}
/**
* @param Server $server
* @param int $fd
* @throws Exception
*/
public function onConnect(Server $server, int $fd)
{
$this->runEvent(Constant::CONNECT, null, [$server, $fd]);
}
/**
* @param Server $server
* @param int $fd
* @throws Exception
*/
public function onConnect(Server $server, int $fd)
{
$this->runEvent(Constant::CONNECT, null, [$server, $fd]);
}
/**
* @param Request $request
* @param Response $response
* @throws Exception
*/
public function onRequest(Request $request, Response $response)
{
try {
/**
* @param Request $request
* @param Response $response
* @throws Exception
*/
public function onRequest(Request $request, Response $response)
{
try {
$node = $this->router->find_path(HSRequest::create($request));
if (!($node instanceof Node)) {
$this->response->setStatusCode(404);
$this->response->setFormat(\HttpServer\Http\Response::HTML);
$data = '<h2>HTTP 404 Not Found</h2><hr><i>Powered by Swoole</i>';
} else {
$this->response->setStatusCode(200);
$data = $node->dispatch();
throw new RequestException('<h2>HTTP 404 Not Found</h2><hr><i>Powered by Swoole</i>', 404);
}
$data = $node->dispatch();
} catch (Error | Throwable $exception) {
$this->response->setStatusCode(500);
$data = jTraceEx($exception);
} finally {
$this->response->send($data, $response);
$this->eventDispatch->dispatch(new OnAfterRequest());
}
}
$code = $exception->getCode() == 0 ? 500 : $exception->getCode();
$data = $code ? $exception->getMessage() : jTraceEx($exception);
} finally {
$this->requestEnd($data, $response);
}
}
/**
* @param Server $server
* @param int $fd
* @throws Exception
*/
public function onDisconnect(Server $server, int $fd)
{
}
/**
* @param mixed $data
* @param \Swoole\Http\Response $response
*/
protected function requestEnd(mixed $data, Response $response)
{
$sResponse = $this->response->getBuilder($data, $response);
$response->end($sResponse->getContent());
$this->eventDispatch->dispatch(new OnAfterRequest());
}
/**
* @param Server $server
* @param int $fd
* @throws Exception
*/
public function onClose(Server $server, int $fd)
{
}
/**
* @param Server $server
* @param int $fd
* @throws Exception
*/
public function onDisconnect(Server $server, int $fd)
{
}
/**
* @param Server $server
* @param int $fd
* @throws Exception
*/
public function onClose(Server $server, int $fd)
{
}
}