cmd_lib 1.9.6

Common rust commandline macros and utils, to write shell script like tasks easily
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

// get disk read bandwidth with multiple threads
//
// Usage: dd_test [-b <block_size>] [-t <thread_num>] -f <file>
//
// e.g:
//! ➜  rust_cmd_lib git:(master) ✗ cargo run --example dd_test -- -b 4096 -f /dev/nvme0n1 -t 4
//!     Finished dev [unoptimized + debuginfo] target(s) in 0.04s
//!      Running `target/debug/examples/dd_test -b 4096 -f /dev/nvme0n1 -t 4`
//! [INFO ] Dropping caches at first
//! [INFO ] Running with thread_num: 4, block_size: 4096
//! [INFO ] thread 3 bandwidth: 317 MB/s
//! [INFO ] thread 1 bandwidth: 289 MB/s
//! [INFO ] thread 0 bandwidth: 281 MB/s
//! [INFO ] thread 2 bandwidth: 279 MB/s
//! [INFO ] Total bandwidth: 1.11 GiB/s
//! ```
use byte_unit::Byte;
use cmd_lib::*;
use rayon::prelude::*;
use std::time::Instant;
use clap::Parser;

const DATA_SIZE: u64 = 10 * 1024 * 1024 * 1024; // 10GB data

#[derive(Parser)]
#[clap(name = "dd_test", about = "Get disk read bandwidth.")]
struct Opt {
    #[clap(short, default_value = "4096")]
    block_size: u64,
    #[clap(short, default_value = "1")]
    thread_num: u64,
    #[clap(short)]
    file: String,
}

#[cmd_lib::main]
fn main() -> CmdResult {
    let Opt {
        block_size,
        thread_num,
        file,
    } = Opt::parse();

    run_cmd! (
        info "Dropping caches at first";
        sudo bash -c "echo 3 > /proc/sys/vm/drop_caches";
        info "Running with thread_num: $thread_num, block_size: $block_size";
    )?;
    let cnt = DATA_SIZE / thread_num / block_size;
    let now = Instant::now();
    (0..thread_num).into_par_iter().for_each(|i| {
        let off = cnt * i;
        let bandwidth = run_fun!(
            sudo bash -c "dd if=$file of=/dev/null bs=$block_size skip=$off count=$cnt 2>&1"
            | awk r#"/copied/{print $(NF-1) " " $NF}"#
        )
        .unwrap_or_else(|_| cmd_die!("thread $i failed"));
        info!("thread {i} bandwidth: {bandwidth}");
    });
    let total_bandwidth =
        Byte::from_bytes((DATA_SIZE / now.elapsed().as_secs()) as u128).get_appropriate_unit(true);
    info!("Total bandwidth: {total_bandwidth}/s");

    Ok(())
}