Skip to content

Commit

Permalink
[1.x] Ensure connection close handler is cleaned up for each request
Browse files Browse the repository at this point in the history
This changeset resolves a small memory leak that causes roughly 1KB per connection tops. Which isn't a big issue but will make memory fluctuate more. The changeset doesn't introduce any performance degradation.

Resolves: reactphp#514
Builds on top of: reactphp#405, reactphp#467, and many others
  • Loading branch information
WyriHaximus committed Feb 19, 2024
1 parent 706edec commit 34a3471
Show file tree
Hide file tree
Showing 2 changed files with 47 additions and 3 deletions.
12 changes: 9 additions & 3 deletions src/Io/StreamingServer.php
Original file line number Diff line number Diff line change
Expand Up @@ -157,10 +157,12 @@ public function handleRequest(ConnectionInterface $conn, ServerRequestInterface
}

// cancel pending promise once connection closes
$connectionOnCloseResponseCanceler = null;
if ($response instanceof PromiseInterface && \method_exists($response, 'cancel')) {
$conn->on('close', function () use ($response) {
$connectionOnCloseResponseCanceler = function () use ($response) {
$response->cancel();
});
};
$conn->on('close', $connectionOnCloseResponseCanceler);
}

// happy path: response returned, handle and return immediately
Expand Down Expand Up @@ -201,7 +203,11 @@ function ($error) use ($that, $conn, $request) {
$that->emit('error', array($exception));
return $that->writeError($conn, Response::STATUS_INTERNAL_SERVER_ERROR, $request);
}
);
)->always(static function () use ($connectionOnCloseResponseCanceler, $conn) {
if ($connectionOnCloseResponseCanceler !== null) {
$conn->removeListener('close', $connectionOnCloseResponseCanceler);
}
});
}

/** @internal */
Expand Down
38 changes: 38 additions & 0 deletions tests/Io/StreamingServerTest.php
Original file line number Diff line number Diff line change
Expand Up @@ -3245,6 +3245,44 @@ public function testNewConnectionWillInvokeParserTwiceAfterInvokingRequestHandle
$this->assertCount(1, $this->connection->listeners('close'));
}

public function testCompletingARequestWillRemoveConnectionOnCloseListener()
{
$connection = $this->getMockBuilder('React\Socket\Connection')
->disableOriginalConstructor()
->setMethods(
array(
'write',
'end',
'close',
'pause',
'resume',
'isReadable',
'isWritable',
'getRemoteAddress',
'getLocalAddress',
'pipe',
'removeListener'
)
)
->getMock();

$connection->method('isWritable')->willReturn(true);
$connection->method('isReadable')->willReturn(true);
$request = new ServerRequest('GET', 'http://localhost/');

$server = new StreamingServer(Loop::get(), function () {
return \React\Promise\resolve(new Response());
});

$server->listen($this->socket);
$this->socket->emit('connection', array($connection));

$connection->expects($this->once())->method('removeListener');

// pretend parser just finished parsing
$server->handleRequest($connection, $request);
}

private function createGetRequest()
{
$data = "GET / HTTP/1.1\r\n";
Expand Down

0 comments on commit 34a3471

Please sign in to comment.