Fix handling lingering bytes after reconnect

Wiebe Cazemier
1 parent 2a3138f9
Showing 1 changed file with 4 additions and 5 deletions
main.cpp
@@ -45,15 +45,14 @@ void do_thread_work(ThreadData *threadData)
                 {
                     if (cur_ev.events & EPOLLIN)
                     {
-                        if (!client->readFdIntoBuffer())
+                        bool readSuccess = client->readFdIntoBuffer();
+                        client->bufferToMqttPackets(packetQueueIn, client);
+
+                        if (!readSuccess)
                         {
                             std::cout << "Disconnect: " << client->repr() << std::endl;
                             threadData->removeClient(client);
                         }
-                        else
-                        {
-                            client->bufferToMqttPackets(packetQueueIn, client);
-                        }
                     }
                     if (cur_ev.events & EPOLLOUT)
                     {