diff --git a/server/Cargo.toml b/server/Cargo.toml index fa5235ef..d52d12a3 100644 --- a/server/Cargo.toml +++ b/server/Cargo.toml @@ -3,14 +3,6 @@ name = "mnml" version = "1.4.1" authors = ["ntr "] -[[bin]] -name = "main" -path = "bin/main.rs" - -[[bin]] -name = "svt" -path = "bin/svt.rs" - [dependencies] serde = "1" serde_derive = "1" diff --git a/server/bin/svt.rs b/server/src/bin/svt.rs similarity index 94% rename from server/bin/svt.rs rename to server/src/bin/svt.rs index 9fae1289..6ae957af 100644 --- a/server/bin/svt.rs +++ b/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 = 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(); diff --git a/server/bin/main.rs b/server/src/main.rs similarity index 100% rename from server/bin/main.rs rename to server/src/main.rs