Move create-or-fetch logic into functions on model types

Signed-off-by: Matthias Beyer <mail@beyermatthias.de>

1 files changed, 25 insertions, 1 deletions

diff --git a/src/db/models/image.rs b/src/db/models/image.rs
index bb050c3..32623d5 100644
--- a/src/db/models/image.rs
+++ b/src/db/models/image.rs
@@ -1,3 +1,11 @@
+use std::ops::Deref;
+
+use anyhow::Error;
+use anyhow::Result;
+use diesel::PgConnection;
+use diesel::prelude::*;
+
+use crate::schema::images::*;
 use crate::schema::images;
 
 #[derive(Queryable)]
@@ -8,7 +16,23 @@ pub struct Image {
 
 #[derive(Insertable)]
 #[table_name="images"]
-pub struct NewImage<'a> {
+struct NewImage<'a> {
     pub name: &'a str,
 }
 
+impl Image {
+    pub fn create_or_fetch(database_connection: &PgConnection, image_name: &str) -> Result<Image> {
+        let new_image = NewImage { name: image_name };
+
+        diesel::insert_into(images::table)
+            .values(&new_image)
+            .on_conflict_do_nothing()
+            .execute(database_connection)?;
+
+        dsl::images
+            .filter(name.eq(image_name))
+            .first::<Image>(database_connection)
+            .map_err(Error::from)
+    }
+}
+