chorus/src/globals.rs

use crate::config::Config;
use crate::filestore::FileStore;
use crate::ip::HashedIp;
use dashmap::DashMap;
use hyper::server::conn::http1;
use hyper_util::rt::tokio::TokioTimer;
use lazy_static::lazy_static;
use parking_lot::RwLock;
use pocket_db::Store;
use std::sync::atomic::{AtomicU64, AtomicUsize};
use std::sync::OnceLock;
use std::time::{Duration, Instant};
use tokio::sync::broadcast::Sender as BroadcastSender;
use tokio::sync::watch::Sender as WatchSender;

pub struct Globals {
    pub start_time: Instant,
    pub bytes_inbound: AtomicU64,
    pub bytes_outbound: AtomicU64,
    pub config: RwLock<Config>,
    pub store: OnceLock<Store>,
    pub filestore: OnceLock<FileStore>,
    pub http1builder: http1::Builder,
    pub rid: OnceLock<String>,

    /// This is a broadcast channel where new incoming events are advertised by their offset.
    /// Every handler needs to listen to it and check if the incoming event matches any
    /// subscribed fitlers for their client, and if so, send the event to their client under
    /// that subscription.
    pub new_events: BroadcastSender<u64>,

    pub num_connections: AtomicUsize,
    pub num_connections_per_ip: DashMap<HashedIp, usize>,
    pub shutting_down: WatchSender<bool>,
}

lazy_static! {
    pub static ref GLOBALS: Globals = {
        let (new_events, _) = tokio::sync::broadcast::channel(512);
        let (shutting_down, _) = tokio::sync::watch::channel(false);

        let mut http1builder = http1::Builder::new();
        http1builder.keep_alive(true);
        http1builder.timer(TokioTimer::new());
        http1builder.header_read_timeout(Duration::from_secs(5));

        Globals {
            start_time: Instant::now(),
            bytes_inbound: AtomicU64::new(0),
            bytes_outbound: AtomicU64::new(0),
            config: RwLock::new(Default::default()),
            store: OnceLock::new(),
            filestore: OnceLock::new(),
            http1builder,
            rid: OnceLock::new(),
            new_events,
            num_connections: AtomicUsize::new(0),
            num_connections_per_ip: DashMap::new(),
            shutting_down,
        }
    };
}