fail nicely if %time col has gone missing

jonhoo · Sep 9, 2023 · dd38ed5 · dd38ed5
1 parent e25381e
commit dd38ed5
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/collapse/ghcprof.rs b/src/collapse/ghcprof.rs
@@ -84,7 +84,14 @@ impl Collapse for Folder {
                 //   - BUT it has a max width of 9 whilst its values can exceed (but are always space separted)
                 // "%time %alloc   %time %alloc  ticks  bytes"
                 let source = match self.opt.source {
-                    Source::PercentTime => l.find("%time").unwrap(),
+                    Source::PercentTime => match l.find("%time") {
+                        Some(col_start) => col_start,
+                        _ => {
+                            return invalid_data_error!(
+                                "Expected '%time' column but it was not present"
+                            )
+                        }
+                    },
                     // See note above about ticks and bytes columns
                     Source::Ticks => one_off_end_of_col_before(l.as_ref(), "ticks")?,
                     Source::Bytes => one_off_end_of_col_before(l.as_ref(), "bytes")?,