Feat - implement queue

database/src/lib.rs

@@ -152,6 +152,16 @@ impl FromStr for CommitType {
     }
 }
 
+impl ToString for CommitType {
+    fn to_string(&self) -> String {
+        match self {
+            CommitType::Try => "try",
+            CommitType::Master => "master",
+        }
+        .to_string()
+    }
+}
+
 #[derive(Debug, Clone, serde::Deserialize, serde::Serialize)]
 pub struct Commit {
     pub sha: String,
@@ -791,3 +801,104 @@ pub struct ArtifactCollection {
     pub duration: Duration,
     pub end_time: DateTime<Utc>,
 }
+
+#[derive(Debug, Clone, serde::Deserialize, serde::Serialize)]
+pub enum CommitJobStatus {
+    Queued,
+    InProgress,
+    Finished,
+}
+
+impl FromStr for CommitJobStatus {
+    type Err = String;
+    fn from_str(s: &str) -> Result<Self, Self::Err> {
+        Ok(match s.to_ascii_lowercase().as_str() {
+            "queued" => CommitJobStatus::Queued,
+            "in_progress" => CommitJobStatus::InProgress,
+            "finished" => CommitJobStatus::Finished,
+            _ => return Err(format!("{} is not a valid `CommitJobStatus`", s)),
+        })
+    }
+}
+
+impl ToString for CommitJobStatus {
+    fn to_string(&self) -> String {
+        match self {
+            CommitJobStatus::Queued => "queued",
+            CommitJobStatus::InProgress => "in_progress",
+            CommitJobStatus::Finished => "finished",
+        }
+        .to_string()
+    }
+}
+
+/// Represents a job in the work queue for collectors
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct CommitJob {
+    pub sha: String,
+    pub parent_sha: Option<String>,
+    pub commit_type: CommitType,
+    pub pr: u32,
+    pub commit_time: Date,
+    pub target: Target,
+    pub machine_id: Option<String>,
+    pub started_at: Option<Date>,
+    pub finished_at: Option<Date>,
+    pub status: CommitJobStatus,
+    pub include: Option<String>,
+    pub exclude: Option<String>,
+    pub runs: Option<i32>,
+    pub backends: Option<String>,
+}
+
+impl CommitJob {
+    pub fn from_db(
+        sha: String,
+        parent_sha: Option<String>,
+        commit_type: CommitType,
+        pr: u32,
+        commit_time: Date,
+        target: Target,
+        machine_id: Option<String>,
+        started_at: Option<Date>,
+        finished_at: Option<Date>,
+        status: CommitJobStatus,
+        include: Option<String>,
+        exclude: Option<String>,
+        runs: Option<i32>,
+        backends: Option<String>,
+    ) -> Self {
+        Self {
+            sha,
+            parent_sha,
+            commit_type,
+            pr,
+            commit_time,
+            target,
+            machine_id,
+            started_at,
+            finished_at,
+            status,
+            include,
+            exclude,
+            runs,
+            backends,
+        }
+    }
+
+    pub fn get_enqueue_column_names() -> Vec<String> {
+        vec![
+            String::from("sha"),
+            String::from("parent_sha"),
+            String::from("commit_type"),
+            String::from("pr"),
+            String::from("commit_time"),
+            String::from("status"),
+            String::from("target"),
+            String::from("include"),
+            String::from("exclude"),
+            String::from("runs"),
+            String::from("backends"),
+        ]
+    }
+}

database/src/pool.rs

@@ -1,5 +1,6 @@
 use crate::{
-    ArtifactCollection, ArtifactId, ArtifactIdNumber, CodegenBackend, CompileBenchmark, Target,
+    ArtifactCollection, ArtifactId, ArtifactIdNumber, CodegenBackend, CommitJob, CompileBenchmark,
+    Target,
 };
 use crate::{CollectionId, Index, Profile, QueuedCommit, Scenario, Step};
 use chrono::{DateTime, Utc};
@@ -178,6 +179,15 @@ pub trait Connection: Send + Sync {
 
     /// Removes all data associated with the given artifact.
     async fn purge_artifact(&self, aid: &ArtifactId);
+
+    /// Add a jobs to the queue
+    async fn enqueue_commit_job(&self, target: Target, jobs: &[CommitJob]);
+
+    /// Dequeue jobs
+    async fn dequeue_commit_job(&self, machine_id: &str, target: Target) -> Option<CommitJob>;
+
+    /// Mark the job as finished
+    async fn finish_commit_job(&self, machine_id: &str, target: Target, sha: String) -> bool;
 }
 
 #[async_trait::async_trait]