simple-pub-sub 0.1.0-rc1

simple message broker
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

pub mod client_handler;
pub mod message;
pub mod topics;
use log::info;
use std::env;
use std::error::Error;
use tokio::net::TcpListener;

use clap::Parser;

/// Simple pub-sub
#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
pub struct Args {
    /// host name
    #[arg(short, long, default_value = "0.0.0.0")]
    pub host: String,

    /// port
    #[arg(short, long, default_value_t = 6480)]
    pub port: u16,

    /// log_level
    #[arg(short, long, default_value = "trace")]
    pub log_level: Option<String>,
}

#[tokio::main]
async fn main() -> Result<(), Box<dyn Error>> {
    let levels: Vec<String> = vec![
        "trace".to_string(),
        "info".to_string(),
        "debug".to_string(),
        "error".to_string(),
    ];

    let args = Args::parse();
    let addr = format!("{}:{}", args.host, args.port);
    if levels.contains(&args.log_level.clone().unwrap()) {
        env::set_var("RUST_LOG", args.log_level.clone().unwrap());
        env_logger::init();
        info!("using log_level: {}", args.log_level.clone().unwrap());
    }
    let listener = TcpListener::bind(&addr).await?;
    info!("Listening on: {}", addr);
    info!("getting global broadcaster");

    let tx = topics::get_global_broadcaster();

    let _topic_handler = tokio::spawn(topics::topic_manager(tx.clone()));

    loop {
        let (socket, addr) = listener.accept().await?;
        info!("addr is: {addr}");
        client_handler::handle_clinet(socket, tx.clone()).await;
    }
}