modify plugin name

This commit is contained in:
2022-02-14 14:28:46 +08:00
parent f272d8e451
commit 7999847ffa
+13 -8
View File
@@ -10,6 +10,7 @@ use Kiri\Server\ProcessManager;
use Kiri\Server\TraitServer; use Kiri\Server\TraitServer;
use Kiri\Task\AsyncTaskExecute; use Kiri\Task\AsyncTaskExecute;
use Kiri\Task\CoroutineTaskExecute; use Kiri\Task\CoroutineTaskExecute;
use Kiri\Websocket\FdCollector;
use Kiri\Websocket\Sender; use Kiri\Websocket\Sender;
use Psr\Container\ContainerExceptionInterface; use Psr\Container\ContainerExceptionInterface;
use Psr\Container\NotFoundExceptionInterface; use Psr\Container\NotFoundExceptionInterface;
@@ -152,21 +153,25 @@ class Http extends Component
$this->servers[$value['port']] = $server; $this->servers[$value['port']] = $server;
$server->handle('/', function (Request $request, Response $response) use ($handshake, $open, $close, $message) { $server->handle('/', function (Request $request, Response $response) use ($handshake, $open, $close, $message) {
$fdCollector = $this->getContainer()->get(FdCollector::class);
try { try {
if (is_null($handshake)) { $response->upgrade();
$response->upgrade(); $fdCollector->set($response->fd, $response);
} else { if (is_callable($open)) {
call_user_func($handshake, $request, $response); $open($request);
}
if ($response->isWritable() && is_callable($open)) {
call_user_func($open, $response);
} }
while (($data = $response->recv()) instanceof Frame) { while (($data = $response->recv()) instanceof Frame) {
call_user_func($message, $data); try {
call_user_func($message, $data);
} catch (\Throwable $throwable) {
$this->logger()->error($throwable->getMessage());
}
} }
call_user_func($close, $response->fd); call_user_func($close, $response->fd);
} catch (\Throwable $throwable) { } catch (\Throwable $throwable) {
$this->logger()->error($throwable->getMessage()); $this->logger()->error($throwable->getMessage());
} finally {
$fdCollector->remove($response->fd);
} }
}); });
} else { } else {