aljaz/turso
1
0
Fork 0
mirror of https://github.com/aljazceru/turso.git synced 2026-01-08 10:44:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

small improvement of stress testing tool

Browse Source
This commit is contained in:
Nikita Sivukhin
2025-09-29 20:18:10 +04:00
parent fda1b89540
commit 5086480b28
1 changed files with 20 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
stress/main.rs
View File

@@ -353,8 +353,15 @@ fn generate_plan(opts: &Opts) -> Result<Plan, Box<dyn std::error::Error + Send +
}
}
plan.ddl_statements = ddl_statements;
for _ in 0..opts.nr_threads {
for id in 0..opts.nr_threads {
writeln!(log_file, "{id}",)?;
let mut queries = vec![];
let mut push = |sql: &str| {
queries.push(sql.to_string());
if !opts.skip_log {
writeln!(log_file, "{sql}").unwrap();
}
};
for i in 0..opts.nr_iterations {
if !opts.silent && !opts.verbose && i % 100 == 0 {
print!(
@@ -369,18 +376,15 @@ fn generate_plan(opts: &Opts) -> Result<Plan, Box<dyn std::error::Error + Send +
None
};
if let Some(tx) = tx {
queries.push(tx.to_string());
push(tx);
}
let sql = generate_random_statement(&schema);
if !opts.skip_log {
writeln!(log_file, "{sql}")?;
}
queries.push(sql);
push(&sql);
if tx.is_some() {
if get_random() % 2 == 0 {
queries.push("COMMIT".to_string());
push("COMMIT");
} else {
queries.push("ROLLBACK".to_string());
push("ROLLBACK");
}
}
}
@@ -415,17 +419,15 @@ fn read_plan_from_log_file(opts: &Opts) -> Result<Plan, Box<dyn std::error::Erro
plan.ddl_statements
.push(lines.next().expect("expected ddl statement").to_string());
}
for _ in 0..plan.nr_threads {
let mut queries = vec![];
for _ in 0..plan.nr_iterations {
queries.push(
lines
.next()
.expect("missing query for thread {}")
.to_string(),
);
while let Some(line) = lines.next() {
if let Ok(_) = line.parse::<i64>() {
plan.queries_per_thread.push(Vec::new());
} else {
plan.queries_per_thread
.last_mut()
.unwrap()
.push(line.to_string());
}
plan.queries_per_thread.push(queries);
}
Ok(plan)
}