MythEclipse
5859af5294
- Updated all references from `enviroment` to `environment` across the codebase. - Removed the old `enviroment` module and replaced it with a new `environment` module that includes centralized configuration management. - Enhanced OTP generation to include secure hashing and expiration handling. - Improved CSRF token generation and validation with better error handling. - Cleaned up logging statements in various modules for clarity and consistency. - Updated response formatting to include versioning from Cargo.toml. - Removed unused mock test module from utils.
81 lines
2.5 KiB
Rust
81 lines
2.5 KiB
Rust
//! Axum server initialization utilities.
|
|
//!
|
|
//! This module provides utilities for initializing and running an Axum web server
|
|
//! with SurrealDB connections for both WebSocket and in-memory databases.
|
|
|
|
use crate::{surrealdb_init_mem, surrealdb_init_ws, SurrealMemClient, SurrealWsClient};
|
|
use axum::{Router, serve};
|
|
use std::{future::Future, net::SocketAddr};
|
|
use tokio::net::TcpListener;
|
|
use crate::environment::ENV;
|
|
|
|
/// Initialize and start the Axum server with SurrealDB connections.
|
|
///
|
|
/// This function sets up both WebSocket and in-memory SurrealDB connections,
|
|
/// builds the router using the provided function, and starts the server.
|
|
///
|
|
/// # Arguments
|
|
/// * `router_fn` - A function that takes SurrealDB clients and returns a Router
|
|
///
|
|
/// # Panics
|
|
/// This function will panic if:
|
|
/// - SurrealDB initialization fails
|
|
/// - TCP listener binding fails
|
|
///
|
|
/// # Example
|
|
/// ```no_run
|
|
/// use axum::Router;
|
|
/// use imphnen_libs::{axum_init, SurrealWsClient, SurrealMemClient};
|
|
///
|
|
/// async fn create_router(ws: SurrealWsClient, mem: SurrealMemClient) -> Router {
|
|
/// Router::new()
|
|
/// // Add your routes here
|
|
/// }
|
|
///
|
|
/// #[tokio::main]
|
|
/// async fn main() {
|
|
/// axum_init(create_router).await;
|
|
/// }
|
|
/// ```
|
|
pub async fn axum_init<F, Fut>(router_fn: F)
|
|
where
|
|
F: FnOnce(SurrealWsClient, SurrealMemClient) -> Fut,
|
|
Fut: Future<Output = Router>,
|
|
{
|
|
let env = &ENV;
|
|
|
|
// Initialize SurrealDB connections
|
|
log::info!("Initializing SurrealDB connections...");
|
|
let surrealdb_ws = surrealdb_init_ws()
|
|
.await
|
|
.expect("Failed to initialize SurrealDB WebSocket connection");
|
|
|
|
let surrealdb_mem = surrealdb_init_mem()
|
|
.await
|
|
.expect("Failed to initialize SurrealDB in-memory connection");
|
|
|
|
log::info!("SurrealDB connections established successfully");
|
|
|
|
// Build the router
|
|
let router = router_fn(surrealdb_ws, surrealdb_mem).await;
|
|
|
|
// Start the server
|
|
let port = env.port;
|
|
let addr = SocketAddr::from(([0, 0, 0, 0], port));
|
|
log::info!("Starting server on {}", addr);
|
|
|
|
let listener = TcpListener::bind(&addr)
|
|
.await
|
|
.unwrap_or_else(|e| {
|
|
log::error!("Failed to bind to address {}: {}", addr, e);
|
|
panic!("Server binding failed: {}", e);
|
|
});
|
|
|
|
log::info!("Server listening on {}", addr);
|
|
|
|
if let Err(err) = serve(listener, router).await {
|
|
log::error!("Server encountered an error: {}", err);
|
|
panic!("Server failed: {}", err);
|
|
}
|
|
}
|