1
Fork 0
mirror of https://github.com/RGBCube/uutils-coreutils synced 2025-08-04 23:17:46 +00:00

factors: Avoid repeatedly locking and flushing stdout

By default, stdout's LineWriter results one syscall per line, i.e. a billion
syscalls when factoring a billion numbers...

Buffering the output yields a ~28% speedup.
This commit is contained in:
nicoo 2020-06-16 15:06:28 +02:00 committed by Alex Lyon
parent 2869248318
commit ef12991ee7

View file

@ -12,8 +12,9 @@ extern crate rand;
extern crate uucore; extern crate uucore;
use std::collections::BTreeMap; use std::collections::BTreeMap;
use std::error::Error;
use std::fmt; use std::fmt;
use std::io::{stdin, BufRead}; use std::io::{self, stdin, stdout, BufRead, Write};
use std::ops; use std::ops;
mod miller_rabin; mod miller_rabin;
@ -101,35 +102,40 @@ fn factor(mut n: u64) -> Factors {
factors factors
} }
fn print_factors(num: u64) { fn print_factors_str(num_str: &str, w: &mut impl io::Write) -> Result<(), Box<dyn Error>> {
print!("{}:{}", num, factor(num)); num_str
println!(); .parse::<u64>()
} .map_err(|e| e.into())
.and_then(|x| writeln!(w, "{}:{}", x, factor(x)).map_err(|e| e.into()))
fn print_factors_str(num_str: &str) {
if let Err(e) = num_str.parse::<u64>().and_then(|x| {
print_factors(x);
Ok(())
}) {
show_warning!("{}: {}", num_str, e);
}
} }
pub fn uumain(args: impl uucore::Args) -> i32 { pub fn uumain(args: impl uucore::Args) -> i32 {
let matches = app!(SYNTAX, SUMMARY, LONG_HELP).parse(args.collect_str()); let matches = app!(SYNTAX, SUMMARY, LONG_HELP).parse(args.collect_str());
let stdout = stdout();
let mut w = io::BufWriter::new(stdout.lock());
if matches.free.is_empty() { if matches.free.is_empty() {
let stdin = stdin(); let stdin = stdin();
for line in stdin.lock().lines() { for line in stdin.lock().lines() {
for number in line.unwrap().split_whitespace() { for number in line.unwrap().split_whitespace() {
print_factors_str(number); if let Err(e) = print_factors_str(number, &mut w) {
show_warning!("{}: {}", number, e);
}
} }
} }
} else { } else {
for num_str in &matches.free { for number in &matches.free {
print_factors_str(num_str); if let Err(e) = print_factors_str(number, &mut w) {
show_warning!("{}: {}", number, e);
} }
} }
}
if let Err(e) = w.flush() {
show_error!("{}", e);
}
0 0
} }