From 4ed42088b74c384531616769d23e5f9c9dbb4661 Mon Sep 17 00:00:00 2001
From: Martin Algesten <martin@algesten.se>
Date: Mon, 23 Aug 2021 21:11:27 +0200
Subject: [PATCH] Add -m and -ct to cureq example

---
 examples/cureq/main.rs | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/examples/cureq/main.rs b/examples/cureq/main.rs
index 0f49f09..24186df 100644
--- a/examples/cureq/main.rs
+++ b/examples/cureq/main.rs
@@ -130,6 +130,8 @@ fn main2() -> Result<(), Error> {
 -d <string>   Use the given data as the request body (useful for POST)
 --wait <n>    Wait n seconds between requests
 -k            Ignore certificate errors
+-m <time>     Max time for the entire request
+-ct <time>    Connection timeout
 
 Fetch url and copy it to stdout.
 "##,
@@ -164,6 +166,24 @@ Fetch url and copy it to stdout.
                     .set_certificate_verifier(Arc::new(AcceptAll {}));
                 builder = builder.tls_config(Arc::new(client_config));
             }
+            "-m" => {
+                let t: f32 = args
+                    .next()
+                    .expect("Timeout arg")
+                    .parse()
+                    .expect("Parse timeout");
+
+                builder = builder.timeout(Duration::from_millis((t * 1000.0) as u64));
+            }
+            "-ct" => {
+                let t: f32 = args
+                    .next()
+                    .expect("Connection timeout arg")
+                    .parse()
+                    .expect("Parse connection timeout");
+
+                builder = builder.timeout_connect(Duration::from_millis((t * 1000.0) as u64));
+            }
             arg => {
                 if arg.starts_with("-") {
                     Err(StringError(format!("unrecognized flag '{}'", arg)))?;