rust/ripgrep
1
0
Fork 0
mirror of https://github.com/BurntSushi/ripgrep.git synced 2025-06-30 22:23:44 +02:00
Code Issues Releases Activity

update

Browse Source
This commit is contained in:
Andrew Gallant
2016-04-03 21:22:09 -04:00
parent 07bff7409b
commit 8d9d602945
2 changed files with 47 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
src/main.rs
View File

@ -62,28 +62,42 @@ fn run(args: &Args) -> Result<u64> {
} else {
let searcher =
try!(LineSearcherBuilder::new(&args.arg_pattern).create());
run_mmap(args, &searcher)
if args.flag_count {
run_mmap_count_only(args, &searcher)
} else {
run_mmap(args, &searcher)
}
}
}
#[inline(never)]
fn run_mmap(args: &Args, searcher: &LineSearcher) -> Result<u64> {
use memmap::{Mmap, Protection};
assert!(args.arg_file.len() == 1);
let mut wtr = io::BufWriter::new(io::stdout());
let mut count = 0;
let mmap = try!(Mmap::open_path(&args.arg_file[0], Protection::Read));
let text = unsafe { mmap.as_slice() };
let mut count = 0;
for m in searcher.search(text) {
if !args.flag_count {
try!(wtr.write(&text[m.start..m.end]));
try!(wtr.write(b"\n"));
}
try!(wtr.write(&text[m.start..m.end]));
try!(wtr.write(b"\n"));
count += 1;
}
if args.flag_count {
try!(writeln!(wtr, "{}", count));
}
Ok(count)
}
#[inline(never)]
fn run_mmap_count_only(args: &Args, searcher: &LineSearcher) -> Result<u64> {
use memmap::{Mmap, Protection};
assert!(args.arg_file.len() == 1);
let mut wtr = io::BufWriter::new(io::stdout());
let mmap = try!(Mmap::open_path(&args.arg_file[0], Protection::Read));
let text = unsafe { mmap.as_slice() };
let count = searcher.search(text).last().map_or(0, |m| m.count + 1);
try!(writeln!(wtr, "{}", count));
Ok(count)
}