alemi/postwoman
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat: allow specifying --debug to dbg all routes

Browse source
This commit is contained in:
əlemi 2024-10-19 21:57:00 +02:00
parent 68a73284c7
commit 147186f8dc
Signed by: alemi
GPG key ID: A4895B84D311642C
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/main.rs
View file

@ -37,6 +37,10 @@ pub enum PostWomanActions {
/// repeat request N times /// repeat request N times
#[arg(long, default_value_t = 1)] #[arg(long, default_value_t = 1)]
repeat: u32, repeat: u32,
/// force debug extractor on all routes
#[arg(long, default_value_t = false)]
debug: bool,
}, },
/// show all registered routes in current collection /// show all registered routes in current collection
@ -89,13 +93,14 @@ async fn main() -> Result<(), PostWomanError> {
println!("- {name}: \t{} \t{}", endpoint.method.unwrap_or("GET".into()), endpoint.url); println!("- {name}: \t{} \t{}", endpoint.method.unwrap_or("GET".into()), endpoint.url);
} }
}, },
PostWomanActions::Run { query, parallel, repeat } => { PostWomanActions::Run { query, parallel, repeat, debug } => {
let pattern = regex::Regex::new(&query)?; let pattern = regex::Regex::new(&query)?;
let mut joinset = tokio::task::JoinSet::new(); let mut joinset = tokio::task::JoinSet::new();
let client = Arc::new(config.client); let client = Arc::new(config.client);
let env = Arc::new(config.env); let env = Arc::new(config.env);
for (name, endpoint) in config.route { for (name, mut endpoint) in config.route {
if pattern.find(&name).is_some() { if pattern.find(&name).is_some() {
if debug { endpoint.extract = None };
for i in 0..repeat { for i in 0..repeat {
let suffix = if repeat > 1 { let suffix = if repeat > 1 {
format!("#{} ", i+1) format!("#{} ", i+1)