Merge pull request #886 from rylev/github-perf-labels

Change GitHub comment output depending on perf categorization

Author: rylev

site/src/github.rs

@@ -1,4 +1,5 @@
 use crate::api::{github, ServerResult};
+use crate::comparison::{ComparisonSummary, Direction};
 use crate::load::{Config, InputData, TryCommit};
 use anyhow::Context as _;
 use hashbrown::HashSet;
@@ -631,7 +632,11 @@ pub async fn post_finished(data: &InputData) {
                 "https://perf.rust-lang.org/compare.html?start={}&end={}",
                 commit.parent_sha, commit.sha
             );
-            let summary = categorize_benchmark(&commit, data).await;
+            let (summary, direction) = categorize_benchmark(&commit, data).await;
+            let label = match direction {
+                Some(Direction::Regression | Direction::Mixed) => "+perf-regression",
+                Some(Direction::Improvement) | None => "-perf-regression",
+            };
             post_comment(
                 &data.config,
                 commit.pr,
@@ -651,16 +656,19 @@ non-neutral **do not** roll this PR up -- it will mask other \
 regressions or improvements in the roll up.
 
 @bors rollup=never
-@rustbot label: +S-waiting-on-review -S-waiting-on-perf",
-                    commit.sha, comparison_url, summary
+@rustbot label: +S-waiting-on-review -S-waiting-on-perf {}",
+                    commit.sha, comparison_url, summary, label
                 ),
             )
             .await;
         }
     }
 }
 
-async fn categorize_benchmark(commit: &database::QueuedCommit, data: &InputData) -> String {
+async fn categorize_benchmark(
+    commit: &database::QueuedCommit,
+    data: &InputData,
+) -> (String, Option<Direction>) {
     let comparison = match crate::comparison::compare(
         collector::Bound::Commit(commit.parent_sha.clone()),
         collector::Bound::Commit(commit.sha.clone()),
@@ -670,25 +678,26 @@ async fn categorize_benchmark(commit: &database::QueuedCommit, data: &InputData)
     .await
     {
         Ok(Some(c)) => c,
-        _ => return String::from("ERROR categorizing benchmark run!"),
+        _ => return (String::from("ERROR categorizing benchmark run!"), None),
     };
     const DISAGREEMENT: &str = "If you disagree with this performance assessment, \
     please file an issue in [rust-lang/rustc-perf](https://github.com/rust-lang/rustc-perf/issues/new).";
-    let (summary, direction) =
-        match crate::comparison::ComparisonSummary::summarize_comparison(&comparison) {
-            Some(s) if s.direction().is_some() => {
-                let direction = s.direction().unwrap();
-                (s, direction)
-            }
-            _ => {
-                return format!(
+    let (summary, direction) = match ComparisonSummary::summarize_comparison(&comparison) {
+        Some(s) if s.direction().is_some() => {
+            let direction = s.direction().unwrap();
+            (s, direction)
+        }
+        _ => {
+            return (
+                format!(
                     "This benchmark run did not return any significant changes.\n\n{}",
                     DISAGREEMENT
-                )
-            }
-        };
+                ),
+                None,
+            )
+        }
+    };
 
-    use crate::comparison::Direction;
     let category = match direction {
         Direction::Improvement => "improvements 🎉",
         Direction::Regression => "regressions 😿",
@@ -703,5 +712,5 @@ async fn categorize_benchmark(commit: &database::QueuedCommit, data: &InputData)
         change.summary_line(&mut result, None)
     }
     write!(result, "\n{}", DISAGREEMENT).unwrap();
-    result
+    (result, Some(direction))
 }