Store benchmark category in import SQlite

Author: Kobzol

database/src/bin/import-sqlite.rs

@@ -1,4 +1,5 @@
-use database::{Category, Lookup, Pool};
+use database::{BenchmarkData, Lookup, Pool};
+use hashbrown::HashMap;
 use std::collections::HashSet;
 
 #[tokio::main]
@@ -24,6 +25,12 @@ async fn main() {
     let cid = postgres_conn.collection_id(&cid_name).await;
 
     let mut benchmarks = HashSet::new();
+    let benchmark_data: HashMap<String, BenchmarkData> = sqlite_conn
+        .get_benchmarks()
+        .await
+        .into_iter()
+        .map(|benchmark| (benchmark.name.clone(), benchmark))
+        .collect();
 
     // Starting after the sqlite and postgres db args, the rest are artifact
     // names to import.
@@ -41,7 +48,11 @@ async fn main() {
         for &(benchmark, profile, scenario, metric) in sqlite_idx.all_statistic_descriptions() {
             if benchmarks.insert(benchmark) {
                 postgres_conn
-                    .record_benchmark(benchmark.as_str(), None, Category::Secondary)
+                    .record_benchmark(
+                        benchmark.as_str(),
+                        None,
+                        benchmark_data[benchmark.as_str()].category.clone(),
+                    )
                     .await;
             }
 

database/src/lib.rs

@@ -766,8 +766,8 @@ impl fmt::Display for CollectionId {
 
 #[derive(Debug, Clone, Serialize)]
 pub struct BenchmarkData {
-    name: String,
-    category: Category,
+    pub name: String,
+    pub category: Category,
 }
 
 #[derive(Debug, Clone, Serialize, Deserialize)]