From c216a0b5c1ff0c2f9017f14751621719064d1b45 Mon Sep 17 00:00:00 2001
From: Mike Dilger <mike@mikedilger.com>
Date: Wed, 4 Jan 2023 17:25:42 +1300
Subject: [PATCH] Fix RIF count

---
 src/fetcher.rs | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/fetcher.rs b/src/fetcher.rs
index 84b527e3..adea39ee 100644
--- a/src/fetcher.rs
+++ b/src/fetcher.rs
@@ -150,16 +150,27 @@ impl Fetcher {
         let maybe_response = client.get(url.inner()).timeout(timeout).send().await;
 
         // Deal with response errors
-        let response = maybe_response?;
+        let response = match maybe_response {
+            Ok(r) => r,
+            Err(e) => {
+                GLOBALS
+                    .fetcher
+                    .requests_in_flight
+                    .fetch_sub(1, Ordering::SeqCst);
+                return Err(e.into());
+            }
+        };
 
         // Convert to bytes
-        let bytes = response.bytes().await?;
+        let maybe_bytes = response.bytes().await;
 
         GLOBALS
             .fetcher
             .requests_in_flight
             .fetch_sub(1, Ordering::SeqCst);
 
+        let bytes = maybe_bytes?;
+
         let cache_file = GLOBALS.fetcher.cache_file(&url);
 
         // Write to the file