github-personal/firezone
1
0
Fork 0
mirror of https://github.com/outbackdingo/firezone.git synced 2026-01-27 10:18:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(gateway): start telemetry earlier (#7404)

Browse Source 
By removing the use of the `#[tokio::main]`, we can ensure that
telemetry is initialised as early as possible.

---------

Signed-off-by: Thomas Eizinger <thomas@eizinger.io>
This commit is contained in:
Thomas Eizinger
2024-11-28 20:47:11 +00:00
committed by GitHub
parent 1baf669434
commit 78674a8b14
1 changed files with 15 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
rust/gateway/src/main.rs
View File

@@ -32,8 +32,7 @@ mod eventloop;
const ID_PATH: &str = "/var/lib/firezone/gateway_id";
#[tokio::main]
async fn main() {
fn main() {
rustls::crypto::ring::default_provider()
.install_default()
.expect("Calling `install_default` only once per process should always succeed");
@@ -48,19 +47,22 @@ async fn main() {
);
}
// Enforce errors only being printed on a single line using the technique recommended in the anyhow docs:
// https://docs.rs/anyhow/latest/anyhow/struct.Error.html#display-representations
//
// By default, `anyhow` prints a stacktrace when it exits.
// That looks like a "crash" but we "just" exit with a fatal error.
if let Err(e) = try_main(cli, &mut telemetry).await {
tracing::error!(error = anyhow_dyn_err(&e));
telemetry.stop_on_crash().await;
let runtime = tokio::runtime::Runtime::new().expect("Failed to create tokio runtime");
std::process::exit(1);
match runtime.block_on(try_main(cli, &mut telemetry)) {
Ok(()) => runtime.block_on(telemetry.stop()),
Err(e) => {
// Enforce errors only being printed on a single line using the technique recommended in the anyhow docs:
// https://docs.rs/anyhow/latest/anyhow/struct.Error.html#display-representations
//
// By default, `anyhow` prints a stacktrace when it exits.
// That looks like a "crash" but we "just" exit with a fatal error.
tracing::error!(error = anyhow_dyn_err(&e));
runtime.block_on(telemetry.stop_on_crash());
std::process::exit(1);
}
}
telemetry.stop().await
}
async fn try_main(cli: Cli, telemetry: &mut Telemetry) -> Result<()> {