add method to disable new connections to a WebSocket instance

This commit is contained in:
Me No Dev 2016-05-25 19:48:04 +03:00
parent 8171edd7e1
commit 2b501e74a7
2 changed files with 10 additions and 3 deletions

View File

@ -599,6 +599,7 @@ AsyncWebSocket::AsyncWebSocket(String url)
:_url(url) :_url(url)
,_clients(NULL) ,_clients(NULL)
,_cNextId(1) ,_cNextId(1)
,_enabled(true)
{ {
_eventHandler = NULL; _eventHandler = NULL;
} }
@ -898,6 +899,9 @@ const char * WS_STR_ACCEPT = "Sec-WebSocket-Accept";
const char * WS_STR_UUID = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"; const char * WS_STR_UUID = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
bool AsyncWebSocket::canHandle(AsyncWebServerRequest *request){ bool AsyncWebSocket::canHandle(AsyncWebServerRequest *request){
if(!_enabled)
return false;
if(request->method() != HTTP_GET || !request->url().equals(_url)) if(request->method() != HTTP_GET || !request->url().equals(_url))
return false; return false;

View File

@ -134,10 +134,13 @@ class AsyncWebSocket: public AsyncWebHandler {
AsyncWebSocketClient * _clients; AsyncWebSocketClient * _clients;
uint32_t _cNextId; uint32_t _cNextId;
AwsEventHandler _eventHandler; AwsEventHandler _eventHandler;
bool _enabled;
public: public:
AsyncWebSocket(String url); AsyncWebSocket(String url);
~AsyncWebSocket(); ~AsyncWebSocket();
const char * url(){ return _url.c_str(); } const char * url(){ return _url.c_str(); }
void enable(bool e){ _enabled = e; }
bool enabled() { return _enabled; }
size_t count(); size_t count();
AsyncWebSocketClient * client(uint32_t id); AsyncWebSocketClient * client(uint32_t id);