Skip to content

Commit 9807ab5

Browse files
Resolve PR comments
1 parent b4241f6 commit 9807ab5

File tree

4 files changed

+24
-24
lines changed

4 files changed

+24
-24
lines changed

src/error.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
use thiserror::Error;
22

33
#[derive(Error, Debug)]
4-
pub enum RequestSchedulerError {
4+
pub enum PostSchedulerError {
55
#[error("Invalid value \"{value}\" for argument {argument_name}. {message}")]
66
InvalidArgument {
77
argument_name: String,

src/lib.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,9 @@ mod data;
1313
pub mod error;
1414
mod requests;
1515

16-
use crate::requests::{send_data, RequestSchedulerBuilder};
16+
use crate::requests::{send_data, PostSchedulerBuilder};
1717
use clap::Parser;
18-
use error::RequestSchedulerError;
18+
use error::PostSchedulerError;
1919
use log::debug;
2020
use requests::MAX_NUM_THREADS;
2121
use std::time::Duration;
@@ -32,21 +32,21 @@ const URL: &str = "http://localhost:8000/api/readings/add";
3232
long_about = "This application emulates a real set of hardware sensors that can report on environmental conditions such as temperature, pressure, humidity, etc."
3333
)]
3434
pub struct Cli {
35-
/// Turns verbose console debug output on
35+
/// Turns verbose console debug output on.
3636
#[arg(short, long)]
3737
pub debug: bool,
38-
/// The amount of requests to make
38+
/// The number of sensor readings to post.
3939
/// [DEFAULT: 1]
4040
#[arg(short = 'n', long)]
4141
pub request_amount: Option<u32>,
42-
/// The time between each request (in seconds)
42+
/// The time between each sensor reading post (in seconds).
4343
/// [DEFAULT: 10]
4444
#[arg(short = 't', long = "time-per-request")]
4545
pub time_per_request_s: Option<u64>,
46-
/// The total time over which all the requests must be sent (in seconds, alternative to -t)
46+
/// The total time over which all the sensor reading posts must be sent (in seconds, alternative to -t).
4747
#[arg(short = 'T', long = "total-time")]
4848
pub total_time_s: Option<u64>,
49-
/// The number of threads to spawn. The workload will be cloned to each thread, not divided
49+
/// The number of threads to spawn. The workload will be cloned to each thread, not divided.
5050
/// [DEFAULT: 1]
5151
#[arg(short = 'p', long, value_parser = is_valid_num_of_threads)]
5252
pub num_threads: Option<u32>,
@@ -66,10 +66,10 @@ fn is_valid_num_of_threads(val: &str) -> Result<u32, String> {
6666
}
6767
}
6868

69-
pub fn run(cli: &Cli) -> Result<(), RequestSchedulerError> {
69+
pub fn run(cli: &Cli) -> Result<(), PostSchedulerError> {
7070
debug!("cli: {cli:?}");
7171

72-
let req_scheduler = RequestSchedulerBuilder::default()
72+
let req_scheduler = PostSchedulerBuilder::default()
7373
.with_some_request_amount(&cli.request_amount)
7474
.with_some_time_per_request(&cli.time_per_request_s.map(Duration::from_secs))
7575
.with_some_total_time(&cli.total_time_s.map(Duration::from_secs))

src/main.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,11 @@
1010
//! See the `LICENSE` file for Copyright and license details.
1111
//!
1212
13-
use ambi_mock_client::error::RequestSchedulerError;
13+
use ambi_mock_client::error::PostSchedulerError;
1414
use clap::Parser;
1515
use log::LevelFilter;
1616

17-
fn main() -> Result<(), RequestSchedulerError> {
17+
fn main() -> Result<(), PostSchedulerError> {
1818
// Parses the provided command line interface arguments and flags
1919
let cli = ambi_mock_client::Cli::parse();
2020

src/requests.rs

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ use crate::{
99
random_gen_dust_concentration, random_gen_humidity, random_gen_pressure,
1010
random_gen_temperature, AirPurity, Reading,
1111
},
12-
error::RequestSchedulerError,
12+
error::PostSchedulerError,
1313
URL,
1414
};
1515

@@ -19,16 +19,16 @@ const DEFAULT_NUM_THREADS: u32 = 1;
1919
pub const MAX_NUM_THREADS: u32 = 10;
2020

2121
#[derive(Clone, Copy)]
22-
pub struct RequestSchedulerBuilder {
22+
pub struct PostSchedulerBuilder {
2323
request_amount: Option<u32>,
2424
time_per_request: Option<Duration>,
2525
total_time: Option<Duration>,
2626
num_threads: Option<u32>,
2727
}
2828

29-
impl RequestSchedulerBuilder {
29+
impl PostSchedulerBuilder {
3030
pub fn default() -> Self {
31-
RequestSchedulerBuilder {
31+
PostSchedulerBuilder {
3232
request_amount: None,
3333
time_per_request: None,
3434
total_time: None,
@@ -56,7 +56,7 @@ impl RequestSchedulerBuilder {
5656
self
5757
}
5858

59-
pub fn build(self) -> Result<RequestScheduler, RequestSchedulerError> {
59+
pub fn build(self) -> Result<PostScheduler, PostSchedulerError> {
6060
// Loop indefinitely if no req amt is set. If time per req is also not set then don't loop.
6161
let loop_indefinitely = self.request_amount.is_none() && self.time_per_request.is_some();
6262

@@ -73,7 +73,7 @@ impl RequestSchedulerBuilder {
7373
// At this point we know that the number of threads is in [1, `MAX_NUM_THREADS`].
7474
// Validation is done in `lib::is_valid_num_of_threads`.
7575

76-
Ok(RequestScheduler {
76+
Ok(PostScheduler {
7777
request_amount,
7878
time_per_request,
7979
num_threads,
@@ -83,14 +83,14 @@ impl RequestSchedulerBuilder {
8383
}
8484

8585
#[derive(Clone, Copy)]
86-
pub struct RequestScheduler {
86+
pub struct PostScheduler {
8787
request_amount: u32,
8888
time_per_request: Duration,
8989
num_threads: u32,
9090
loop_indefinitely: bool,
9191
}
9292

93-
pub fn send_data(req_scheduler: RequestScheduler) {
93+
pub fn send_data(req_scheduler: PostScheduler) {
9494
// If 1 thread is specified, we can use the current thread.
9595
if req_scheduler.num_threads == 1 {
9696
debug!("num_threads is set to 1, use current thread.");
@@ -126,7 +126,7 @@ pub fn send_data(req_scheduler: RequestScheduler) {
126126
///
127127
/// You can log the thread id by prepending `[Thread {thread_id}]: ` to your logs.
128128
fn send_data_internal(
129-
req_scheduler: RequestScheduler,
129+
req_scheduler: PostScheduler,
130130
thread_id: u32,
131131
client: Client,
132132
rng: &mut ThreadRng,
@@ -205,13 +205,13 @@ fn generate_random_reading(rng: &mut ThreadRng) -> String {
205205

206206
#[cfg(test)]
207207
mod tests {
208-
use super::{RequestSchedulerBuilder, MAX_NUM_THREADS};
208+
use super::{PostSchedulerBuilder, MAX_NUM_THREADS};
209209

210210
// TODO: Add some tests for the parameter logic.
211211

212212
#[test]
213213
fn test_invalid_num_threads_low() {
214-
let req_scheduler = RequestSchedulerBuilder::default()
214+
let req_scheduler = PostSchedulerBuilder::default()
215215
.with_some_num_threads(&Some(0))
216216
.build();
217217

@@ -220,7 +220,7 @@ mod tests {
220220

221221
#[test]
222222
fn test_invalid_num_threads_high() {
223-
let req_scheduler = RequestSchedulerBuilder::default()
223+
let req_scheduler = PostSchedulerBuilder::default()
224224
.with_some_num_threads(&Some(MAX_NUM_THREADS + 1))
225225
.build();
226226

0 commit comments

Comments
 (0)