aboutsummaryrefslogtreecommitdiffstats
path: root/src/bin
diff options
context:
space:
mode:
Diffstat (limited to 'src/bin')
-rw-r--r--src/bin/geniza-net.rs9
1 files changed, 5 insertions, 4 deletions
diff --git a/src/bin/geniza-net.rs b/src/bin/geniza-net.rs
index 11bcee8..6d38eae 100644
--- a/src/bin/geniza-net.rs
+++ b/src/bin/geniza-net.rs
@@ -43,7 +43,8 @@ fn run() -> Result<()> {
SubCommand::with_name("receive-some")
.about("Connects to a peer, pulls some metadata and content")
.arg_from_usage("<host_port> 'peer host:port to connect to'")
- .arg_from_usage("<dat_key> 'dat key (public key) to register with'"),
+ .arg_from_usage("<dat_key> 'dat key (public key) to register with'")
+ .arg_from_usage("<count> 'how many entries to pull'"),
)
.subcommand(
SubCommand::with_name("discovery-key")
@@ -87,11 +88,11 @@ fn run() -> Result<()> {
("receive-some", Some(subm)) => {
let host_port = subm.value_of("host_port").unwrap();
let dat_key = subm.value_of("dat_key").unwrap();
+ let count: u64 = subm.value_of("count").unwrap().parse().unwrap();
let key_bytes = parse_dat_key(&dat_key)?;
let mut dc = DatConnection::connect(host_port, &key_bytes, false)?;
- // XXX: number here totally arbitrary
- dc.receive_all(false, 10)?;
- dc.receive_all(true, 10)?;
+ dc.receive_some(false, count)?;
+ dc.receive_some(true, count)?;
println!("Done!");
}
("discovery-key", Some(subm)) => {