aion-server 0.1.0

Deployable HTTP, gRPC, WebSocket, and worker endpoint for Aion workflows.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

//! Structured JSON tracing subscriber initialization.

use tracing_subscriber::EnvFilter;

use crate::ServerError;

/// Initialize the process-global tracing subscriber for production server output.
///
/// Events are written as JSON to stdout. `AION_LOG` takes precedence over
/// `RUST_LOG`, and the subscriber falls back to `info` when neither is set.
///
/// # Errors
///
/// Returns [`ServerError`] when the configured filter directive is invalid or
/// another subscriber has already been installed.
pub fn init() -> Result<(), ServerError> {
    let filter = env_filter()?;

    tracing_subscriber::fmt()
        .json()
        .with_env_filter(filter)
        .with_current_span(true)
        .with_span_list(true)
        .with_target(true)
        .with_timer(tracing_subscriber::fmt::time::SystemTime)
        .try_init()
        .map_err(|error| ServerError::Config {
            message: format!("failed to initialize tracing subscriber: {error}"),
        })
}

fn env_filter() -> Result<EnvFilter, ServerError> {
    let directive = std::env::var("AION_LOG")
        .or_else(|_| std::env::var("RUST_LOG"))
        .unwrap_or_else(|_| "info".to_owned());

    EnvFilter::try_new(&directive).map_err(|error| ServerError::Config {
        message: format!("invalid log filter `{directive}`: {error}"),
    })
}