/**
* Close the channel, this operation will terminate all receivers that cannot receive any data.
*
* Closing a closed channel has no effect and does not throw an error!
*
* @param \Throwable $e Error will be propagated to all receivers that cannot receive a buffered value anymore.
*/
public function close(\Throwable $e = null)
{
if ($this->closed) {
return;
}
$this->closed = $e ?? true;
if ($this->generator) {
$this->generator->cancel($e ?? new \RuntimeException('Channel closed'));
}
// Dispose of queued receivers that cannot receive a value from the channel.
$sources = ($this->senders ? $this->countActiveSubscriptions($this->senders) : 0) + ($this->buffer ? $this->buffer->count() : 0);
$sinks = $this->receivers ? $this->countActiveSubscriptions($this->receivers) : 0;
$fails = new \SplStack();
$fails->setIteratorMode(\SplStack::IT_MODE_LIFO | \SplStack::IT_MODE_DELETE);
while ($sinks > $sources) {
$receiver = $this->receivers->pop();
if ($receiver->active) {
$sinks--;
$fails->push($receiver);
}
}
foreach ($fails as $subscription) {
if ($e) {
$subscription->fail($e);
} else {
$subscription->resolve($subscription->data);
}
}
}
/**
* Shut down connection.
*
* @return Awaitable
*/
public function shutdown() : Awaitable
{
if ($this->processor) {
try {
return new AwaitPending($this->processor->cancel(new \RuntimeException('WebSocket connection shutdown')));
} finally {
$this->processor = null;
}
}
return new Success(null);
}
protected function sendDeferredResponse(SocketStream $socket, HttpRequest $request, HttpResponse $response, bool $nobody, bool $close) : \Generator
{
if ($this->logger) {
$this->logger->info('{ip} "{method} {target} HTTP/{protocol}" {status} {size}', ['ip' => $request->getClientAddress(), 'method' => $request->getMethod(), 'target' => $request->getRequestTarget(), 'protocol' => $request->getProtocolVersion(), 'status' => $response->getStatusCode(), 'size' => '-']);
}
if (!$nobody) {
$close = true;
}
(yield $socket->write($this->serializeHeaders($response, $close, null, $nobody, true) . "\r\n"));
(yield $socket->flush());
if ($nobody) {
$response->getBody()->close(false);
return !$close;
}
$watcher = null;
$task = new Coroutine(function () use(&$watcher, $socket, $request, $response) {
$body = $response->getBody();
$body->start($request, $this->logger);
$bodyStream = (yield $body->getReadableStream());
$e = null;
try {
while (null !== ($chunk = (yield $bodyStream->read()))) {
(yield $socket->write($chunk));
}
} catch (StreamClosedException $e) {
// Client disconnected from server.
} finally {
try {
$bodyStream->close();
} finally {
$body->close($e ? true : false);
}
$watcher->cancel(new \RuntimeException());
}
return false;
}, true);
$watcher = new Coroutine(function () use($socket, $task) {
$socket = $socket->getSocket();
while (\is_resource($socket) && !\feof($socket)) {
(yield new AwaitRead($socket));
}
$task->cancel(new StreamClosedException('Client disconnected'));
});
return (yield $task);
}
/**
* Shutdown the inbound record handler of the connection.
*/
public function shutdown() : Awaitable
{
$stop = [];
if ($this->processor) {
$stop = $this->processor->cancel(new \RuntimeException('Connection closed'));
}
return new AwaitPending($stop);
}
