From d3f28d5516c53c74f19f975646333efc660ab3b5 Mon Sep 17 00:00:00 2001
From: Matthias Beyer <matthias.beyer@atos.net>
Date: Thu, 22 Apr 2021 08:28:04 +0200
Subject: Add helper functions to load Package from DB

Signed-off-by: Matthias Beyer <matthias.beyer@atos.net>
---
 src/db/models/package.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/db/models/package.rs b/src/db/models/package.rs
index 6a5b646..88839b6 100644
--- a/src/db/models/package.rs
+++ b/src/db/models/package.rs
@@ -57,4 +57,16 @@ impl Package {
             .first::<Package>(database_connection)
             .map_err(Error::from)
     }
+
+    pub fn fetch_for_job(database_connection: &PgConnection, j: &crate::db::models::Job) -> Result<Option<Package>> {
+        Self::fetch_by_id(database_connection, j.package_id)
+    }
+
+    pub fn fetch_by_id(database_connection: &PgConnection, pid: i32) -> Result<Option<Package>> {
+        match dsl::packages.filter(id.eq(pid)).first::<Package>(database_connection) {
+            Err(diesel::result::Error::NotFound) => Ok(None),
+            Err(e) => Err(Error::from(e)),
+            Ok(p) => Ok(Some(p)),
+        }
+    }
 }
-- 
cgit v1.2.3