spawner.rs

//! A thin wrapper for tokio spawn to launch the UDP server launcher as a new task.
use std::net::SocketAddr;
use std::sync::Arc;
use std::time::Duration;

use bittorrent_udp_tracker_core::container::UdpTrackerCoreContainer;
use derive_more::derive::Display;
use derive_more::Constructor;
use tokio::sync::oneshot;
use tokio::task::JoinHandle;
use torrust_server_lib::signals::{Halted, Started};

use super::launcher::Launcher;
use crate::container::UdpTrackerServerContainer;

#[derive(Constructor, Copy, Clone, Debug, Display)]
#[display("(with socket): {bind_to}")]
pub struct Spawner {
    pub bind_to: SocketAddr,
}

impl Spawner {
    /// It spawns a new task to run the UDP server instance.
    ///
    /// # Panics
    ///
    /// It would panic if unable to resolve the `local_addr` from the supplied ´socket´.
    #[must_use]
    pub fn spawn_launcher(
        &self,
        udp_tracker_core_container: Arc<UdpTrackerCoreContainer>,
        udp_tracker_server_container: Arc<UdpTrackerServerContainer>,
        cookie_lifetime: Duration,
        tx_start: oneshot::Sender<Started>,
        rx_halt: oneshot::Receiver<Halted>,
    ) -> JoinHandle<Spawner> {
        let spawner = Self::new(self.bind_to);

        tokio::spawn(async move {
            Launcher::run_with_graceful_shutdown(
                udp_tracker_core_container,
                udp_tracker_server_container,
                spawner.bind_to,
                cookie_lifetime,
                tx_start,
                rx_halt,
            )
            .await;
            spawner
        })
    }
}