Updated the disk read test to use fio instead of dd; added new test for 4k random reads (previous test was for sequential reads)
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
This commit is contained in:
parent
6f86266dee
commit
e41a4dfffb
20
src/main.rs
20
src/main.rs
@ -28,9 +28,20 @@ struct Disk {
|
|||||||
|
|
||||||
#[derive(Subcommand)]
|
#[derive(Subcommand)]
|
||||||
enum DiskCommands {
|
enum DiskCommands {
|
||||||
// disk read subcommand
|
// sequential disk read subcommand
|
||||||
#[clap(name = "read-test", about = "Read from /dev/zero to determine sequential disk read speeds.")]
|
#[clap(name = "read_seq", about = "Sequential disk read speed test.")]
|
||||||
ReadTest {
|
ReadSeqTest {
|
||||||
|
#[clap(short = 'f', default_value_t = String::from("/tmp/disk-test.tmp"))]
|
||||||
|
tempfile: String,
|
||||||
|
#[clap(short = 's', default_value_t = 15)]
|
||||||
|
size: u8,
|
||||||
|
},
|
||||||
|
|
||||||
|
// random disk read subcommand
|
||||||
|
#[clap(name = "read_rand", about = "Random 4K disk read speed test.")]
|
||||||
|
ReadRandTest {
|
||||||
|
#[clap(short = 'f', default_value_t = String::from("/tmp/disk-test.tmp"))]
|
||||||
|
tempfile: String,
|
||||||
#[clap(short = 's', default_value_t = 15)]
|
#[clap(short = 's', default_value_t = 15)]
|
||||||
size: u8,
|
size: u8,
|
||||||
},
|
},
|
||||||
@ -78,7 +89,8 @@ fn main() {
|
|||||||
// map subcommands back to the main command
|
// map subcommands back to the main command
|
||||||
match &cli.command {
|
match &cli.command {
|
||||||
Commands::Disk(args) => match &args.disk_commands {
|
Commands::Disk(args) => match &args.disk_commands {
|
||||||
DiskCommands::ReadTest { size } => tests::disk::disk_read_test(size),
|
DiskCommands::ReadSeqTest { tempfile, size } => tests::disk::disk_read_seq_test(tempfile, size),
|
||||||
|
DiskCommands::ReadRandTest { tempfile, size } => tests::disk::disk_read_rand_test(tempfile, size),
|
||||||
DiskCommands::WriteTest { tempfile, size } => tests::disk::disk_write_test(tempfile, size),
|
DiskCommands::WriteTest { tempfile, size } => tests::disk::disk_write_test(tempfile, size),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1,31 +1,58 @@
|
|||||||
use std::fs;
|
use std::fs;
|
||||||
use std::process::Command;
|
use std::process::Command;
|
||||||
|
|
||||||
// test disk read speeds by reading for /dev/zero and writing it to /dev/null
|
// test disk sequential read speeds w/ fio
|
||||||
pub fn disk_read_test(size: &u8) {
|
pub fn disk_read_seq_test(tempfile: &str, size: &u8) {
|
||||||
// convert size in Gigabytes down to Megabytes
|
// run the fio command
|
||||||
let size_gigs: u32 = (*size as u32 * 1024 * 8).into();
|
let output = Command::new("fio")
|
||||||
|
.arg("--name=TEST")
|
||||||
// run sync to clear out any disk caches prior to running
|
.arg(format!("--filename={}", tempfile))
|
||||||
Command::new("sync");
|
.arg("--rw=read")
|
||||||
|
.arg("--size=2g")
|
||||||
// run the dd command
|
.arg(format!("--io_size={}g", size))
|
||||||
let output = Command::new("dd")
|
.arg("--blocksize=1024k")
|
||||||
.arg("bs=128k")
|
.arg("--ioengine=libaio")
|
||||||
.arg(format!("count={}", size_gigs))
|
.arg("--fsync=10000")
|
||||||
.arg("if=/dev/zero")
|
.arg("--iodepth=32")
|
||||||
.arg("of=/dev/null")
|
.arg("--direct=1")
|
||||||
|
.arg("--numjobs=1")
|
||||||
|
.arg("--runtime=60")
|
||||||
|
.arg("--group_reporting")
|
||||||
.output()
|
.output()
|
||||||
.expect("Failed to execute command");
|
.expect("Failed to execute command");
|
||||||
|
|
||||||
// check that the command succeeded
|
// check that the command succeeded
|
||||||
assert!(output.status.success());
|
assert!(output.status.success());
|
||||||
|
|
||||||
// dd's output ends up in stderr
|
// print the test's output
|
||||||
println!("{}", String::from_utf8_lossy(&output.stderr));
|
println!("{}", String::from_utf8_lossy(&output.stdout));
|
||||||
|
}
|
||||||
|
|
||||||
// run another sync to clear out the disk's cache
|
// test disk 4K random read speeds w/ fio
|
||||||
Command::new("sync");
|
pub fn disk_read_rand_test(tempfile: &str, size: &u8) {
|
||||||
|
// run the fio command
|
||||||
|
let output = Command::new("fio")
|
||||||
|
.arg("--name=TEST")
|
||||||
|
.arg(format!("--filename={}", tempfile))
|
||||||
|
.arg("--rw=randread")
|
||||||
|
.arg("--size=2g")
|
||||||
|
.arg(format!("--io_size={}g", size))
|
||||||
|
.arg("--blocksize=4k")
|
||||||
|
.arg("--ioengine=libaio")
|
||||||
|
.arg("--fsync=1")
|
||||||
|
.arg("--iodepth=32")
|
||||||
|
.arg("--direct=1")
|
||||||
|
.arg("--numjobs=1")
|
||||||
|
.arg("--runtime=60")
|
||||||
|
.arg("--group_reporting")
|
||||||
|
.output()
|
||||||
|
.expect("Failed to execute command");
|
||||||
|
|
||||||
|
// check that the command succeeded
|
||||||
|
assert!(output.status.success());
|
||||||
|
|
||||||
|
// print the test's output
|
||||||
|
println!("{}", String::from_utf8_lossy(&output.stdout));
|
||||||
}
|
}
|
||||||
|
|
||||||
// test disk write speeds by continually writing zeroes to it
|
// test disk write speeds by continually writing zeroes to it
|
||||||
|
Loading…
Reference in New Issue
Block a user