drop-app/src-tauri/download_manager/src/util/download_thread_control_flag.rs

use std::sync::{
    atomic::{AtomicBool, Ordering},
    Arc,
};

#[derive(PartialEq, Eq, PartialOrd, Ord)]
pub enum DownloadThreadControlFlag {
    Stop,
    Go,
}
/// Go => true
/// Stop => false
impl From<DownloadThreadControlFlag> for bool {
    fn from(value: DownloadThreadControlFlag) -> Self {
        match value {
            DownloadThreadControlFlag::Go => true,
            DownloadThreadControlFlag::Stop => false,
        }
    }
}
/// true => Go
/// false => Stop
impl From<bool> for DownloadThreadControlFlag {
    fn from(value: bool) -> Self {
        if value { DownloadThreadControlFlag::Go } else { DownloadThreadControlFlag::Stop }
    }
}

#[derive(Clone)]
pub struct DownloadThreadControl {
    inner: Arc<AtomicBool>,
}

impl DownloadThreadControl {
    pub fn new(flag: DownloadThreadControlFlag) -> Self {
        Self {
            inner: Arc::new(AtomicBool::new(flag.into())),
        }
    }
    pub fn get(&self) -> DownloadThreadControlFlag {
        self.inner.load(Ordering::Acquire).into()
    }
    pub fn set(&self, flag: DownloadThreadControlFlag) {
        self.inner.store(flag.into(), Ordering::Release);
    }
}