52 lines
1.3 KiB
Rust
52 lines
1.3 KiB
Rust
use {
|
|
parking_lot::{Condvar, Mutex},
|
|
std::{mem, rc::Rc, sync::Arc},
|
|
uapi::OwnedFd,
|
|
};
|
|
|
|
pub struct FdCloser {
|
|
fds: Mutex<Vec<OwnedFd>>,
|
|
cv: Condvar,
|
|
}
|
|
|
|
impl FdCloser {
|
|
pub fn new() -> Arc<Self> {
|
|
let slf = Arc::new(Self {
|
|
fds: Mutex::new(Vec::new()),
|
|
cv: Condvar::new(),
|
|
});
|
|
let slf2 = slf.clone();
|
|
std::thread::Builder::new()
|
|
.name("fd closer".to_string())
|
|
.spawn(move || {
|
|
let mut fds = vec![];
|
|
let mut lock = slf2.fds.lock();
|
|
loop {
|
|
mem::swap(&mut *lock, &mut fds);
|
|
if fds.len() > 0 {
|
|
drop(lock);
|
|
fds.clear();
|
|
lock = slf2.fds.lock();
|
|
} else {
|
|
slf2.cv.wait(&mut lock);
|
|
}
|
|
}
|
|
})
|
|
.unwrap();
|
|
slf
|
|
}
|
|
|
|
pub fn close(&self, fd: Rc<OwnedFd>) {
|
|
match Rc::try_unwrap(fd) {
|
|
Ok(fd) => {
|
|
self.fds.lock().push(fd);
|
|
self.cv.notify_all();
|
|
}
|
|
Err(_e) => {
|
|
log::warn!(
|
|
"Could not close file descriptor in separate thread. There are still references."
|
|
);
|
|
}
|
|
}
|
|
}
|
|
}
|