Return job ID from enqueue_benchmark_job

Author: Kobzol

database/src/pool.rs

@@ -214,15 +214,15 @@ pub trait Connection: Send + Sync {
         commit_date: DateTime<Utc>,
     ) -> anyhow::Result<()>;
 
-    /// Add a benchmark job to the job queue.
+    /// Add a benchmark job to the job queue and return its ID.
     async fn enqueue_benchmark_job(
         &self,
         request_tag: &str,
         target: Target,
         backend: CodegenBackend,
         profile: Profile,
         benchmark_set: u32,
-    ) -> anyhow::Result<()>;
+    ) -> anyhow::Result<u32>;
 
     /// Returns a set of compile-time benchmark test cases that were already computed for the
     /// given artifact.

database/src/pool/postgres.rs

@@ -1637,9 +1637,10 @@ where
         backend: CodegenBackend,
         profile: Profile,
         benchmark_set: u32,
-    ) -> anyhow::Result<()> {
-        self.conn()
-            .execute(
+    ) -> anyhow::Result<u32> {
+        let row = self
+            .conn()
+            .query_one(
                 r#"
             INSERT INTO job_queue(
                 request_tag,
@@ -1651,6 +1652,7 @@ where
             )
             VALUES ($1, $2, $3, $4, $5, $6)
             ON CONFLICT DO NOTHING
+            RETURNING job_queue.id
                 "#,
                 &[
                     &request_tag,
@@ -1663,7 +1665,7 @@ where
             )
             .await
             .context("failed to insert benchmark_job")?;
-        Ok(())
+        Ok(row.get::<_, i32>(0) as u32)
     }
 
     async fn get_compile_test_cases_with_measurements(

database/src/pool/sqlite.rs

@@ -1301,7 +1301,7 @@ impl Connection for SqliteConnection {
         _backend: CodegenBackend,
         _profile: Profile,
         _benchmark_set: u32,
-    ) -> anyhow::Result<()> {
+    ) -> anyhow::Result<u32> {
         no_queue_implementation_abort!()
     }