Skip to content

Commit ae94ffe

Browse files
authored
feat(bench-bot): add cooling down(cd) arg for each benchmark (#17)
1 parent b6682b7 commit ae94ffe

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

bench-bot/src/main.rs

+12-1
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,10 @@ struct Args {
4141
/// Url for each benchmark.
4242
#[clap(short, default_value = "http://127.0.0.1:3000")]
4343
url: String,
44+
45+
/// Cooling down for each benchmark.
46+
#[clap(long, default_value = "5")]
47+
cd: u64,
4448
}
4549

4650
#[derive(Debug, Serialize, Deserialize)]
@@ -100,6 +104,8 @@ fn main() {
100104
let cpu_count = sys.cpus().len().to_string();
101105
let conn_count = args.connections.to_string();
102106
let duration = format!("{}s", args.duration);
107+
let cd = args.cd;
108+
let members_len = members.len();
103109

104110
let rewrk_args = [
105111
"-t",
@@ -131,7 +137,7 @@ fn main() {
131137
let mut output_map = HashMap::new();
132138
let mut reports = Vec::with_capacity(members.len());
133139

134-
for member in &members {
140+
for (index, member) in members.iter().enumerate() {
135141
if exclude.contains(member) {
136142
log::warn!("Skipping {:?} because build was failed.", member);
137143
} else {
@@ -204,6 +210,11 @@ fn main() {
204210
log::warn!("Could not parse benchmark result: {}", stdout);
205211
}
206212
}
213+
214+
// lets CPU cooling down, ignore last member.
215+
if index != members_len - 1 {
216+
thread::sleep(Duration::from_secs(cd));
217+
}
207218
}
208219
}
209220

0 commit comments

Comments
 (0)