Changeset View
Changeset View
Standalone View
Standalone View
services/tunnelbroker/src/main.rs
| pub mod constants; | pub mod constants; | ||||
| pub mod grpc; | pub mod grpc; | ||||
| pub mod websockets; | pub mod websockets; | ||||
| use dashmap::DashMap; | use dashmap::DashMap; | ||||
| use once_cell::sync::Lazy; | use once_cell::sync::Lazy; | ||||
| use std::io::{self, Error, ErrorKind}; | use std::io::{self, Error, ErrorKind}; | ||||
| use tokio::sync::mpsc::UnboundedSender; | use tokio::sync::mpsc::UnboundedSender; | ||||
| use tracing; | use tracing::{self, Level}; | ||||
| use tracing_subscriber::EnvFilter; | |||||
| use tunnelbroker_messages::Messages; | use tunnelbroker_messages::Messages; | ||||
| pub static ACTIVE_CONNECTIONS: Lazy< | pub static ACTIVE_CONNECTIONS: Lazy< | ||||
| DashMap<String, UnboundedSender<Messages>>, | DashMap<String, UnboundedSender<Messages>>, | ||||
| > = Lazy::new(DashMap::new); | > = Lazy::new(DashMap::new); | ||||
| #[tokio::main] | #[tokio::main] | ||||
| async fn main() -> Result<(), io::Error> { | async fn main() -> Result<(), io::Error> { | ||||
| let subscriber = tracing_subscriber::FmtSubscriber::new(); | let filter = EnvFilter::builder() | ||||
| .with_default_directive(Level::INFO.into()) | |||||
| .with_env_var(constants::LOG_LEVEL_ENV_VAR) | |||||
| .from_env_lossy(); | |||||
| let subscriber = tracing_subscriber::fmt().with_env_filter(filter).finish(); | |||||
| tracing::subscriber::set_global_default(subscriber) | tracing::subscriber::set_global_default(subscriber) | ||||
| .expect("Unable to configure tracing"); | .expect("Unable to configure tracing"); | ||||
| let grpc_server = grpc::run_server(); | let grpc_server = grpc::run_server(); | ||||
| let websocket_server = websockets::run_server(); | let websocket_server = websockets::run_server(); | ||||
| tokio::select! { | tokio::select! { | ||||
| Ok(_) = grpc_server => { Ok(()) }, | Ok(_) = grpc_server => { Ok(()) }, | ||||
| Ok(_) = websocket_server => { Ok(()) }, | Ok(_) = websocket_server => { Ok(()) }, | ||||
| else => { | else => { | ||||
| tracing::error!("A grpc or websocket server crashed."); | tracing::error!("A grpc or websocket server crashed."); | ||||
| Err(Error::new(ErrorKind::Other, "A grpc or websocket server crashed.")) | Err(Error::new(ErrorKind::Other, "A grpc or websocket server crashed.")) | ||||
| } | } | ||||
| } | } | ||||
| } | } | ||||