Nulo/tofufirme
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

rust: simplificar

Browse source
This commit is contained in:
Cat /dev/Nulo 2023-04-23 21:40:13 -03:00
parent 92a9ab8340
commit c4a1418cc2
1 changed files with 25 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
rust/src/main.rs
View file

@ -16,18 +16,19 @@ fn db_path() -> &'static str {
fn main() -> Result<(), Box<dyn Error>> { fn main() -> Result<(), Box<dyn Error>> {
_ = remove_file(db_path()); _ = remove_file(db_path());
{
let conn = make_conn()?; let conn = make_conn()?;
make_db_schema(&conn)?; make_db_schema(&conn)?;
}
let input = Path::new("./input"); let input = Path::new("./input");
let paths = iterator(input).collect::<Vec<PathBuf>>(); let paths = iterator(input).collect::<Vec<PathBuf>>();
let chunks = paths.chunks(paths.len() / 4); let chunks = paths.chunks(paths.len() / 4 + paths.len() % 4);
println!("{}", chunks.len()); println!("threads: {}", chunks.len());
let (rx, threads) = {
let (tx, rx): (Sender<Entry>, Receiver<Entry>) = mpsc::channel(); let (tx, rx): (Sender<Entry>, Receiver<Entry>) = mpsc::channel();
let producers = chunks
let threads = chunks
.map(|chunk| { .map(|chunk| {
let chunkk = chunk.to_vec(); let chunkk = chunk.to_vec();
let ttx = tx.clone(); let ttx = tx.clone();
@ -40,9 +41,11 @@ fn main() -> Result<(), Box<dyn Error>> {
}) })
}) })
.collect::<Vec<JoinHandle<()>>>(); .collect::<Vec<JoinHandle<()>>>();
let collector = thread::spawn(|| {
(rx, producers)
};
{
let mut conn = make_conn().unwrap(); let mut conn = make_conn().unwrap();
println!("thread");
let trans = conn.transaction().unwrap(); let trans = conn.transaction().unwrap();
{ {
let mut insert_stmt = trans let mut insert_stmt = trans
@ -53,10 +56,7 @@ fn main() -> Result<(), Box<dyn Error>> {
} }
} }
trans.commit().unwrap(); trans.commit().unwrap();
}); }
drop(tx);
collector.join().unwrap();
for thread in threads { for thread in threads {
thread.join().unwrap(); thread.join().unwrap();
} }