svt

server/Cargo.toml

@@ -3,14 +3,6 @@ name = "mnml"
 version = "1.4.1"
 authors = ["ntr <ntr@smokestack.io>"]
 
-[[bin]]
-name = "main"
-path = "bin/main.rs"
-
-[[bin]]
-name = "svt"
-path = "bin/svt.rs"
-
 [dependencies]
 serde = "1"
 serde_derive = "1"

server/src/bin/svt.rs

@@ -4,6 +4,8 @@ extern crate ws;
 extern crate mnml;
 extern crate serde_cbor;
 
+use std::env;
+
 use std::thread;
 
 use serde_cbor::{from_slice, to_vec};
@@ -45,11 +47,14 @@ impl Handler for Bot {
 }
 
 fn main() {
+    let args: Vec<String> = env::args().collect();
+    let num_clients: usize = args[1].parse().unwrap();
+
     setup_logger().unwrap();
 
     let mut clients = vec![];
 
-    for i in 0..1000 {
+    for i in 0..num_clients {
         clients.push(thread::Builder::new().name(i.to_string()).spawn(|| {
             let mut rng = thread_rng();
             let name: String = iter::repeat(()).map(|()| rng.sample(Alphanumeric)).take(12).collect();