use vin::*;

#[vin::task]
#[derive(Debug, Clone, PartialEq, Eq)]
struct MyTaskActor<T: Clone + Send + Sync + 'static> {
    pub number: u32,
    _phantom: std::marker::PhantomData<T>,
}

#[async_trait]
impl<T: Clone + Send + Sync + 'static> vin::Task for MyTaskActor<T> {
    async fn task(&self, ctx: Self::Context) -> anyhow::Result<()> {
        for i in 0..ctx.number {
            log::info!("{}. iteration", i);
            tokio::time::sleep(std::time::Duration::from_millis(50)).await;
        }

        Err(anyhow::anyhow!("hi, i am error"))
    }
}

#[cfg(test)]
mod tests {
    use std::time::Duration;
    use tracing::Level;
    use super::*;

    #[tokio::test]
    async fn test1() {
        tracing_subscriber::fmt()
            .with_max_level(Level::TRACE)
            .init();

        let ctx = VinContextMyTaskActor::<u32> {
            number: 5,
            _phantom: std::marker::PhantomData,
        };
        let actor = MyTaskActor::start("test_task", ctx);
        drop(actor);
        tokio::time::sleep(Duration::from_millis(100)).await;
        // actor.close();
        vin::wait_for_shutdowns().await;
        // assert_eq!(actor.state(), State::Closed);
        // assert_eq!(actor.is_closed(), true);
    }
}