rust_htslib/
tpool.rs

1use std::cell::RefCell;
2use std::sync::Arc;
3
4pub use crate::errors::{Error, Result};
5use crate::htslib;
6
7/// An HTSlib thread pool. Create a thread pool and use `set_thread_pool()` methods
8/// to share a thread pool across multiple BAM readers & writers.
9/// The Rust wrapper holds the htslib thread pool behind a Rc, and a Rc reference
10/// to the thread pool is held by each reader / writer so you don't need to
11/// explicitly manage the lifetime of the `ThreadPool`.
12#[derive(Clone, Debug)]
13pub struct ThreadPool {
14    pub(crate) handle: Arc<RefCell<InnerThreadPool>>,
15}
16
17impl ThreadPool {
18    /// Create a new thread pool with `n_threads` threads.
19    pub fn new(n_threads: u32) -> Result<ThreadPool> {
20        let ret = unsafe { htslib::hts_tpool_init(n_threads as i32) };
21
22        if ret.is_null() {
23            Err(Error::ThreadPool)
24        } else {
25            let inner = htslib::htsThreadPool {
26                pool: ret,
27                // this matches the default size
28                // used in hts_set_threads.
29                qsize: n_threads as i32 * 2,
30            };
31            let inner = InnerThreadPool { inner };
32
33            let handle = Arc::new(RefCell::new(inner));
34            Ok(ThreadPool { handle })
35        }
36    }
37}
38
39/// Internal htsThreadPool
40#[derive(Clone, Debug)]
41pub struct InnerThreadPool {
42    pub(crate) inner: htslib::htsThreadPool,
43}
44
45impl Drop for InnerThreadPool {
46    fn drop(&mut self) {
47        if !self.inner.pool.is_null() {
48            unsafe {
49                htslib::hts_tpool_destroy(self.inner.pool);
50            }
51        }
52
53        self.inner.pool = std::ptr::null_mut();
54    }
55}
rust_htslib/tpool.rs

rust_htslib/
tpool.rs
