systemprompt_logging/
lib.rs

1pub mod extension;
2pub mod layer;
3pub mod models;
4pub mod repository;
5pub mod services;
6pub mod trace;
7
8pub use extension::LoggingExtension;
9
10pub use layer::DatabaseLayer;
11pub use models::{LogEntry, LogFilter, LogLevel};
12pub use repository::{AnalyticsEvent, AnalyticsRepository, LoggingRepository};
13pub use services::{
14    is_startup_mode, publish_log, set_log_publisher, set_startup_mode, CliService,
15    DatabaseLogService, FilterSystemFields, LoggingMaintenanceService, RequestSpan,
16    RequestSpanBuilder, SystemSpan,
17};
18pub use trace::{
19    AiRequestInfo, AiRequestSummary, AiTraceService, ConversationMessage, ExecutionStep,
20    ExecutionStepSummary, McpExecutionSummary, McpToolExecution, TaskArtifact, TaskInfo,
21    ToolLogEntry, TraceEvent, TraceQueryService,
22};
23
24use std::sync::OnceLock;
25
26use systemprompt_database::DbPool;
27use tracing_subscriber::layer::SubscriberExt;
28use tracing_subscriber::util::SubscriberInitExt;
29use tracing_subscriber::{EnvFilter, Layer};
30
31static LOGGING_INITIALIZED: OnceLock<()> = OnceLock::new();
32
33pub fn init_logging(db_pool: DbPool) {
34    if LOGGING_INITIALIZED.set(()).is_err() {
35        return;
36    }
37
38    let console_filter = if is_startup_mode() {
39        EnvFilter::new("warn")
40    } else {
41        EnvFilter::try_from_default_env()
42            .unwrap_or_else(|_| EnvFilter::new("info,tokio_cron_scheduler=warn"))
43    };
44
45    let fmt_layer = tracing_subscriber::fmt::layer()
46        .fmt_fields(FilterSystemFields::new())
47        .with_target(true)
48        .with_writer(std::io::stderr)
49        .with_filter(console_filter);
50
51    let db_layer =
52        DatabaseLayer::new(db_pool).with_filter(tracing_subscriber::filter::LevelFilter::INFO);
53
54    tracing_subscriber::registry()
55        .with(fmt_layer)
56        .with(db_layer)
57        .init();
58}
59
60pub fn init_console_logging() {
61    if LOGGING_INITIALIZED.set(()).is_err() {
62        return; // Already initialized
63    }
64
65    let env_filter = EnvFilter::try_from_default_env()
66        .unwrap_or_else(|_| EnvFilter::new("info,tokio_cron_scheduler=warn"));
67
68    tracing_subscriber::fmt().with_env_filter(env_filter).init();
69}
systemprompt_logging/lib.rs

systemprompt_logging/
lib.rs
