eldegoss 0.2.2

eldegoss net
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

use clap::Parser;
use color_eyre::Result;
use common_x::signal::shutdown_signal;
use eldegoss::{
    config::Config,
    protocol::Message,
    session::{Session, Subscriber},
    util::Args,
};
use tracing::info;

#[tokio::main(flavor = "multi_thread", worker_threads = 30)]
async fn main() -> Result<()> {
    common_x::log::init_log_filter("info");
    let args = Args::parse();
    let config: Config = common_x::configure::file_config(&args.config)?;

    info!("config: {:#?}", config);

    let mut stats = eldegoss::util::Stats::new(100000);
    let callback = vec![Subscriber::new("topic", move |_net_msg| {
        stats.increment();
    })];

    // for graceful shutdown
    let session = Session::serve(config, callback).await;
    let sender = session.sender();

    let sender_handle = tokio::spawn(async move {
        loop {
            let msg = Message::put("topic", vec![0; 1024]);
            sender.send_async(msg).await.ok();
        }
    });
    shutdown_signal().await;
    sender_handle.abort();
    info!("shutdown");
    Ok(())
}