faeez/mxmbsocket
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Repo. Project Releases Wiki Activity

* @param int $keepAliveInterval Seconds between ping calls

Browse Source
This commit is contained in:
Illia Kovalov 2017-10-04 12:03:25 +02:00
parent 467c4552a5
commit e2b7a8f95a
2 changed files with 23 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/Ratchet/App.php
View File

@ -97,24 +97,21 @@ class App {
* @param ComponentInterface $controller Your application to server for the route. If not specified, assumed to be for a WebSocket
* @param array $allowedOrigins An array of hosts allowed to connect (same host by default), ['*'] for any
* @param string $httpHost Override the $httpHost variable provided in the __construct
* @param int $keepAliveInterval Seconds between ping calls. Works for WsServer only
* @return ComponentInterface|WsServer
*/
public function route($path, ComponentInterface $controller, array $allowedOrigins = array(), $httpHost = null, $keepAliveInterval = 30) {
public function route($path, ComponentInterface $controller, array $allowedOrigins = array(), $httpHost = null) {
if ($controller instanceof HttpServerInterface || $controller instanceof WsServer) {
$decorated = $controller;
} elseif ($controller instanceof WampServerInterface) {
$decorated = new WsServer(new WampServer($controller));
$decorated->enableKeepAlive($this->_server->loop);
} elseif ($controller instanceof MessageComponentInterface) {
$decorated = new WsServer($controller);
$decorated->enableKeepAlive($this->_server->loop);
} else {
$decorated = $controller;
}
if ($decorated instanceof WsServer) {
$decorated->enableKeepAlive($this->_server->loop, $keepAliveInterval);
}
if ($httpHost === null) {
$httpHost = $this->httpHost;
}

23
src/Ratchet/WebSocket/WsServer.php
View File

@ -61,11 +61,22 @@ class WsServer implements HttpServerInterface {
*/
private $msgCb;
/**
* @var bool
*/
private $keepAliveEnabled;
/**
* @var int
*/
private $keepAliveTimeout = 30;
/**
* @param \Ratchet\WebSocket\MessageComponentInterface|\Ratchet\MessageComponentInterface $component Your application to run with WebSockets
* @param int $keepAliveTimeout
* @note If you want to enable sub-protocols have your component implement WsServerInterface as well
*/
public function __construct(ComponentInterface $component) {
public function __construct(ComponentInterface $component, $keepAliveTimeout = 30) {
if ($component instanceof MessageComponentInterface) {
$this->msgCb = function(ConnectionInterface $conn, MessageInterface $msg) {
$this->delegate->onMessage($conn, $msg);
@ -88,6 +99,7 @@ class WsServer implements HttpServerInterface {
$this->closeFrameChecker = new CloseFrameChecker;
$this->handshakeNegotiator = new ServerNegotiator(new RequestVerifier);
$this->handshakeNegotiator->setStrictSubProtocolCheck(true);
$this->keepAliveEnabled = false;
if ($component instanceof WsServerInterface) {
$this->handshakeNegotiator->setSupportedSubProtocols($component->getSubProtocols());
@ -99,6 +111,7 @@ class WsServer implements HttpServerInterface {
$this->ueFlowFactory = function() use ($reusableUnderflowException) {
return $reusableUnderflowException;
};
$this->keepAliveTimeout = $keepAliveTimeout;
}
/**
@ -195,7 +208,11 @@ class WsServer implements HttpServerInterface {
$this->handshakeNegotiator->setStrictSubProtocolCheck($enable);
}
public function enableKeepAlive(LoopInterface $loop, $interval = 30) {
public function enableKeepAlive(LoopInterface $loop) {
if ($this->keepAliveEnabled) {
return;
}
$this->keepAliveEnabled = true;
$lastPing = new Frame(uniqid(), true, Frame::OP_PING);
$pingedConnections = new \SplObjectStorage;
$splClearer = new \SplObjectStorage;
@ -206,7 +223,7 @@ class WsServer implements HttpServerInterface {
}
};
$loop->addPeriodicTimer((int)$interval, function() use ($pingedConnections, &$lastPing, $splClearer) {
$loop->addPeriodicTimer((int)$this->keepAliveTimeout, function() use ($pingedConnections, &$lastPing, $splClearer) {
foreach ($pingedConnections as $wsConn) {
$wsConn->close();
}