From fccc22cd4a4a84211d8b61cf561d0f4463f500f1 Mon Sep 17 00:00:00 2001
From: jakka <jakkadoujin@gmail.com>
Date: Mon, 7 Jul 2025 13:18:57 +0300
Subject: better threading handling

---
 src/main.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index d1e4562..122ce60 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -116,7 +116,11 @@ fn main() -> Result<()> {
     if args.get_flag("doit") {
         let hanlder = running.clone();
         let threads = *args.get_one::<usize>("threads").unwrap();
-        files::reencode_files(conn, hanlder, threads)?;
+        let pool = rayon::ThreadPoolBuilder::new()
+            .num_threads(threads)
+            .build()?;
+
+        pool.install(|| files::reencode_files(conn, hanlder))?;
     }
     Ok::<(), anyhow::Error>(())
 }
-- 
cgit v1.3.1