Send additional headers on close connection in Router.

This commit is contained in:
samizdam 2016-11-17 19:42:22 +03:00
parent 5eb1dfa98d
commit 7215ffe9e6

View File

@ -34,7 +34,7 @@ class Router implements HttpServerInterface {
try { try {
$route = $this->_matcher->match($request->getPath()); $route = $this->_matcher->match($request->getPath());
} catch (MethodNotAllowedException $nae) { } catch (MethodNotAllowedException $nae) {
return $this->close($conn, 405); return $this->close($conn, 405, array('Allow' => $nae->getAllowedMethods()));
} catch (ResourceNotFoundException $nfe) { } catch (ResourceNotFoundException $nfe) {
return $this->close($conn, 404); return $this->close($conn, 404);
} }
@ -91,13 +91,15 @@ class Router implements HttpServerInterface {
/** /**
* Close a connection with an HTTP response * Close a connection with an HTTP response
* @param \Ratchet\ConnectionInterface $conn * @param \Ratchet\ConnectionInterface $conn
* @param int $code HTTP status code * @param int $code HTTP status code
* @param array $additionalHeaders
* @return null * @return null
*/ */
protected function close(ConnectionInterface $conn, $code = 400) { protected function close(ConnectionInterface $conn, $code = 400, array $additionalHeaders = array()) {
$response = new Response($code, array( $headers = array_merge(array(
'X-Powered-By' => \Ratchet\VERSION 'X-Powered-By' => \Ratchet\VERSION
)); ), $additionalHeaders);
$response = new Response($code, $headers);
$conn->send((string)$response); $conn->send((string)$response);
$conn->close(); $conn->close();