cleanup code layout

Re-organize the various client types to provide more coherent structure.

src/clients/memcache/mod.rs → src/clients/cache/memcache/mod.rs

@@ -1,3 +1,4 @@
+use crate::clients::ResponseError;
 use super::*;
 use crate::net::Connector;
 use protocol_memcache::{Compose, Parse, Request, Response, Ttl};

src/clients/cache/mod.rs

@@ -0,0 +1,52 @@
+use crate::workload::ClientRequest;
+use crate::*;
+
+use async_channel::Receiver;
+use tokio::io::*;
+use tokio::runtime::Runtime;
+use tokio::time::{timeout, Duration};
+use workload::ClientWorkItemKind;
+
+use std::io::{Error, ErrorKind, Result};
+use std::time::Instant;
+
+mod memcache;
+mod momento;
+mod redis;
+
+pub fn launch(
+    config: &Config,
+    work_receiver: Receiver<ClientWorkItemKind<ClientRequest>>,
+) -> Option<Runtime> {
+    debug!("Launching clients...");
+
+    config.client()?;
+
+    // spawn the request drivers on their own runtime
+    let mut client_rt = Builder::new_multi_thread()
+        .enable_all()
+        .worker_threads(config.client().unwrap().threads())
+        .build()
+        .expect("failed to initialize tokio runtime");
+
+    match config.general().protocol() {
+        Protocol::Memcache => {
+            memcache::launch_tasks(&mut client_rt, config.clone(), work_receiver)
+        }
+        Protocol::Momento => {
+            momento::launch_tasks(&mut client_rt, config.clone(), work_receiver)
+        }
+        Protocol::Ping => {
+            crate::clients::ping::ascii::launch_tasks(&mut client_rt, config.clone(), work_receiver)
+        }
+        Protocol::Resp => {
+            redis::launch_tasks(&mut client_rt, config.clone(), work_receiver)
+        }
+        protocol => {
+            error!("keyspace is not supported for the {:?} protocol", protocol);
+            std::process::exit(1);
+        }
+    }
+
+    Some(client_rt)
+}

src/clients/momento/mod.rs → src/clients/cache/momento/mod.rs

@@ -1,4 +1,5 @@
 use super::*;
+use crate::clients::*;
 
 use ::momento::cache::configurations::LowLatency;
 use ::momento::*;

src/clients/redis/mod.rs → src/clients/cache/redis/mod.rs

@@ -1,4 +1,5 @@
-use super::*;
+use crate::clients::cache::*;
+use crate::clients::*;
 use crate::net::Connector;
 use ::redis::aio::Connection;
 use ::redis::{AsyncCommands, RedisConnectionInfo};