mirror of https://github.com/poanetwork/hbbft.git
65 lines
1.6 KiB
Rust
65 lines
1.6 KiB
Rust
//! Example of a consensus node that uses the `hbbft::node::Node` struct for
|
|
//! running the distributed consensus state machine.
|
|
|
|
mod network;
|
|
|
|
use std::collections::HashSet;
|
|
use std::net::SocketAddr;
|
|
use std::process;
|
|
use std::vec::Vec;
|
|
|
|
use docopt::Docopt;
|
|
|
|
use crate::network::node::Node;
|
|
|
|
const VERSION: &str = "0.1.0";
|
|
const USAGE: &str = "
|
|
Consensus node example
|
|
|
|
Usage:
|
|
consensus-node --bind-address=<host:port> [--value=VALUE] [--remote-address=<host:port>]...
|
|
consensus-node --help | -h
|
|
consensus-node --version
|
|
";
|
|
|
|
#[derive(Debug)]
|
|
struct Args {
|
|
bind_address: SocketAddr,
|
|
remote_addresses: HashSet<SocketAddr>,
|
|
value: Option<Vec<u8>>,
|
|
}
|
|
|
|
fn parse_args() -> Args {
|
|
let args = Docopt::new(USAGE)
|
|
.unwrap_or_else(|e| e.exit())
|
|
.version(Some(VERSION.to_owned()))
|
|
.parse()
|
|
.unwrap_or_else(|e| e.exit());
|
|
if args.get_count("-h") > 0 || args.get_count("--help") > 0 {
|
|
print!("{}", USAGE);
|
|
process::exit(0);
|
|
}
|
|
Args {
|
|
value: if args.get_count("--value") > 0 {
|
|
Some(args.get_str("--value").as_bytes().to_vec())
|
|
} else {
|
|
None
|
|
},
|
|
bind_address: args.get_str("--bind-address").parse().unwrap(),
|
|
remote_addresses: args
|
|
.get_vec("--remote-address")
|
|
.iter()
|
|
.map(|s| s.parse().unwrap())
|
|
.collect(),
|
|
}
|
|
}
|
|
|
|
pub fn main() {
|
|
env_logger::init();
|
|
let args: Args = parse_args();
|
|
println!("{:?}", args);
|
|
|
|
let node = Node::new(args.bind_address, args.remote_addresses, args.value);
|
|
node.run().expect("Node failed");
|
|
}
|