From 0adc36b1f6f5ee3dd98c3d88f513dc7aca0e2520 Mon Sep 17 00:00:00 2001
From: me-no-dev <me-no-dev@vdnsbg.com>
Date: Fri, 23 Sep 2016 21:56:28 +0300
Subject: [PATCH] fix mistake

---
 src/AsyncEventSource.cpp | 4 ++--
 src/AsyncWebSocket.cpp   | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/AsyncEventSource.cpp b/src/AsyncEventSource.cpp
index bbe28b3..5e65e26 100644
--- a/src/AsyncEventSource.cpp
+++ b/src/AsyncEventSource.cpp
@@ -113,7 +113,8 @@ AsyncEventSourceClient::AsyncEventSourceClient(AsyncWebServerRequest *request, A
   next = NULL;
   if(request->hasHeader("Last-Event-ID"))
     _lastId = atoi(request->getHeader("Last-Event-ID")->value().c_str());
-
+    
+  _client->setRxTimeout(0);
   _client->onError(NULL, NULL);
   _client->onAck(NULL, NULL);
   _client->onPoll(NULL, NULL);
@@ -175,7 +176,6 @@ void AsyncEventSource::onConnect(ArEventHandlerFunction cb){
 }
 
 void AsyncEventSource::_addClient(AsyncEventSourceClient * client){
-  client->setRxTimeout(0);
   /*char * temp = (char *)malloc(2054);
   if(temp != NULL){
     memset(temp+1,' ',2048);
diff --git a/src/AsyncWebSocket.cpp b/src/AsyncWebSocket.cpp
index 79691e1..8d38de0 100644
--- a/src/AsyncWebSocket.cpp
+++ b/src/AsyncWebSocket.cpp
@@ -250,6 +250,7 @@ AsyncWebSocketClient::AsyncWebSocketClient(AsyncWebServerRequest *request, Async
   _lastMessageTime = millis();
   _keepAlivePeriod = 0;
   next = NULL;
+  _client->setRxTimeout(0);
   _client->onError([](void *r, AsyncClient* c, int8_t error){ ((AsyncWebSocketClient*)(r))->_onError(error); }, this);
   _client->onAck([](void *r, AsyncClient* c, size_t len, uint32_t time){ ((AsyncWebSocketClient*)(r))->_onAck(len, time); }, this);
   _client->onDisconnect([](void *r, AsyncClient* c){ ((AsyncWebSocketClient*)(r))->_onDisconnect(); delete c; }, this);
@@ -621,7 +622,6 @@ void AsyncWebSocket::_handleEvent(AsyncWebSocketClient * client, AwsEventType ty
 }
 
 void AsyncWebSocket::_addClient(AsyncWebSocketClient * client){
-  client->setRxTimeout(0);
   if(_clients == NULL){
     _clients = client;
     return;