82 lines
2.2 KiB
Rust
82 lines
2.2 KiB
Rust
use std::{path::PathBuf, sync::Arc};
|
|
|
|
use tokio::sync::Mutex;
|
|
use tuwunel_core::{
|
|
Error, Result,
|
|
config::Config,
|
|
implement, info,
|
|
utils::{stream, sys},
|
|
};
|
|
|
|
use crate::{Args, args, logging::TracingFlameGuard, runtime};
|
|
|
|
/// Server runtime state; complete
|
|
pub struct Server {
|
|
/// Server runtime state; public portion
|
|
pub server: Arc<tuwunel_core::Server>,
|
|
|
|
pub services: Mutex<Option<Arc<tuwunel_service::Services>>>,
|
|
|
|
_tracing_flame_guard: TracingFlameGuard,
|
|
|
|
#[cfg(feature = "sentry_telemetry")]
|
|
_sentry_guard: Option<::sentry::ClientInitGuard>,
|
|
|
|
#[cfg(all(tuwunel_mods, feature = "tuwunel_mods"))]
|
|
// Module instances; TODO: move to mods::loaded mgmt vector
|
|
pub(crate) mods: tokio::sync::RwLock<Vec<tuwunel_core::mods::Module>>,
|
|
}
|
|
|
|
#[implement(Server)]
|
|
pub fn new(args: Option<&Args>, runtime: Option<&runtime::Handle>) -> Result<Arc<Self>, Error> {
|
|
let _runtime_guard = runtime.map(runtime::Handle::enter);
|
|
|
|
let args_default = args.is_none().then(Args::default);
|
|
let args = args.unwrap_or_else(|| args_default.as_ref().expect("default arguments"));
|
|
let config_paths = args
|
|
.config
|
|
.as_deref()
|
|
.into_iter()
|
|
.flat_map(<[_]>::iter)
|
|
.map(PathBuf::as_path);
|
|
|
|
let config = Config::load(config_paths)
|
|
.and_then(|raw| args::update(raw, args))
|
|
.and_then(|raw| Config::new(&raw))?;
|
|
|
|
let (tracing_flame_guard, logger) = crate::logging::init(&config)?;
|
|
|
|
config.check()?;
|
|
|
|
#[cfg(feature = "sentry_telemetry")]
|
|
let sentry_guard = crate::sentry::init(&config);
|
|
|
|
sys::maximize_fd_limit()
|
|
.expect("Unable to increase maximum soft and hard file descriptor limit");
|
|
|
|
let (_old_width, _new_width) = stream::set_width(config.stream_width_default);
|
|
let (_old_amp, _new_amp) = stream::set_amplification(config.stream_amplification);
|
|
|
|
info!(
|
|
server_name = %config.server_name,
|
|
database_path = ?config.database_path,
|
|
log_levels = %config.log,
|
|
"{}",
|
|
tuwunel_core::version(),
|
|
);
|
|
|
|
Ok(Arc::new(Self {
|
|
server: Arc::new(tuwunel_core::Server::new(config, runtime.cloned(), logger)),
|
|
|
|
services: None.into(),
|
|
|
|
_tracing_flame_guard: tracing_flame_guard,
|
|
|
|
#[cfg(feature = "sentry_telemetry")]
|
|
_sentry_guard: sentry_guard,
|
|
|
|
#[cfg(all(tuwunel_mods, feature = "tuwunel_mods"))]
|
|
mods: tokio::sync::RwLock::new(Vec::new()),
|
|
}))
|
|
}
|