bug fix

eric9n · Aug 19, 2024 · f109e8d · f109e8d
1 parent 27095d8
commit f109e8d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/kr2r/src/bin/resolve.rs b/kr2r/src/bin/resolve.rs
@@ -167,14 +167,17 @@ fn process_batch<P: AsRef<Path>>(
                     );
                     Some(output_line)
                 } else {
+                    eprintln!("can't find {} in sample_id map file", k);
                     None
                 }
             },
             |result| {
-                while let Some(Some(res)) = result.next() {
-                    writer
-                        .write_all(res.as_bytes())
-                        .expect("write output content error");
+                while let Some(output) = result.next() {
+                    if let Some(res) = output {
+                        writer
+                            .write_all(res.as_bytes())
+                            .expect("write output content error");
+                    }
                 }
             },
         )

diff --git a/seqkmer/src/parallel.rs b/seqkmer/src/parallel.rs
@@ -7,6 +7,7 @@ use scoped_threadpool::Pool;
 use std::collections::HashMap;
 use std::io::Result;
 use std::sync::Arc;
+
 pub struct ParallelResult<P>
 where
     P: Send,
@@ -164,7 +165,7 @@ where
             let _ = func(&mut parallel_result);
         });
 
-        // pool_scope.join_all();
+        pool_scope.join_all();
     });
 
     Ok(())