From 53b4035a5985642b078444f618eb4aa768e3ffc6 Mon Sep 17 00:00:00 2001
From: Shina <53410646+s7tya@users.noreply.github.com>
Date: Sat, 12 Oct 2024 17:10:42 +0900
Subject: [PATCH] Mark the local benchmarks as 'master'

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

diff --git a/collector/src/bin/collector.rs b/collector/src/bin/collector.rs
index 3d8ee7078..9eb8526fb 100644
--- a/collector/src/bin/collector.rs
+++ b/collector/src/bin/collector.rs
@@ -16,6 +16,7 @@ use std::time::Duration;
 use std::{str, time::Instant};
 
 use anyhow::Context;
+use chrono::Utc;
 use clap::builder::TypedValueParser;
 use clap::{Arg, Parser};
 use collector::compare::compare_artifacts;
@@ -731,7 +732,12 @@ fn main_result() -> anyhow::Result<i32> {
 
             let mut rt = build_async_runtime();
             let mut conn = rt.block_on(pool.connection());
-            let artifact_id = ArtifactId::Tag(toolchain.id.clone());
+            let artifact_id = ArtifactId::Commit(Commit {
+                sha: toolchain.id.clone(),
+                date: Utc::now().into(),
+                r#type: CommitType::Master,
+            });
+
             rt.block_on(purge_old_data(conn.as_mut(), &artifact_id, purge.purge));
 
             let runtime_suite = rt.block_on(load_runtime_benchmarks(
@@ -880,7 +886,12 @@ fn main_result() -> anyhow::Result<i32> {
             )?;
             benchmarks.retain(|b| local.category.0.contains(&b.category()));
 
-            let artifact_id = ArtifactId::Tag(toolchain.id.clone());
+            let artifact_id = ArtifactId::Commit(Commit {
+                sha: toolchain.id.clone(),
+                date: Utc::now().into(),
+                r#type: CommitType::Master,
+            });
+
             let mut rt = build_async_runtime();
             let mut conn = rt.block_on(pool.connection());
             rt.block_on(purge_old_data(conn.as_mut(), &artifact_id, purge.purge));