Move loading of Submit object for a specific uuid to helper method

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

1 files changed, 6 insertions, 1 deletions

diff --git a/src/db/models/submit.rs b/src/db/models/submit.rs
index cfd29a3..731012a 100644
--- a/src/db/models/submit.rs
+++ b/src/db/models/submit.rs
@@ -43,7 +43,7 @@ impl Submit {
                   requested_package: &Package,
                   repo_hash: &GitHash)
         -> Result<Submit>
-{
+    {
         let tree_json = serde_json::to_value(t)
             .context("Converting tree to JSON string")
             .with_context(|| anyhow!("Tree = {:#?}", t))?;
@@ -63,11 +63,16 @@ impl Submit {
             .execute(database_connection)
             .context("Inserting new submit into submits table")?;
 
+        Self::with_id(database_connection, submit_id)
+    }
+
+    pub fn with_id(database_connection: &PgConnection, submit_id: &::uuid::Uuid) -> Result<Submit> {
         dsl::submits
             .filter(submits::uuid.eq(submit_id))
             .first::<Submit>(database_connection)
             .context("Loading submit")
             .map_err(Error::from)
     }
+
 }