spacegate_kernel/helper_layers/
reload.rs

use std::sync::{Arc, OnceLock};

use crossbeam_utils::sync::ShardedLock;
use tower_layer::Layer;

#[derive(Default, Debug, Clone)]
pub struct ReloadLayer<S> {
    pub reloader: Reloader<S>,
}

impl<S> Layer<S> for ReloadLayer<S>
where
    S: Send,
{
    type Service = Reload<S>;

    fn layer(&self, inner: S) -> Self::Service {
        let inner = Arc::new(ShardedLock::new(inner));
        self.reloader.setup(inner.clone());
        Reload { service: inner }
    }
}

#[derive(Debug, Clone)]
pub struct Reload<S>
where
    S: Send,
{
    pub(crate) service: Arc<ShardedLock<S>>,
}

#[derive(Debug)]
pub struct Reloader<S> {
    pub service: Arc<OnceLock<Arc<ShardedLock<S>>>>,
}

impl<S> Default for Reloader<S> {
    fn default() -> Self {
        Self { service: Default::default() }
    }
}

impl<S> Clone for Reloader<S> {
    fn clone(&self) -> Self {
        Self { service: self.service.clone() }
    }
}

impl<S> Reloader<S> {
    pub fn setup(&self, service: Arc<ShardedLock<S>>) {
        if self.service.set(service).is_err() {
            tracing::warn!("reloader already settled");
        }
    }
    pub fn reload(&self, service: S) {
        if let Some(wg) = self.service.get() {
            let mut wg = wg.write().expect("should never be poisoned");
            *wg = service;
        } else {
            tracing::warn!("reloader not initialized");
        }
    }
    pub fn into_layer(self) -> ReloadLayer<S> {
        ReloadLayer { reloader: self }
    }
}

impl<Request, S> hyper::service::Service<Request> for Reload<S>
where
    Request: Send + Sync + 'static,
    S: hyper::service::Service<Request> + Send + Sync + 'static,
    <S as hyper::service::Service<Request>>::Future: std::marker::Send,
{
    type Response = S::Response;

    type Error = S::Error;

    type Future = <S as hyper::service::Service<Request>>::Future;

    fn call(&self, req: Request) -> Self::Future {
        let service = self.service.clone();
        let rg = service.read().expect("should never be posisoned");
        let fut = rg.call(req);
        drop(rg);
        fut
    }
}
spacegate_kernel/helper_layers/reload.rs

spacegate_kernel/helper_layers/
reload.rs
