forked from Deuxfleurs/garage
77 lines
1.5 KiB
Rust
77 lines
1.5 KiB
Rust
mod error;
|
|
mod data;
|
|
mod proto;
|
|
mod rpc;
|
|
mod api;
|
|
|
|
use structopt::StructOpt;
|
|
use futures::channel::oneshot;
|
|
use tokio::sync::Mutex;
|
|
use hyper::client::Client;
|
|
|
|
use data::*;
|
|
|
|
|
|
#[derive(StructOpt, Debug)]
|
|
#[structopt(name = "garage")]
|
|
pub struct Opt {
|
|
#[structopt(long = "api-port", default_value = "3900")]
|
|
api_port: u16,
|
|
|
|
#[structopt(long = "rpc-port", default_value = "3901")]
|
|
rpc_port: u16,
|
|
}
|
|
|
|
pub struct System {
|
|
pub opt: Opt,
|
|
|
|
pub rpc_client: Client<hyper::client::HttpConnector, hyper::Body>,
|
|
|
|
pub network_members: Mutex<NetworkMembers>,
|
|
}
|
|
|
|
async fn shutdown_signal(chans: Vec<oneshot::Sender<()>>) {
|
|
// Wait for the CTRL+C signal
|
|
tokio::signal::ctrl_c()
|
|
.await
|
|
.expect("failed to install CTRL+C signal handler");
|
|
for ch in chans {
|
|
ch.send(()).unwrap();
|
|
}
|
|
}
|
|
|
|
async fn wait_from(chan: oneshot::Receiver<()>) -> () {
|
|
chan.await.unwrap()
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let opt = Opt::from_args();
|
|
let rpc_port = opt.rpc_port;
|
|
let api_port = opt.api_port;
|
|
|
|
let sys = System{
|
|
opt,
|
|
rpc_client: Client::new(),
|
|
network_members: Mutex::new(NetworkMembers::default()),
|
|
};
|
|
|
|
let (tx1, rx1) = oneshot::channel();
|
|
let (tx2, rx2) = oneshot::channel();
|
|
|
|
tokio::spawn(shutdown_signal(vec![tx1, tx2]));
|
|
|
|
let rpc_server = rpc::run_rpc_server(&sys, rpc_port, wait_from(rx1));
|
|
let api_server = api::run_api_server(&sys, api_port, wait_from(rx2));
|
|
|
|
let (e1, e2) = futures::join![rpc_server, api_server];
|
|
|
|
if let Err(e) = e1 {
|
|
eprintln!("RPC server error: {}", e)
|
|
}
|
|
|
|
if let Err(e) = e2 {
|
|
eprintln!("API server error: {}", e)
|
|
}
|
|
}
|