bubbles 4.1.0

Bubble integration server for powder diffraction
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

#[macro_use]
extern crate clap;
#[macro_use]
extern crate log;
use crate::dispatcher::Dispatcher;
use crate::params::Params;
use crate::server::{Server, TcpConnect};
use actix::{Actor, Addr, AsyncContext, SyncArbiter, System};
use std::str::FromStr;
use std::sync::Arc;
use std::{fmt, net};
use tokio::net::TcpListener;

pub mod dispatcher;
pub mod notifier;
pub mod session;
pub mod utils;
#[macro_use]
pub mod macros;
pub mod averager;
pub mod codec;
pub mod integrator;
pub mod logger;
pub mod params;
pub mod results;
pub mod server;
pub mod walker;

#[actix::main]
async fn main() {
    let params = Params::from_command_line();
    info!("{} thread(s) will be used for integration", params.threads);
    let params = Arc::new(params);
    let address = params.address();
    let addr = net::SocketAddr::from_str(&address).unwrap();
    let listener = match TcpListener::bind(&addr).await {
        Ok(listener) => listener,
        Err(err) => die!(
            "Failed to start TCP/IP listener at port {}: {}",
            params.port,
            err
        ),
    };
    let waxs_dispatcher = start_dispatcher(params.clone(), IType::WAXS);
    let saxs_dispatcher = start_dispatcher(params.clone(), IType::SAXS);
    let stream = async_stream::stream! {
        while let Ok((st, addr)) = listener.accept().await {
            yield TcpConnect(st, addr);
        }
    };
    Server::create(move |ctx| {
        ctx.add_message_stream(stream);
        Server {
            params,
            waxs_dispatcher,
            saxs_dispatcher,
        }
    });
    info!("Awaiting requests from clients at {}", &address);
    tokio::signal::ctrl_c().await.unwrap();
    info!("Ctrl-C received, shutting down");
    System::current().stop();
}

fn start_dispatcher(params: Arc<Params>, itype: IType) -> Addr<Dispatcher> {
    SyncArbiter::start(1, move || Dispatcher::new(itype, params.clone()))
}

#[derive(PartialEq, Copy, Clone)]
pub enum IType {
    SAXS,
    WAXS,
}

impl fmt::Display for IType {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        match self {
            IType::SAXS => write!(f, "SAXS"),
            IType::WAXS => write!(f, "WAXS"),
        }
    }
}