Last active
May 27, 2020 18:55
-
-
Save praveenperera/e605dd48f32cd0d42ad22a4b73d854e6 to your computer and use it in GitHub Desktop.
build a path list in parallel
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
use crossbeam_channel as channel; | |
use ignore::WalkBuilder; | |
use std::path::{Path, PathBuf}; | |
#[derive(Debug)] | |
enum Message { | |
FoundPath(Result<ignore::DirEntry, ignore::Error>), | |
DoneScanning, | |
} | |
fn get_search_paths_from_starting_path(starting_path: &Path) -> Vec<PathBuf> { | |
let (tx, rx) = channel::unbounded(); | |
let mut paths: Vec<ignore::DirEntry> = vec![]; | |
WalkBuilder::new(starting_path).build_parallel().run(|| { | |
let tx = tx.clone(); | |
Box::new(move |result| { | |
tx.send(Message::FoundPath(result)).unwrap(); | |
ignore::WalkState::Continue | |
}) | |
}); | |
tx.send(Message::DoneScanning).unwrap(); | |
loop { | |
match rx.recv() { | |
Ok(Message::DoneScanning) => break, | |
Ok(Message::FoundPath(Ok(path))) => paths.push(path), | |
_ => (), | |
} | |
} | |
drop(rx); | |
paths | |
.iter() | |
.filter(|f| f.path().is_file()) | |
.map(|file| file.path().to_owned()) | |
.collect() | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment