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

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

(limited to 'src/db')

diff --git a/src/db/models/endpoint.rs b/src/db/models/endpoint.rs
index d7a7986..3909b2b 100644
--- a/src/db/models/endpoint.rs
+++ b/src/db/models/endpoint.rs
@@ -43,4 +43,16 @@ impl Endpoint {
             .first::<Endpoint>(database_connection)
             .map_err(Error::from)
     }
+
+    pub fn fetch_for_job(database_connection: &PgConnection, j: &crate::db::models::Job) -> Result<Option<Endpoint>> {
+        Self::fetch_by_id(database_connection, j.endpoint_id)
+    }
+
+    pub fn fetch_by_id(database_connection: &PgConnection, eid: i32) -> Result<Option<Endpoint>> {
+        match dsl::endpoints.filter(id.eq(eid)).first::<Endpoint>(database_connection) {
+            Err(diesel::result::Error::NotFound) => Ok(None),
+            Err(e) => Err(Error::from(e)),
+            Ok(e) => Ok(Some(e)),
+        }
+    }
 }
-- 
cgit v1.2.3