Post a summary table to PRs after perf. runs

Author: Kobzol

collector/src/category.rs

@@ -0,0 +1,49 @@
+use serde::{Deserialize, Serialize};
+use std::fmt;
+
+#[derive(Debug, Clone, Copy, PartialEq, Serialize, Deserialize, clap::ArgEnum)]
+#[serde(rename_all = "kebab-case")]
+pub enum Category {
+    Primary,
+    Secondary,
+    Stable,
+}
+
+impl Category {
+    pub fn is_stable(self) -> bool {
+        self == Category::Stable
+    }
+
+    pub fn is_primary_or_secondary(self) -> bool {
+        self == Category::Primary || self == Category::Secondary
+    }
+
+    // Within the DB, `Category` is represented in two fields:
+    // - a `supports_stable` bool,
+    // - a `category` which is either "primary" or "secondary".
+    pub fn db_representation(self) -> (bool, String) {
+        match self {
+            Category::Primary => (false, "primary".to_string()),
+            Category::Secondary => (false, "secondary".to_string()),
+            Category::Stable => (true, "primary".to_string()),
+        }
+    }
+
+    pub fn from_db_representation(text: &str) -> anyhow::Result<Self> {
+        match text {
+            "primary" => Ok(Self::Primary),
+            "secondary" => Ok(Self::Secondary),
+            _ => Err(anyhow::anyhow!("Unknown category {text}")),
+        }
+    }
+}
+
+impl fmt::Display for Category {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        match self {
+            Category::Primary => f.write_str("primary"),
+            Category::Secondary => f.write_str("secondary"),
+            Category::Stable => f.write_str("stable"),
+        }
+    }
+}

collector/src/execute.rs

@@ -2,12 +2,12 @@
 
 use crate::{Compiler, Profile, Scenario};
 use anyhow::{bail, Context};
+use collector::category::Category;
 use collector::command_output;
 use collector::etw_parser;
 use database::{PatchName, QueryLabel};
 use futures::stream::FuturesUnordered;
 use futures::stream::StreamExt;
-use serde::{Deserialize, Serialize};
 use std::collections::HashMap;
 use std::env;
 use std::fmt;
@@ -107,45 +107,6 @@ fn touch_all(path: &Path) -> anyhow::Result<()> {
     Ok(())
 }
 
-#[derive(Debug, Clone, Copy, PartialEq, Serialize, Deserialize, clap::ArgEnum)]
-#[serde(rename_all = "kebab-case")]
-pub enum Category {
-    Primary,
-    Secondary,
-    Stable,
-}
-
-impl Category {
-    pub fn is_stable(self) -> bool {
-        self == Category::Stable
-    }
-
-    pub fn is_primary_or_secondary(self) -> bool {
-        self == Category::Primary || self == Category::Secondary
-    }
-
-    // Within the DB, `Category` is represented in two fields:
-    // - a `supports_stable` bool,
-    // - a `category` which is either "primary" or "secondary".
-    pub fn db_representation(self) -> (bool, String) {
-        match self {
-            Category::Primary => (false, "primary".to_string()),
-            Category::Secondary => (false, "secondary".to_string()),
-            Category::Stable => (true, "primary".to_string()),
-        }
-    }
-}
-
-impl fmt::Display for Category {
-    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
-        match self {
-            Category::Primary => f.write_str("primary"),
-            Category::Secondary => f.write_str("secondary"),
-            Category::Stable => f.write_str("stable"),
-        }
-    }
-}
-
 fn default_runs() -> usize {
     3
 }

collector/src/lib.rs

@@ -6,6 +6,7 @@ use std::fmt;
 use std::process::{self, Command};
 
 pub mod api;
+pub mod category;
 pub mod etw_parser;
 mod read2;
 pub mod self_profile;

collector/src/main.rs

@@ -2,6 +2,7 @@
 
 use anyhow::{bail, Context};
 use clap::Parser;
+use collector::category::Category;
 use database::{ArtifactId, Commit};
 use log::debug;
 use std::collections::HashSet;
@@ -18,7 +19,7 @@ use tokio::runtime::Runtime;
 mod execute;
 mod sysroot;
 
-use execute::{BenchProcessor, Benchmark, BenchmarkName, Category, ProfileProcessor, Profiler};
+use execute::{BenchProcessor, Benchmark, BenchmarkName, ProfileProcessor, Profiler};
 use sysroot::Sysroot;
 
 #[derive(Debug, Copy, Clone)]