techno-tanoc / progress_test Goto Github PK

struct ProgressInner<T> {
    size: u64,
    buf: T,
}

pub struct Progress<T> {
    inner: Arc<Mutex<ProgressInner<T>>>,
}

impl<T> Progress<T> {
    pub fn new(buf: T) -> Self {
        let inner = Arc::new(Mutex::new(ProgressInner {
            size: 0,
            buf,
        }));
        Progress { inner }
    }

    pub async fn to_size(&self) -> u64 {
        let pg = self.inner.lock().await;
        pg.size
    }
}