aquatic/aquatic_udp/src/lib/mio/common.rs

use aquatic_common::access_list::AccessListArcSwap;
use parking_lot::Mutex;
use std::sync::{atomic::AtomicUsize, Arc};

use crate::common::*;

pub enum ConnectedRequest {
    Announce(AnnounceRequest),
    Scrape(ScrapeRequest),
}

pub enum ConnectedResponse {
    Announce(AnnounceResponse),
    Scrape(ScrapeResponse),
}

impl Into<Response> for ConnectedResponse {
    fn into(self) -> Response {
        match self {
            Self::Announce(response) => Response::Announce(response),
            Self::Scrape(response) => Response::Scrape(response),
        }
    }
}

#[derive(Default)]
pub struct Statistics {
    pub requests_received: AtomicUsize,
    pub responses_sent: AtomicUsize,
    pub bytes_received: AtomicUsize,
    pub bytes_sent: AtomicUsize,
}

#[derive(Clone)]
pub struct State {
    pub access_list: Arc<AccessListArcSwap>,
    pub torrents: Arc<Mutex<TorrentMaps>>,
    pub statistics: Arc<Statistics>,
}

impl Default for State {
    fn default() -> Self {
        Self {
            access_list: Arc::new(AccessListArcSwap::default()),
            torrents: Arc::new(Mutex::new(TorrentMaps::default())),
            statistics: Arc::new(Statistics::default()),
        }
    }
}