split: add benchmarks

Author: sylvestre

src/uu/split/Cargo.toml

@@ -27,3 +27,12 @@ fluent = { workspace = true }
 [[bin]]
 name = "split"
 path = "src/main.rs"
+
+[dev-dependencies]
+divan = { workspace = true }
+tempfile = { workspace = true }
+uucore = { workspace = true, features = ["benchmark"] }
+
+[[bench]]
+name = "split_bench"
+harness = false

src/uu/split/benches/split_bench.rs

@@ -0,0 +1,97 @@
+// This file is part of the uutils coreutils package.
+//
+// For the full copyright and license information, please view the LICENSE
+// file that was distributed with this source code.
+
+use divan::{Bencher, black_box};
+use tempfile::TempDir;
+use uu_split::uumain;
+use uucore::benchmark::{run_util_function, setup_test_file, text_data};
+
+/// Benchmark splitting by line count
+#[divan::bench]
+fn split_lines(bencher: Bencher) {
+    let data = text_data::generate_by_lines(100_000, 80);
+    let file_path = setup_test_file(&data);
+
+    bencher
+        .with_inputs(|| {
+            let output_dir = TempDir::new().unwrap();
+            let prefix = output_dir.path().join("x");
+            (output_dir, prefix.to_str().unwrap().to_string())
+        })
+        .bench_values(|(output_dir, prefix)| {
+            black_box(run_util_function(
+                uumain,
+                &["-l", "1000", file_path.to_str().unwrap(), &prefix],
+            ));
+            drop(output_dir);
+        });
+}
+
+/// Benchmark splitting by byte size
+#[divan::bench]
+fn split_bytes(bencher: Bencher) {
+    let data = text_data::generate_by_size(10, 80);
+    let file_path = setup_test_file(&data);
+
+    bencher
+        .with_inputs(|| {
+            let output_dir = TempDir::new().unwrap();
+            let prefix = output_dir.path().join("x");
+            (output_dir, prefix.to_str().unwrap().to_string())
+        })
+        .bench_values(|(output_dir, prefix)| {
+            black_box(run_util_function(
+                uumain,
+                &["-b", "100K", file_path.to_str().unwrap(), &prefix],
+            ));
+            drop(output_dir);
+        });
+}
+
+/// Benchmark splitting by number of chunks
+#[divan::bench]
+fn split_number_chunks(bencher: Bencher) {
+    let data = text_data::generate_by_lines(100_000, 80);
+    let file_path = setup_test_file(&data);
+
+    bencher
+        .with_inputs(|| {
+            let output_dir = TempDir::new().unwrap();
+            let prefix = output_dir.path().join("x");
+            (output_dir, prefix.to_str().unwrap().to_string())
+        })
+        .bench_values(|(output_dir, prefix)| {
+            black_box(run_util_function(
+                uumain,
+                &["-n", "10", file_path.to_str().unwrap(), &prefix],
+            ));
+            drop(output_dir);
+        });
+}
+
+/// Benchmark splitting with numeric suffix
+#[divan::bench]
+fn split_numeric_suffix(bencher: Bencher) {
+    let data = text_data::generate_by_lines(100_000, 80);
+    let file_path = setup_test_file(&data);
+
+    bencher
+        .with_inputs(|| {
+            let output_dir = TempDir::new().unwrap();
+            let prefix = output_dir.path().join("x");
+            (output_dir, prefix.to_str().unwrap().to_string())
+        })
+        .bench_values(|(output_dir, prefix)| {
+            black_box(run_util_function(
+                uumain,
+                &["-d", "-l", "500", file_path.to_str().unwrap(), &prefix],
+            ));
+            drop(output_dir);
+        });
+}
+
+fn main() {
+    divan::main();
+}