use crate::{
linear_owned_reusable::LinearOwnedReusable, linear_page::LinearPage,
linear_reusable::LinearReusable,
};
use std::sync::Arc;
pub struct LinearObjectPool<T> {
reset: Box<dyn Fn(&mut T) + Send + Sync>,
init: Box<dyn Fn() -> T + Send + Sync>,
head: LinearPage<T>,
}
impl<T> LinearObjectPool<T> {
#[inline]
pub fn new<R, I>(init: I, reset: R) -> Self
where
R: Fn(&mut T) + 'static + Send + Sync,
I: Fn() -> T + 'static + Clone + Send + Sync,
{
Self {
reset: Box::new(reset),
init: Box::new(init.clone()),
head: LinearPage::new(init),
}
}
#[inline]
pub fn pull(&self) -> LinearReusable<T> {
let (page, page_id) = self.head.alloc(&self.init);
unsafe { LinearReusable::new(self, page_id, page) }
}
#[inline]
pub fn pull_owned(self: &Arc<Self>) -> LinearOwnedReusable<T> {
let (page, page_id) = self.head.alloc(&self.init);
unsafe { LinearOwnedReusable::new(self.clone(), page_id, page) }
}
#[inline]
pub(crate) fn get_reset_callback(&self) -> &dyn Fn(&mut T) {
&self.reset
}
}