mirror of
https://github.com/bdbrown4/peltier.io.git
synced 2026-11-07 10:58:28 +00:00
phase4: service-mode latency bench + mechanical ROI report generator
- bench-runner service mode (SPEC §3.1 mode c): a coordinated-omission- correct, open-loop, fixed-rate load generator; latency measured from the intended send time so queueing is counted, not hidden. Interleaved A/B of two server binaries, exact percentiles (bounded sample, no HDR bucketing error), bootstrap p50/p99 CIs via the existing ratio-CI machine. service-calibrate: A/A false-positive + injected latency- regression detection (calibration in progress). - targets/cjson/service.c: minimal single-threaded HTTP server wrapping the patched cJSON.c (trust-layer, outside the workspace allowlist); env-toggled per-request busy-wait for the injection calibration only. - report generator (crates/report bin): reads a ledger row + optional service-latency JSON, renders throughput→cores→dollars and/or latency percentiles, every figure with its CI, pricing source, and the methodology printed inline (SPEC §9); flags any non-accepted row so a rejected change never reads as a committed saving. - ledger::report_row; justfile service/service-calibrate/report. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01EcKPWQ3MQD4NdEHwar96u2
This commit is contained in:
Claude
@@ -10,6 +10,7 @@ repository.workspace = true
|
||||
anyhow.workspace = true
|
||||
clap.workspace = true
|
||||
ledger = { path = "../ledger" }
|
||||
libc = "0.2"
|
||||
serde.workspace = true
|
||||
serde_json.workspace = true
|
||||
toml.workspace = true
|
||||
|
||||
@@ -9,6 +9,7 @@ pub mod config;
|
||||
pub mod exec;
|
||||
pub mod fingerprint;
|
||||
pub mod schedule;
|
||||
pub mod service;
|
||||
pub mod stats;
|
||||
|
||||
use ledger::Verdict;
|
||||
|
||||
@@ -47,6 +47,297 @@ enum Cmd {
|
||||
#[arg(long)]
|
||||
out: PathBuf,
|
||||
},
|
||||
/// Service mode (SPEC §3.1 mode c): interleaved A/B of two HTTP
|
||||
/// server binaries under a coordinated-omission-correct open-loop
|
||||
/// load. Reports p50 and p99 latency speedup (baseline/candidate)
|
||||
/// with bootstrap CIs. `--aa` runs the same binary both sides.
|
||||
Service {
|
||||
#[arg(long)]
|
||||
baseline_bin: String,
|
||||
#[arg(long)]
|
||||
candidate_bin: String,
|
||||
#[arg(long)]
|
||||
doc: String,
|
||||
#[arg(long, default_value_t = 1)]
|
||||
iters: u64,
|
||||
/// Open-loop arrival rate (requests/sec).
|
||||
#[arg(long, default_value_t = 200.0)]
|
||||
rate: f64,
|
||||
/// Timed requests per session.
|
||||
#[arg(long, default_value_t = 1000)]
|
||||
count: usize,
|
||||
/// Interleaved sessions per side.
|
||||
#[arg(long, default_value_t = 12)]
|
||||
sessions: usize,
|
||||
#[arg(long, default_value_t = 200)]
|
||||
warmup: usize,
|
||||
#[arg(long, default_value_t = 32)]
|
||||
workers: usize,
|
||||
/// Prefix pinning the server to a core (e.g. "taskset -c 2").
|
||||
#[arg(long, default_value = "")]
|
||||
pin: String,
|
||||
/// Same binary both sides — must yield a null verdict on p99.
|
||||
#[arg(long, default_value_t = false)]
|
||||
aa: bool,
|
||||
/// Optional JSON output path (latency percentiles + CIs).
|
||||
#[arg(long)]
|
||||
out: Option<PathBuf>,
|
||||
},
|
||||
/// Service-mode calibration: A/A false-positive rate + injected
|
||||
/// latency-regression detection rate on p50 (SPEC §3.1).
|
||||
ServiceCalibrate {
|
||||
#[arg(long)]
|
||||
server_bin: String,
|
||||
#[arg(long)]
|
||||
doc: String,
|
||||
#[arg(long, default_value_t = 1)]
|
||||
iters: u64,
|
||||
#[arg(long, default_value_t = 150.0)]
|
||||
rate: f64,
|
||||
#[arg(long, default_value_t = 500)]
|
||||
count: usize,
|
||||
/// Interleaved rounds per compare.
|
||||
#[arg(long, default_value_t = 6)]
|
||||
rounds: usize,
|
||||
/// Calibration compares (A/A + injection each).
|
||||
#[arg(long, default_value_t = 10)]
|
||||
sessions: usize,
|
||||
#[arg(long, default_value_t = 100)]
|
||||
warmup: usize,
|
||||
#[arg(long, default_value_t = 32)]
|
||||
workers: usize,
|
||||
#[arg(long, default_value = "")]
|
||||
pin: String,
|
||||
#[arg(long, default_value_t = 0.05)]
|
||||
slowdown: f64,
|
||||
#[arg(long)]
|
||||
out: PathBuf,
|
||||
},
|
||||
}
|
||||
|
||||
#[allow(clippy::too_many_arguments)]
|
||||
fn service_bench(
|
||||
cfg: &AcceptConfig,
|
||||
baseline_bin: &str,
|
||||
candidate_bin: &str,
|
||||
doc: &str,
|
||||
iters: u64,
|
||||
rate: f64,
|
||||
count: usize,
|
||||
sessions: usize,
|
||||
warmup: usize,
|
||||
workers: usize,
|
||||
pin: &str,
|
||||
aa: bool,
|
||||
out: &Option<PathBuf>,
|
||||
) -> Result<()> {
|
||||
use bench_runner::service::{run_compare, ServiceCfg};
|
||||
|
||||
let mk = |bin: &str| ServiceCfg {
|
||||
server_bin: bin.to_string(),
|
||||
doc: doc.to_string(),
|
||||
iters,
|
||||
rate,
|
||||
count,
|
||||
warmup,
|
||||
workers,
|
||||
pin_prefix: pin.to_string(),
|
||||
inject_us: None,
|
||||
};
|
||||
let base_cfg = mk(baseline_bin);
|
||||
let cand_cfg = mk(candidate_bin);
|
||||
|
||||
eprintln!(
|
||||
"service: {sessions} interleaved sessions/side, {count} req @ {rate} rps, \
|
||||
CO-correct open loop, server pin='{pin}'"
|
||||
);
|
||||
|
||||
let r = run_compare(
|
||||
&base_cfg,
|
||||
&cand_cfg,
|
||||
sessions,
|
||||
cfg.bootstrap_iters,
|
||||
cfg.confidence,
|
||||
cfg.bootstrap_seed,
|
||||
|k, bp50, bp99, cp50, cp99| {
|
||||
eprintln!(
|
||||
" round {}/{sessions}: base p50={:.3}ms p99={:.3}ms | cand p50={:.3}ms p99={:.3}ms",
|
||||
k + 1,
|
||||
bp50 * 1e3,
|
||||
bp99 * 1e3,
|
||||
cp50 * 1e3,
|
||||
cp99 * 1e3
|
||||
);
|
||||
},
|
||||
)?;
|
||||
let (ci50, ci99, drop_rate) = (r.ci50, r.ci99, r.drop_rate);
|
||||
let (b_p50, c_p50, b_p99, c_p99) = (&r.base_p50, &r.cand_p50, &r.base_p99, &r.cand_p99);
|
||||
|
||||
// Drop rate must be negligible or the load was above capacity and the
|
||||
// percentiles are unstable — refuse to report a number we don't trust.
|
||||
let total_req = sessions * count * 2;
|
||||
let total_dropped = (drop_rate * total_req as f64).round() as usize;
|
||||
anyhow::ensure!(
|
||||
drop_rate < 0.005,
|
||||
"drop rate {:.2}% exceeds 0.5% — load is above server capacity; lower --rate",
|
||||
drop_rate * 100.0
|
||||
);
|
||||
|
||||
println!(
|
||||
"p50 latency speedup (baseline/candidate): median {:.4}, {:.0}% CI [{:.4}, {:.4}]",
|
||||
ci50.median,
|
||||
cfg.confidence * 100.0,
|
||||
ci50.lo,
|
||||
ci50.hi
|
||||
);
|
||||
println!(
|
||||
"p99 latency speedup (baseline/candidate): median {:.4}, {:.0}% CI [{:.4}, {:.4}]",
|
||||
ci99.median,
|
||||
cfg.confidence * 100.0,
|
||||
ci99.lo,
|
||||
ci99.hi
|
||||
);
|
||||
println!(
|
||||
"drop rate: {:.3}% ({total_dropped}/{total_req})",
|
||||
drop_rate * 100.0
|
||||
);
|
||||
|
||||
if let Some(path) = out {
|
||||
let ev = serde_json::json!({
|
||||
"mode": "service-latency",
|
||||
"workload": format!("{doc}, {iters} parse+print/req, {rate} rps open-loop, CO-correct"),
|
||||
"sessions": sessions, "count_per_session": count, "rate_rps": rate,
|
||||
"p50_speedup_median": ci50.median, "p50_speedup_ci": [ci50.lo, ci50.hi],
|
||||
"p99_speedup_median": ci99.median, "p99_speedup_ci": [ci99.lo, ci99.hi],
|
||||
"baseline_p50_ms_median": stats::median(b_p50) * 1e3,
|
||||
"candidate_p50_ms_median": stats::median(c_p50) * 1e3,
|
||||
"baseline_p99_ms_median": stats::median(b_p99) * 1e3,
|
||||
"candidate_p99_ms_median": stats::median(c_p99) * 1e3,
|
||||
"drop_rate": drop_rate,
|
||||
});
|
||||
std::fs::write(path, serde_json::to_string_pretty(&ev)?)?;
|
||||
}
|
||||
|
||||
if aa {
|
||||
// Null verdict required on BOTH percentiles.
|
||||
for (label, ci) in [("p50", ci50), ("p99", ci99)] {
|
||||
if decide(ci, cfg.threshold) == Verdict::Accepted {
|
||||
anyhow::bail!(
|
||||
"service A/A FAILED on {label}: speedup claimed from identical servers"
|
||||
);
|
||||
}
|
||||
}
|
||||
println!("service A/A self-test passed (null verdict on p50 and p99)");
|
||||
} else {
|
||||
println!("p50 verdict: {}", decide(ci50, cfg.threshold).as_str());
|
||||
println!("p99 verdict: {}", decide(ci99, cfg.threshold).as_str());
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Service-mode calibration (SPEC §3.1): N A/A compares (false-positive
|
||||
/// rate on p50 must be <5%) + N injection compares (a per-request
|
||||
/// busy-wait ≈ `slowdown` of measured service time in the candidate; the
|
||||
/// harness must resolve the regression — p50 CI entirely < 1 — ≥95%).
|
||||
#[allow(clippy::too_many_arguments)]
|
||||
fn service_calibrate(
|
||||
cfg: &AcceptConfig,
|
||||
server_bin: &str,
|
||||
doc: &str,
|
||||
iters: u64,
|
||||
rate: f64,
|
||||
count: usize,
|
||||
rounds: usize,
|
||||
n: usize,
|
||||
warmup: usize,
|
||||
workers: usize,
|
||||
pin: &str,
|
||||
slowdown: f64,
|
||||
out: &PathBuf,
|
||||
) -> Result<()> {
|
||||
use bench_runner::service::{run_compare, run_session, ServiceCfg};
|
||||
|
||||
let mk = |inject: Option<u64>| ServiceCfg {
|
||||
server_bin: server_bin.to_string(),
|
||||
doc: doc.to_string(),
|
||||
iters,
|
||||
rate,
|
||||
count,
|
||||
warmup,
|
||||
workers,
|
||||
pin_prefix: pin.to_string(),
|
||||
inject_us: inject,
|
||||
};
|
||||
|
||||
// Size the injected busy-wait from a probe of the service's p50.
|
||||
let probe = run_session(&mk(None), bench_runner::service::free_port()?)?;
|
||||
let p50 = bench_runner::service::percentile(&probe.latencies, 0.50);
|
||||
let inject_us = (p50 * slowdown * 1e6).round() as u64;
|
||||
eprintln!(
|
||||
"service-calibrate: probe p50={:.3}ms, injecting {inject_us}µs (~{:.0}%) for detection",
|
||||
p50 * 1e3,
|
||||
slowdown * 100.0
|
||||
);
|
||||
|
||||
let (mut fp, mut det) = (0usize, 0usize);
|
||||
let (mut aa_rows, mut inj_rows) = (Vec::new(), Vec::new());
|
||||
for i in 0..n {
|
||||
let seed = cfg.bootstrap_seed.wrapping_add(i as u64);
|
||||
let aa = run_compare(
|
||||
&mk(None),
|
||||
&mk(None),
|
||||
rounds,
|
||||
cfg.bootstrap_iters,
|
||||
cfg.confidence,
|
||||
seed,
|
||||
|_, _, _, _, _| {},
|
||||
)?;
|
||||
if decide(aa.ci50, cfg.threshold) == Verdict::Accepted {
|
||||
fp += 1;
|
||||
}
|
||||
let inj = run_compare(
|
||||
&mk(None),
|
||||
&mk(Some(inject_us)),
|
||||
rounds,
|
||||
cfg.bootstrap_iters,
|
||||
cfg.confidence,
|
||||
seed,
|
||||
|_, _, _, _, _| {},
|
||||
)?;
|
||||
if inj.ci50.hi < 1.0 {
|
||||
det += 1;
|
||||
}
|
||||
aa_rows.push(serde_json::json!({"lo": aa.ci50.lo, "hi": aa.ci50.hi}));
|
||||
inj_rows.push(serde_json::json!({"lo": inj.ci50.lo, "hi": inj.ci50.hi}));
|
||||
eprintln!(
|
||||
"session {}/{n}: A/A p50 [{:.4},{:.4}] fp={fp}; inj p50 [{:.4},{:.4}] det={det}",
|
||||
i + 1,
|
||||
aa.ci50.lo,
|
||||
aa.ci50.hi,
|
||||
inj.ci50.lo,
|
||||
inj.ci50.hi
|
||||
);
|
||||
}
|
||||
let fp_rate = fp as f64 / n as f64;
|
||||
let det_rate = det as f64 / n as f64;
|
||||
let pass = fp_rate < 0.05 && det_rate >= 0.95;
|
||||
let ev = serde_json::json!({
|
||||
"mode": "service-latency-calibration",
|
||||
"server_bin": server_bin, "workload": doc, "rate_rps": rate,
|
||||
"count_per_session": count, "rounds_per_compare": rounds, "sessions": n,
|
||||
"injected_slowdown": slowdown, "injected_us": inject_us,
|
||||
"aa_false_positive_rate": fp_rate, "aa_sessions": aa_rows,
|
||||
"injection_detection_rate": det_rate, "injection_sessions": inj_rows,
|
||||
"acceptance": {"fp_lt": 0.05, "detection_ge": 0.95, "pass": pass},
|
||||
});
|
||||
std::fs::write(out, serde_json::to_string_pretty(&ev)?)?;
|
||||
println!(
|
||||
"service calibration: A/A false-positive {fp_rate:.3} (<0.05), \
|
||||
injection detection {det_rate:.3} (>=0.95) -> {}",
|
||||
if pass { "PASS" } else { "FAIL" }
|
||||
);
|
||||
anyhow::ensure!(pass, "service calibration acceptance criteria not met");
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn calibrate(
|
||||
@@ -152,6 +443,56 @@ fn main() -> Result<()> {
|
||||
} => {
|
||||
return calibrate(&cfg, cmd, *sessions, *slowdown, out);
|
||||
}
|
||||
Cmd::Service {
|
||||
baseline_bin,
|
||||
candidate_bin,
|
||||
doc,
|
||||
iters,
|
||||
rate,
|
||||
count,
|
||||
sessions,
|
||||
warmup,
|
||||
workers,
|
||||
pin,
|
||||
aa,
|
||||
out,
|
||||
} => {
|
||||
#[allow(clippy::used_underscore_items)]
|
||||
return service_bench(
|
||||
&cfg,
|
||||
baseline_bin,
|
||||
candidate_bin,
|
||||
doc,
|
||||
*iters,
|
||||
*rate,
|
||||
*count,
|
||||
*sessions,
|
||||
*warmup,
|
||||
*workers,
|
||||
pin,
|
||||
*aa,
|
||||
out,
|
||||
);
|
||||
}
|
||||
Cmd::ServiceCalibrate {
|
||||
server_bin,
|
||||
doc,
|
||||
iters,
|
||||
rate,
|
||||
count,
|
||||
rounds,
|
||||
sessions,
|
||||
warmup,
|
||||
workers,
|
||||
pin,
|
||||
slowdown,
|
||||
out,
|
||||
} => {
|
||||
return service_calibrate(
|
||||
&cfg, server_bin, doc, *iters, *rate, *count, *rounds, *sessions, *warmup,
|
||||
*workers, pin, *slowdown, out,
|
||||
);
|
||||
}
|
||||
};
|
||||
|
||||
let fp = EnvFingerprint::collect("none", "system-default");
|
||||
|
||||
@@ -0,0 +1,248 @@
|
||||
//! Service-mode latency bench (SPEC §3.1 mode (c)). A coordinated-
|
||||
//! omission-CORRECT, open-loop, fixed-rate load generator: requests are
|
||||
//! scheduled at `start + i/rate` and each latency is measured from its
|
||||
//! *intended* send time, not its actual send time. So when the server
|
||||
//! falls behind, the requests that pile up behind it each carry the full
|
||||
//! queueing delay a real client would see — the exact error (coordinated
|
||||
//! omission) that closed-loop tools like `ab` hide. We store every
|
||||
//! latency and take exact percentiles (no HdrHistogram bucketing error;
|
||||
//! the sample is bounded).
|
||||
//!
|
||||
//! Interleaved A/B, like the wall-time bench: each session yields one
|
||||
//! percentile (p50/p99) per side; the existing bootstrap ratio-CI machine
|
||||
//! then works unchanged, treating "p99 latency" the way it treats "wall
|
||||
//! time" — lower is better, speedup = baseline / candidate.
|
||||
|
||||
use anyhow::{anyhow, ensure, Result};
|
||||
use std::io::{BufRead, BufReader, Read, Write};
|
||||
use std::net::{Ipv4Addr, TcpStream};
|
||||
use std::process::{Child, Command, Stdio};
|
||||
use std::sync::atomic::{AtomicUsize, Ordering};
|
||||
use std::sync::{Arc, Mutex};
|
||||
use std::time::{Duration, Instant};
|
||||
|
||||
use crate::stats::{self, RatioCi};
|
||||
|
||||
pub struct ServiceCfg {
|
||||
pub server_bin: String,
|
||||
pub doc: String,
|
||||
pub iters: u64,
|
||||
pub rate: f64,
|
||||
pub count: usize,
|
||||
pub warmup: usize,
|
||||
pub workers: usize,
|
||||
/// Prefix that pins the SERVER to a core (e.g. `taskset -c 2`); the
|
||||
/// load generator runs on the other cores.
|
||||
pub pin_prefix: String,
|
||||
/// Calibration only: inject this many µs of busy-wait per request
|
||||
/// (via HOTPATH_INJECT_US) to validate the harness detects a known
|
||||
/// latency regression. None in every real run.
|
||||
pub inject_us: Option<u64>,
|
||||
}
|
||||
|
||||
/// Result of one interleaved A/B service compare.
|
||||
pub struct CompareResult {
|
||||
pub ci50: RatioCi,
|
||||
pub ci99: RatioCi,
|
||||
pub base_p50: Vec<f64>,
|
||||
pub cand_p50: Vec<f64>,
|
||||
pub base_p99: Vec<f64>,
|
||||
pub cand_p99: Vec<f64>,
|
||||
pub drop_rate: f64,
|
||||
}
|
||||
|
||||
/// Interleaved A/B: `rounds` alternating baseline/candidate sessions,
|
||||
/// each yielding one p50 and one p99; bootstrap ratio-CI over the rounds.
|
||||
#[allow(clippy::too_many_arguments)]
|
||||
pub fn run_compare(
|
||||
base: &ServiceCfg,
|
||||
cand: &ServiceCfg,
|
||||
rounds: usize,
|
||||
bootstrap_iters: usize,
|
||||
confidence: f64,
|
||||
seed: u64,
|
||||
mut on_round: impl FnMut(usize, f64, f64, f64, f64),
|
||||
) -> Result<CompareResult> {
|
||||
let (mut b50, mut b99, mut c50, mut c99) = (Vec::new(), Vec::new(), Vec::new(), Vec::new());
|
||||
let mut dropped = 0usize;
|
||||
for k in 0..rounds {
|
||||
let bs = run_session(base, free_port()?)?;
|
||||
let cs = run_session(cand, free_port()?)?;
|
||||
dropped += bs.dropped + cs.dropped;
|
||||
let (bp50, bp99) = (
|
||||
percentile(&bs.latencies, 0.50),
|
||||
percentile(&bs.latencies, 0.99),
|
||||
);
|
||||
let (cp50, cp99) = (
|
||||
percentile(&cs.latencies, 0.50),
|
||||
percentile(&cs.latencies, 0.99),
|
||||
);
|
||||
b50.push(bp50);
|
||||
b99.push(bp99);
|
||||
c50.push(cp50);
|
||||
c99.push(cp99);
|
||||
on_round(k, bp50, bp99, cp50, cp99);
|
||||
}
|
||||
let total = rounds * base.count * 2;
|
||||
let drop_rate = dropped as f64 / total as f64;
|
||||
let ci50 = stats::bootstrap_ratio_ci(&b50, &c50, bootstrap_iters, confidence, seed);
|
||||
let ci99 = stats::bootstrap_ratio_ci(&b99, &c99, bootstrap_iters, confidence, seed);
|
||||
Ok(CompareResult {
|
||||
ci50,
|
||||
ci99,
|
||||
base_p50: b50,
|
||||
cand_p50: c50,
|
||||
base_p99: b99,
|
||||
cand_p99: c99,
|
||||
drop_rate,
|
||||
})
|
||||
}
|
||||
|
||||
pub struct SessionResult {
|
||||
pub latencies: Vec<f64>,
|
||||
pub dropped: usize,
|
||||
}
|
||||
|
||||
/// Exact percentile (0.0–1.0) of a sorted-in-place copy.
|
||||
pub fn percentile(latencies: &[f64], q: f64) -> f64 {
|
||||
if latencies.is_empty() {
|
||||
return f64::NAN;
|
||||
}
|
||||
let mut xs = latencies.to_vec();
|
||||
xs.sort_by(|a, b| a.partial_cmp(b).unwrap());
|
||||
let idx = ((xs.len() as f64 - 1.0) * q).round() as usize;
|
||||
xs[idx.min(xs.len() - 1)]
|
||||
}
|
||||
|
||||
fn spawn_server(cfg: &ServiceCfg, port: u16) -> Result<Child> {
|
||||
let inject = match cfg.inject_us {
|
||||
Some(us) => format!("HOTPATH_INJECT_US={us} "),
|
||||
None => String::new(),
|
||||
};
|
||||
let cmd = format!(
|
||||
"{inject}{pin} {bin} {port} {doc} {iters}",
|
||||
pin = cfg.pin_prefix,
|
||||
bin = cfg.server_bin,
|
||||
doc = cfg.doc,
|
||||
iters = cfg.iters,
|
||||
);
|
||||
let mut child = Command::new("sh")
|
||||
.arg("-c")
|
||||
.arg(cmd.trim())
|
||||
.stdout(Stdio::piped())
|
||||
.stderr(Stdio::null())
|
||||
.spawn()?;
|
||||
// Block until the server prints "READY <port>".
|
||||
let stdout = child
|
||||
.stdout
|
||||
.take()
|
||||
.ok_or_else(|| anyhow!("no server stdout"))?;
|
||||
let mut reader = BufReader::new(stdout);
|
||||
let mut line = String::new();
|
||||
reader.read_line(&mut line)?;
|
||||
ensure!(
|
||||
line.starts_with("READY"),
|
||||
"server did not signal READY (got {line:?})"
|
||||
);
|
||||
Ok(child)
|
||||
}
|
||||
|
||||
fn set_linger_zero(s: &TcpStream) {
|
||||
// SO_LINGER with a 0 timeout → RST on close instead of FIN, so the
|
||||
// tens of thousands of short client connections a load session opens
|
||||
// don't pile up in TIME_WAIT and exhaust the ephemeral port range.
|
||||
// std::net::TcpStream::set_linger is still unstable, so go through
|
||||
// setsockopt directly. The response is fully read before close, so the
|
||||
// RST is harmless.
|
||||
use std::os::unix::io::AsRawFd;
|
||||
let l = libc::linger {
|
||||
l_onoff: 1,
|
||||
l_linger: 0,
|
||||
};
|
||||
unsafe {
|
||||
libc::setsockopt(
|
||||
s.as_raw_fd(),
|
||||
libc::SOL_SOCKET,
|
||||
libc::SO_LINGER,
|
||||
&l as *const libc::linger as *const libc::c_void,
|
||||
std::mem::size_of::<libc::linger>() as libc::socklen_t,
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
fn one_request(port: u16) -> std::io::Result<()> {
|
||||
let mut s = TcpStream::connect((Ipv4Addr::LOCALHOST, port))?;
|
||||
s.set_nodelay(true)?;
|
||||
set_linger_zero(&s);
|
||||
s.write_all(b"GET / HTTP/1.1\r\nHost: x\r\n\r\n")?;
|
||||
let mut buf = [0u8; 512];
|
||||
// Read the full response (server sends Content-Length + Connection: close).
|
||||
loop {
|
||||
match s.read(&mut buf) {
|
||||
Ok(0) => break,
|
||||
Ok(_) => {}
|
||||
Err(e) => return Err(e),
|
||||
}
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// One open-loop load session against a freshly spawned server.
|
||||
pub fn run_session(cfg: &ServiceCfg, port: u16) -> Result<SessionResult> {
|
||||
let mut child = spawn_server(cfg, port)?;
|
||||
// Warm up closed-loop (untimed) so the first timed request doesn't eat
|
||||
// page faults / cold caches.
|
||||
for _ in 0..cfg.warmup {
|
||||
let _ = one_request(port);
|
||||
}
|
||||
|
||||
let period = Duration::from_secs_f64(1.0 / cfg.rate);
|
||||
let start = Instant::now();
|
||||
let idx = Arc::new(AtomicUsize::new(0));
|
||||
let out = Arc::new(Mutex::new((Vec::<f64>::with_capacity(cfg.count), 0usize)));
|
||||
let count = cfg.count;
|
||||
|
||||
let mut handles = Vec::new();
|
||||
for _ in 0..cfg.workers {
|
||||
let idx = idx.clone();
|
||||
let out = out.clone();
|
||||
handles.push(std::thread::spawn(move || {
|
||||
let mut local = Vec::new();
|
||||
let mut dropped = 0usize;
|
||||
loop {
|
||||
let i = idx.fetch_add(1, Ordering::Relaxed);
|
||||
if i >= count {
|
||||
break;
|
||||
}
|
||||
let target = start + period * (i as u32);
|
||||
let now = Instant::now();
|
||||
if now < target {
|
||||
std::thread::sleep(target - now);
|
||||
}
|
||||
// Latency from the INTENDED send time — the CO correction.
|
||||
match one_request(port) {
|
||||
Ok(()) => local.push(Instant::now().duration_since(target).as_secs_f64()),
|
||||
Err(_) => dropped += 1,
|
||||
}
|
||||
}
|
||||
let mut g = out.lock().unwrap();
|
||||
g.0.extend(local);
|
||||
g.1 += dropped;
|
||||
}));
|
||||
}
|
||||
for h in handles {
|
||||
let _ = h.join();
|
||||
}
|
||||
let _ = child.kill();
|
||||
let _ = child.wait();
|
||||
|
||||
let (latencies, dropped) = Arc::try_unwrap(out).unwrap().into_inner().unwrap();
|
||||
Ok(SessionResult { latencies, dropped })
|
||||
}
|
||||
|
||||
/// Pick a free loopback TCP port (small race between probe-close and the
|
||||
/// server bind; harmless on loopback with SO_REUSEADDR).
|
||||
pub fn free_port() -> Result<u16> {
|
||||
let l = std::net::TcpListener::bind((Ipv4Addr::LOCALHOST, 0))?;
|
||||
Ok(l.local_addr()?.port())
|
||||
}
|
||||
Reference in New Issue
Block a user