pyth-lazer-agent 0.11.1

Pyth Lazer Agent
Documentation 
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use pyth_lazer_agent::{config, http_server, lazer_publisher::LazerPublisher};
use {
    anyhow::Context,
    clap::Parser,
    serde::Deserialize,
    tracing::{info, level_filters::LevelFilter},
    tracing_subscriber::{EnvFilter, fmt::format::FmtSpan},
};

#[derive(Parser, Deserialize)]
#[command(version)]
struct Cli {
    #[clap(short, long, default_value = "config/config.toml")]
    config: String,
    #[clap(short, long, default_value = "json")]
    log_format: LogFormat,
}

#[derive(clap::ValueEnum, Clone, Deserialize, Default)]
enum LogFormat {
    #[default]
    Json,
    Compact,
    Pretty,
}

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let args = Cli::parse();
    init_tracing_subscriber(args.log_format);

    let config =
        config::load_config(args.config.to_string()).context("Failed to read config file")?;
    info!(?config, "starting lazer-agent");

    let lazer_publisher = LazerPublisher::new(&config).await;
    http_server::run(config, lazer_publisher).await?;

    Ok(())
}

fn init_tracing_subscriber(log_format: LogFormat) {
    #[allow(
        clippy::expect_used,
        reason = "application can fail on invalid RUST_LOG"
    )]
    let subscriber = tracing_subscriber::fmt()
        .with_env_filter(
            EnvFilter::builder()
                .with_default_directive(LevelFilter::INFO.into())
                .from_env()
                .expect("invalid RUST_LOG env var"),
        )
        .with_span_events(FmtSpan::NONE);

    match log_format {
        LogFormat::Json => {
            subscriber.json().with_span_list(false).init();
        }
        LogFormat::Compact => {
            subscriber.compact().init();
        }
        LogFormat::Pretty => {
            subscriber.pretty().init();
        }
    }
}