mirror of
https://github.com/windmill-labs/windmill.git
synced 2026-09-07 08:02:40 +00:00
wendrul
ac841400ba
* Add spans to show worker_name and job_id Always show worker_name and job_id when relevant, by creating spans that live as long as the worker and the job. * Remove redundant worker names from fn run_worker Remove the now redundant worker_name tracing, as the information now lives at the span level * Use span when spawning new thread * Instrument more pertinent functions Move spans to more pertinent functions. Remove more redundant info, and change worker_name to just worker * Change tracing subscriber to log path:line_no Change from logging target (module name usually) to file path and line number
82 lines
2.3 KiB
Rust
82 lines
2.3 KiB
Rust
/*
|
|
* Author: Ruben Fiszel
|
|
* Copyright: Windmill Labs, Inc 2022
|
|
* This file and its contents are licensed under the AGPLv3 License.
|
|
* Please see the included NOTICE for copyright information and
|
|
* LICENSE-AGPL for a copy of the license.
|
|
*/
|
|
|
|
use tracing_subscriber::{
|
|
fmt::{format, Layer},
|
|
prelude::*,
|
|
EnvFilter,
|
|
};
|
|
|
|
fn json_layer<S>() -> Layer<S, format::JsonFields, format::Format<format::Json>> {
|
|
tracing_subscriber::fmt::layer()
|
|
.json()
|
|
.flatten_event(true)
|
|
.with_span_list(false)
|
|
.with_current_span(true)
|
|
}
|
|
|
|
fn compact_layer<S>() -> Layer<S, format::DefaultFields, format::Format<format::Compact>> {
|
|
tracing_subscriber::fmt::layer().compact()
|
|
}
|
|
|
|
pub fn initialize_tracing() {
|
|
let style = std::env::var("RUST_LOG_STYLE").unwrap_or_else(|_| "auto".into());
|
|
let json_fmt = std::env::var("JSON_FMT")
|
|
.map(|x| x == "true")
|
|
.unwrap_or(false);
|
|
|
|
if std::env::var("RUST_LOG").is_ok_and(|x| x == "debug" || x == "info") {
|
|
std::env::set_var(
|
|
"RUST_LOG",
|
|
&format!("windmill={}", std::env::var("RUST_LOG").unwrap()),
|
|
)
|
|
}
|
|
|
|
let env_filter = EnvFilter::from_default_env();
|
|
|
|
let ts_base = tracing_subscriber::registry().with(env_filter);
|
|
|
|
#[cfg(feature = "loki")]
|
|
let ts_base = {
|
|
let (layer, task) = tracing_loki::builder()
|
|
.build_url(reqwest::Url::parse("http://127.0.0.1:3100").unwrap())
|
|
.expect("build loki url");
|
|
tokio::spawn(task);
|
|
ts_base.with(layer)
|
|
};
|
|
|
|
match json_fmt {
|
|
true => ts_base.with(json_layer().flatten_event(true)).init(),
|
|
false => ts_base
|
|
.with(
|
|
compact_layer()
|
|
.with_ansi(style.to_lowercase() != "never")
|
|
.with_file(true)
|
|
.with_line_number(true)
|
|
.with_target(false),
|
|
)
|
|
.init(),
|
|
}
|
|
}
|
|
|
|
#[cfg(feature = "flamegraph")]
|
|
use tracing_flame::FlameLayer;
|
|
|
|
#[cfg(feature = "flamegraph")]
|
|
pub fn setup_flamegraph() -> impl Drop {
|
|
// let fmt_layer = Layer::default();
|
|
|
|
let (flame_layer, _guard) = FlameLayer::with_file("./tracing.folded").unwrap();
|
|
|
|
tracing_subscriber::registry()
|
|
// .with(fmt_layer)
|
|
.with(flame_layer)
|
|
.init();
|
|
_guard
|
|
}
|