Better drain

resetius · Dec 1, 2023 · 9245b3e · 9245b3e
1 parent aa48e7e
commit 9245b3e
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/server.cpp b/src/server.cpp
@@ -90,16 +90,17 @@ void TNode<TPoller>::Drain() {
 
 template<typename TPoller>
 NNet::TTestTask TNode<TPoller>::DoDrain() {
-    auto tosend = std::move(Messages);
     try {
-        for (auto&& m : tosend) {
-            co_await TWriter(Socket).Write(std::move(m));
+        while (!Messages.empty()) {
+            auto tosend = std::move(Messages); Messages.clear();
+            for (auto&& m : tosend) {
+                co_await TWriter(Socket).Write(std::move(m));
+            }
         }
     } catch (const std::exception& ex) {
         std::cout << "Error on write: " << ex.what() << "\n";
         Connect();
     }
-    Messages.clear();
     co_return;
 }