This commit is contained in:
Chris Boden 2013-03-31 11:50:39 -04:00
parent 0be4299fd6
commit 777e928cea
2 changed files with 47 additions and 47 deletions

View File

@ -88,7 +88,10 @@ class WsServer implements MessageComponentInterface {
* {@inheritdoc} * {@inheritdoc}
*/ */
public function onMessage(ConnectionInterface $from, $msg) { public function onMessage(ConnectionInterface $from, $msg) {
if (true !== $from->WebSocket->established) { if (true === $from->WebSocket->established) {
return $from->WebSocket->version->onMessage($this->connections[$from], $msg);
}
if (isset($from->WebSocket->request)) { if (isset($from->WebSocket->request)) {
$from->WebSocket->request->getBody()->write($msg); $from->WebSocket->request->getBody()->write($msg);
} else { } else {
@ -113,13 +116,13 @@ class WsServer implements MessageComponentInterface {
} catch (\UnderflowException $e) { } catch (\UnderflowException $e) {
return; return;
} }
$response->setHeader('X-Powered-By', \Ratchet\VERSION);
// This needs to be refactored later on, incorporated with routing // This needs to be refactored later on, incorporated with routing
if ('' !== ($agreedSubProtocols = $this->getSubProtocolString($from->WebSocket->request->getTokenizedHeader('Sec-WebSocket-Protocol', ',')))) { if ('' !== ($agreedSubProtocols = $this->getSubProtocolString($from->WebSocket->request->getTokenizedHeader('Sec-WebSocket-Protocol', ',')))) {
$response->setHeader('Sec-WebSocket-Protocol', $agreedSubProtocols); $response->setHeader('Sec-WebSocket-Protocol', $agreedSubProtocols);
} }
$response->setHeader('X-Powered-By', \Ratchet\VERSION);
$from->send((string)$response); $from->send((string)$response);
if (101 != $response->getStatusCode()) { if (101 != $response->getStatusCode()) {
@ -135,9 +138,6 @@ class WsServer implements MessageComponentInterface {
return $this->_decorating->onOpen($upgraded); return $this->_decorating->onOpen($upgraded);
} }
$from->WebSocket->version->onMessage($this->connections[$from], $msg);
}
/** /**
* {@inheritdoc} * {@inheritdoc}
*/ */