feat: add worker_job_update api endpoint

DESIGN.md

@@ -50,6 +50,13 @@ Job result:
 1. List of successful builds
 2. Failed package and link to build log (on buildit.aosc.io)
 
+Job status:
+
+1. created: can be assigned to worker
+2. assigned: assigned to worker
+3. error: unexpected error
+4. finished: finished, successful or failed
+
 ## Authentication
 
 Authentication:

server/migrations/2024-03-10-154337_add_error_message_to_jobs/down.sql

@@ -0,0 +1,2 @@
+-- This file should undo anything in `up.sql`
+ALTER TABLE jobs DROP COLUMN error_message;

server/migrations/2024-03-10-154337_add_error_message_to_jobs/up.sql

@@ -0,0 +1,2 @@
+-- Your SQL goes here
+ALTER TABLE jobs ADD COLUMN error_message TEXT;

server/migrations/2024-03-10-155259_add_elapsed_secs_to_jobs/down.sql

@@ -0,0 +1,2 @@
+-- This file should undo anything in `up.sql`
+ALTER TABLE jobs DROP COLUMN elapsed_secs;

server/migrations/2024-03-10-155259_add_elapsed_secs_to_jobs/up.sql

@@ -0,0 +1,2 @@
+-- Your SQL goes here
+ALTER TABLE jobs ADD COLUMN elapsed_secs BIGINT;

server/src/models.rs

@@ -48,6 +48,8 @@ pub struct Job {
     pub skipped_packages: Option<String>,
     pub log_url: Option<String>,
     pub finish_time: Option<chrono::DateTime<chrono::Utc>>,
+    pub error_message: Option<String>,
+    pub elapsed_secs: Option<i64>,
 }
 
 #[derive(Insertable)]

server/src/routes.rs

@@ -189,17 +189,17 @@ pub enum JobResult {
 #[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct JobOk {
     /// Is the build successful?
-    pub success: bool,
+    pub build_success: bool,
     /// List of packages successfully built
     pub successful_packages: Vec<String>,
     /// List of packages failed to build
     pub failed_package: Option<String>,
     /// List of packages skipped
     pub skipped_packages: Vec<String>,
     /// URL to build log
-    pub log: Option<String>,
+    pub log_url: Option<String>,
     /// Elapsed time of the job
-    pub elapsed: Duration,
+    pub elapsed_secs: i64,
     /// If pushpkg succeeded
     pub pushpkg_success: bool,
 }
@@ -216,5 +216,32 @@ pub async fn worker_job_update(
     State(pool): State<DbPool>,
     Json(payload): Json<WorkerJobUpdateRequest>,
 ) -> Result<(), AnyhowError> {
+    let mut conn = pool
+        .get()
+        .context("Failed to get db connection from pool")?;
+
+    use crate::schema::jobs::dsl::*;
+    match payload.result {
+        JobResult::Ok(res) => {
+            diesel::update(jobs.filter(id.eq(payload.job_id)))
+                .set((
+                    status.eq("finished"),
+                    build_success.eq(res.build_success),
+                    pushpkg_success.eq(res.pushpkg_success),
+                    successful_packages.eq(res.successful_packages.join(",")),
+                    failed_package.eq(res.failed_package),
+                    skipped_packages.eq(res.skipped_packages.join(",")),
+                    log_url.eq(res.log_url),
+                    finish_time.eq(chrono::Utc::now()),
+                    elapsed_secs.eq(res.elapsed_secs),
+                ))
+                .execute(&mut conn)?;
+        }
+        JobResult::Error(err) => {
+            diesel::update(jobs.filter(id.eq(payload.job_id)))
+                .set((status.eq("error"), error_message.eq(err)))
+                .execute(&mut conn)?;
+        }
+    }
     Ok(())
 }

server/src/schema.rs

@@ -16,6 +16,8 @@ diesel::table! {
         skipped_packages -> Nullable<Text>,
         log_url -> Nullable<Text>,
         finish_time -> Nullable<Timestamptz>,
+        error_message -> Nullable<Text>,
+        elapsed_secs -> Nullable<Int8>,
     }
 }
 
@@ -47,4 +49,8 @@ diesel::table! {
 
 diesel::joinable!(jobs -> pipelines (pipeline_id));
 
-diesel::allow_tables_to_appear_in_same_query!(jobs, pipelines, workers,);
+diesel::allow_tables_to_appear_in_same_query!(
+    jobs,
+    pipelines,
+    workers,
+);