1 files changed, 77 insertions, 0 deletions
diff --git a/cli/bench/websocket.rs b/cli/bench/websocket.rs
new file mode 100644
index 000000000..84a799660
--- /dev/null
+++ b/cli/bench/websocket.rs
@@ -0,0 +1,77 @@
+// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.
+use std::collections::HashMap;
+use std::io::BufRead;
+use std::path::Path;
+use std::process::Command;
+use std::process::Stdio;
+use std::time::Duration;
+
+use super::Result;
+
+pub fn benchmark() -> Result<HashMap<String, f64>> {
+  let deno_exe = test_util::deno_exe_path();
+  let deno_exe = deno_exe.to_str().unwrap();
+
+  let mut res = HashMap::new();
+  let manifest_dir = Path::new(env!("CARGO_MANIFEST_DIR"));
+  let ws_dir = manifest_dir.join("bench").join("websocket");
+  for entry in std::fs::read_dir(&ws_dir)? {
+    let port = crate::http::get_port();
+    let entry = entry?;
+    let pathbuf = entry.path();
+    let path = pathbuf.to_str().unwrap();
+    let file_stem = pathbuf.file_stem().unwrap().to_str().unwrap();
+
+    let mut cmd = Command::new(deno_exe);
+    let mut server = cmd
+      .arg("run")
+      .arg("-A")
+      .arg("--unstable")
+      .arg(path)
+      .arg(&port.to_string())
+      .spawn()
+      .unwrap();
+
+    std::thread::sleep(Duration::from_secs(5)); // wait for server to wake up.
+
+    let load_test = test_util::prebuilt_tool_path("load_test");
+    assert!(load_test.is_file());
+    // ./load_test 100 0.0.0.0 8000 0 0
+    // Running benchmark now...
+    // Msg/sec: 161327.500000
+    // Msg/sec: 163977.000000
+    // ^C⏎
+    let mut cmd = Command::new(load_test);
+    let mut process = cmd
+      .stdout(Stdio::piped())
+      .arg("100")
+      .arg("0.0.0.0")
+      .arg(&port.to_string())
+      .arg("0")
+      .arg("0")
+      .spawn()
+      .unwrap();
+
+    let mut lines = Vec::new();
+
+    let mut stdout =
+      std::io::BufReader::new(process.stdout.take().unwrap()).lines();
+    for _ in 0..5 {
+      let line = stdout.next().unwrap().unwrap();
+      lines.push(line);
+    }
+
+    process.kill().unwrap();
+    let msg_per_sec = lines
+      .into_iter()
+      .filter(|line| line.starts_with("Msg/sec:"))
+      .map(|line| line.split(": ").nth(1).unwrap().parse::<f64>().unwrap())
+      .max_by(|a, b| a.partial_cmp(b).unwrap())
+      .unwrap();
+
+    res.insert(file_stem.to_string(), msg_per_sec);
+    server.kill().unwrap();
+  }
+
+  Ok(res)
+}