Merge pull request 'network-bandwidth-short' (#1) from network-bandwidth-short into master

Reviewed-on: #1
This commit is contained in:
Gregory Ballantine 2022-02-25 18:08:12 -05:00
commit a315e2662c
3 changed files with 38 additions and 2 deletions

View File

@ -14,4 +14,5 @@ path = "src/nettest.rs"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
chrono = "0.4.19"
clap = { version = "3.1.2", features = ["derive"] }

3
bin/docker-build.sh Normal file
View File

@ -0,0 +1,3 @@
#!/bin/sh
docker run --rm --user "$(id -u)":"$(id -g)" -v "$PWD":/usr/src/hardware-tests -w /usr/src/hardware-tests rust:1.59 cargo build --release

View File

@ -1,5 +1,6 @@
use chrono::prelude::*;
use clap::{Parser, Subcommand};
use std::process;
use std::{fs,process};
#[derive(Parser)]
#[clap(name = "Bit Goblin Network Tester", author, version, about = "Network testing app.", long_about = None)]
@ -16,6 +17,14 @@ enum Commands {
#[clap(default_value_t = String::from("8.8.8.8"))]
host: String
},
// bandwidth test subcommand
Bandwidth {
#[clap(default_value_t = String::from("https://www.bitgoblin.tech/hardware-tests/export-01.mp4"))]
download: String,
#[clap(default_value_t = String::from("./tempfile"))]
output: String,
},
}
fn main() {
@ -23,7 +32,8 @@ fn main() {
// map subcommands back to the main command
match &cli.command {
Commands::Ping { host } => ping_host(host)
Commands::Ping { host } => ping_host(host),
Commands::Bandwidth { download, output } => bandwidth_test(download, output)
}
}
@ -44,3 +54,25 @@ fn ping_host(host: &str) {
// print out the ping results from stdout
println!("{}", String::from_utf8_lossy(&output.stdout));
}
// timed file copy test to guage bandwidth speeds
fn bandwidth_test(download: &str, output: &str) {
println!("Testing network bandwidth by downloading {}", download, output);
// get start time so we can track how long it takes to complete
let start_time = Utc::now();
// do the download
// get finish time
let finish_time = Utc::now();
// compute time to complete
let comp_time = finish_time - start_time;
println!("{}", comp_time.num_milliseconds());
// clean up the test file
match fs::remove_file(output) {
Ok(()) => println!("Cleaning up..."),
Err(e) => println!("There was a problem during cleanup - {}", e),
}
}