nym/nym-node/src/http/mod.rs

// Copyright 2023 - Nym Technologies SA <contact@nymtech.net>
// SPDX-License-Identifier: GPL-3.0-only

use axum::extract::connect_info::IntoMakeServiceWithConnectInfo;
use axum::Router;
use hyper::server::conn::AddrIncoming;
use hyper::Server;
use nym_task::TaskClient;
use std::net::SocketAddr;
use tracing::{debug, error, info};

pub mod middleware;
pub mod router;
pub mod state;

pub use router::{api, landing_page, Config, NymNodeRouter};

pub struct NymNodeHTTPServer {
    task_client: Option<TaskClient>,
    inner: Server<AddrIncoming, IntoMakeServiceWithConnectInfo<Router, SocketAddr>>,
}

impl NymNodeHTTPServer {
    pub(crate) fn new(
        inner: Server<AddrIncoming, IntoMakeServiceWithConnectInfo<Router, SocketAddr>>,
    ) -> Self {
        NymNodeHTTPServer {
            task_client: None,
            inner,
        }
    }

    #[must_use]
    pub fn with_task_client(mut self, task_client: TaskClient) -> Self {
        self.task_client = Some(task_client);
        self
    }

    async fn run_server_forever(
        server: Server<AddrIncoming, IntoMakeServiceWithConnectInfo<Router, SocketAddr>>,
    ) {
        if let Err(err) = server.await {
            error!("the HTTP server has terminated with the error: {err}");
        } else {
            error!("the HTTP server has terminated with producing any errors");
        }
    }

    pub async fn run(self) {
        info!("Started NymNodeHTTPServer on {}", self.inner.local_addr());
        if let Some(mut task_client) = self.task_client {
            tokio::select! {
                _ = task_client.recv_with_delay() => {
                    debug!("NymNodeHTTPServer: Received shutdown");
                }
                _ = Self::run_server_forever(self.inner) => { }
            }
        } else {
            Self::run_server_forever(self.inner).await
        }

        debug!("NymNodeHTTPServer: Exiting");
    }
}