init4-bin-base 0.19.0

Internal utilities for binaries produced by the init4 team
//! This example is for testing exporting data to an OTLP collector.
//!
//! It produces
//! - 1 spawn for the lifetime of the program
//! - 1 span every 5 seconds
//! - 1 event every 5 seconds
//!
//! It can be killed via sigint or sigterm

use eyre::WrapErr;
use init4_bin_base::{
    deps::tracing::{info, info_span},
    utils::{from_env::FromEnv, metrics::MetricsConfig, tracing::TracingConfig},
    Init4Config,
};
use std::sync::{
    atomic::{AtomicBool, Ordering},
    Arc,
};

#[derive(Debug, FromEnv)]
struct Config {
    tracing: TracingConfig,
    metrics: MetricsConfig,
}

impl Init4Config for Config {
    fn tracing(&self) -> &TracingConfig {
        &self.tracing
    }
    fn metrics(&self) -> &MetricsConfig {
        &self.metrics
    }
}

#[tokio::main]
async fn main() -> eyre::Result<()> {
    let term: Arc<AtomicBool> = Default::default();
    signal_hook::flag::register(signal_hook::consts::SIGTERM, Arc::clone(&term))
        .wrap_err("failed to register SIGTERM hook")?;
    signal_hook::flag::register(signal_hook::consts::SIGINT, Arc::clone(&term))
        .wrap_err("failed to register SIGINT hook")?;

    let _config_and_guard = init4_bin_base::init::<Config>()?;
    let mut counter = 0;
    let _outer = info_span!("outer span").entered();

    while !term.load(Ordering::Relaxed) {
        let _inner = info_span!("inner span").entered();

        tokio::time::sleep(std::time::Duration::from_secs(5)).await;

        counter += 1;
        info!(counter, "this is an event");
    }

    info!("signal received, shutting down");

    Ok(())
}