Keep connections open (#23)

src/worker/network.rs

@@ -10,6 +10,7 @@ use std::{
     fmt::Display,
     io::{prelude::*, BufReader},
     net::TcpListener,
+    thread,
 };
 
 use crate::{BaseConfig, Worker, WorkerError, Workload, WorkloadConfig};
@@ -51,7 +52,16 @@ impl NetworkWorker {
 
         for stream in listener.incoming() {
             let mut stream = stream.unwrap();
-            loop {
+
+            // As a simplest solution to keep a connection open, spawn a
+            // thread.  It's not the best one though, as we waste resources.
+            // For the purpose of only keeping connections open we could e.g.
+            // spawn only two threads, where the first one receives connections
+            // and adds streams into the list of active, and the second iterates
+            // through streams and replies. This way the connections will have
+            // high latency, but for the purpose of networking workload it
+            // doesn't matter.
+            thread::spawn(move || loop {
                 let mut buf_reader = BufReader::new(&stream);
                 let mut buffer = String::new();
 
@@ -67,7 +77,6 @@ impl NetworkWorker {
                         match stream.write_all(response.as_bytes()) {
                             Ok(_) => {
                                 // Response is sent, handle the next one
-                                break;
                             }
                             Err(e) => {
                                 trace!("ERROR: sending response, {}", e);
@@ -79,7 +88,7 @@ impl NetworkWorker {
                         trace!("ERROR: reading a line, {}", e)
                     }
                 }
-            }
+            });
         }
 
         Ok(())