sagitta 0.1.3

Rust framework for building analytical data services on Arrow Flight and DataFusion.
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

//! Arrow Flight server entry point.

use clap::Parser;
use sagitta::{Config, LoggingConfig, Sagitta};
use tracing_subscriber::{EnvFilter, fmt, layer::SubscriberExt, util::SubscriberInitExt};

/// Arrow Flight SQL server for data infrastructure.
#[derive(Debug, Parser)]
#[command(version, about, long_about = None)]
struct Args {
    /// Path to the configuration file (TOML format).
    #[arg(short, long, value_name = "FILE")]
    config: Option<String>,
}

fn init_logging(config: &LoggingConfig) {
    let filter =
        EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new(&config.level));

    let registry = tracing_subscriber::registry().with(filter);

    if config.format == "json" {
        registry.with(fmt::layer().json()).init();
    } else {
        registry.with(fmt::layer()).init();
    }
}

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

    let config = match args.config {
        Some(path) => Config::from_file(&path)?,
        None => Config::load()?,
    };

    init_logging(&config.logging);

    Sagitta::builder().config(config).serve().await
}