diff --git a/crates/bench/src/data_gen.rs b/crates/bench/src/data_gen.rs index 8d5d675..e885ccc 100644 --- a/crates/bench/src/data_gen.rs +++ b/crates/bench/src/data_gen.rs @@ -83,11 +83,11 @@ pub fn generate_growable_file(dir: &Path) -> std::io::Result { /// Append `count` lines to the file pub fn append_lines(path: &Path, count: usize) -> std::io::Result<()> { + let existing_lines = count_existing_lines(path)?; let mut file = BufWriter::with_capacity( 64 * 1024, fs::OpenOptions::new().append(true).open(path)?, ); - let existing_lines = count_existing_lines(path).unwrap_or(0); for i in 0..count { writeln!( file, @@ -117,7 +117,12 @@ pub fn rotate_file(path: &Path) -> std::io::Result { fn count_existing_lines(path: &Path) -> std::io::Result { let file = fs::File::open(path)?; let reader = BufReader::new(file); - Ok(reader.lines().count() as u64) + let mut count = 0u64; + for line in reader.lines() { + line?; + count += 1; + } + Ok(count) } #[cfg(test)]