use clap::Parser;
use liquid_ml::{error::LiquidError, kv::Key, LiquidML};
use log::Level;
use simple_logger;

/// This is a simple example showing how to load a sor file from disk and
/// distribute it across nodes
#[derive(Parser)]
#[clap(version = "1.0", author = "Samedh G. & Thomas H.")]
struct Opts {
    /// The IP:Port at which the registration server is running
    #[clap(
        short = 's',
        long = "server_addr",
        default_value = "127.0.0.1:9000"
    )]
    server_address: String,
    /// The IP:Port at which this application must run
    #[clap(short = 'm', long = "my_addr", default_value = "127.0.0.2:9002")]
    my_address: String,
}

#[tokio::main]
async fn main() -> Result<(), LiquidError> {
    let opts: Opts = Opts::parse();
    simple_logger::init_with_level(Level::Debug).unwrap();
    let mut app =
        LiquidML::new(&opts.my_address, &opts.server_address, 3).await?;
    app.df_from_sor("tests/distributed.sor", "my-distributed-df")
        .await?;

    let k = Key::new("my-distributed-df", app.node_id);
    let df = app.kv.get(&k).await?;
    println!("{:?}", df.n_rows());
    println!("{}", df);
    app.kill_notifier.notified().await;

    Ok(())
}