From c4a1418cc299066b03509e534734e3ae06b73145 Mon Sep 17 00:00:00 2001
From: Nulo <git@nulo.in>
Date: Sun, 23 Apr 2023 21:40:13 -0300
Subject: [PATCH] rust: simplificar

---
 rust/src/main.rs | 50 ++++++++++++++++++++++++------------------------
 1 file changed, 25 insertions(+), 25 deletions(-)

diff --git a/rust/src/main.rs b/rust/src/main.rs
index 5932d64..1ceb148 100644
--- a/rust/src/main.rs
+++ b/rust/src/main.rs
@@ -16,33 +16,36 @@ fn db_path() -> &'static str {
 
 fn main() -> Result<(), Box<dyn Error>> {
     _ = remove_file(db_path());
-    let conn = make_conn()?;
-    make_db_schema(&conn)?;
+    {
+        let conn = make_conn()?;
+        make_db_schema(&conn)?;
+    }
     let input = Path::new("./input");
-
     let paths = iterator(input).collect::<Vec<PathBuf>>();
 
-    let chunks = paths.chunks(paths.len() / 4);
-    println!("{}", chunks.len());
+    let chunks = paths.chunks(paths.len() / 4 + paths.len() % 4);
+    println!("threads: {}", chunks.len());
 
-    let (tx, rx): (Sender<Entry>, Receiver<Entry>) = mpsc::channel();
-
-    let threads = chunks
-        .map(|chunk| {
-            let chunkk = chunk.to_vec();
-            let ttx = tx.clone();
-            thread::spawn(move || {
-                let input = Path::new("./input");
-                let params = file_compress_params();
-                for path in chunkk {
-                    compress_file(&path, input, &params, &ttx);
-                }
+    let (rx, threads) = {
+        let (tx, rx): (Sender<Entry>, Receiver<Entry>) = mpsc::channel();
+        let producers = chunks
+            .map(|chunk| {
+                let chunkk = chunk.to_vec();
+                let ttx = tx.clone();
+                thread::spawn(move || {
+                    let input = Path::new("./input");
+                    let params = file_compress_params();
+                    for path in chunkk {
+                        compress_file(&path, input, &params, &ttx);
+                    }
+                })
             })
-        })
-        .collect::<Vec<JoinHandle<()>>>();
-    let collector = thread::spawn(|| {
+            .collect::<Vec<JoinHandle<()>>>();
+
+        (rx, producers)
+    };
+    {
         let mut conn = make_conn().unwrap();
-        println!("thread");
         let trans = conn.transaction().unwrap();
         {
             let mut insert_stmt = trans
@@ -53,10 +56,7 @@ fn main() -> Result<(), Box<dyn Error>> {
             }
         }
         trans.commit().unwrap();
-    });
-    drop(tx);
-
-    collector.join().unwrap();
+    }
     for thread in threads {
         thread.join().unwrap();
     }