Clean up

Author: rylev

analyzeme/src/stack_collapse.rs

@@ -121,6 +121,7 @@ mod test {
         b.interval("Query", "e2", 0, 3, 9, |b| {
             b.interval("Query", "e1", 0, 4, 8, |b| {
                 b.interval("Query", "e3", 0, 5, 7, |_| {});
+                b.integer("ArtifactSize", "e3", 0, 100);
             });
         });
 
@@ -156,9 +157,12 @@ mod test {
 
         b.interval("Query", "e1", 1, 1, 2, |_| {});
         b.interval("Query", "e1", 1, 3, 4, |_| {});
-        b.interval("Query", "e1", 2, 1, 2, |_| {});
+        b.interval("Query", "e1", 2, 1, 2, |b| {
+            b.instant("Instant", "e4", 2, 100);
+        });
         b.interval("Query", "e2", 2, 2, 5, |b| {
             b.interval("Query", "e3", 2, 3, 4, |_| {});
+            b.integer("ArtifactSize", "e4", 2, 1);
         });
 
         let profiling_data = b.into_profiling_data();

summarize/src/diff.rs

@@ -80,8 +80,8 @@ pub fn calculate_diff(base: Results, change: Results) -> DiffResults {
 
             match (b, c) {
                 (Some(b), Some(c)) => c.clone() - b.clone(),
-                (Some(_b), None) => todo!(),
-                (None, Some(_c)) => todo!(),
+                (Some(b), None) => b.invert(),
+                (None, Some(c)) => c.as_artifact_size_diff(),
                 (None, None) => unreachable!(),
             }
         })

summarize/src/query_data.rs

@@ -147,6 +147,20 @@ impl ArtifactSize {
     pub fn new(label: String, value: u64) -> Self {
         Self { label, value }
     }
+
+    pub fn invert(&self) -> ArtifactSizeDiff {
+        ArtifactSizeDiff {
+            label: self.label.clone(),
+            size_change: -(self.value as i64),
+        }
+    }
+
+    pub fn as_artifact_size_diff(&self) -> ArtifactSizeDiff {
+        ArtifactSizeDiff {
+            label: self.label.clone(),
+            size_change: self.value as i64,
+        }
+    }
 }
 
 #[derive(Serialize, Deserialize, Debug)]