Add custom error type to database module

This commit is contained in:
Elias Projahn 2021-02-04 17:43:36 +01:00
parent 5a41d5008f
commit d7fb996183
11 changed files with 232 additions and 116 deletions

View file

@ -1,6 +1,5 @@
use super::schema::ensembles;
use super::Database;
use anyhow::Result;
use super::{Database, DatabaseResult};
use diesel::prelude::*;
use serde::{Deserialize, Serialize};
@ -14,7 +13,7 @@ pub struct Ensemble {
impl Database {
/// Update an existing ensemble or insert a new one.
pub fn update_ensemble(&self, ensemble: Ensemble) -> Result<()> {
pub fn update_ensemble(&self, ensemble: Ensemble) -> DatabaseResult<()> {
self.defer_foreign_keys()?;
self.connection.transaction(|| {
@ -27,7 +26,7 @@ impl Database {
}
/// Get an existing ensemble.
pub fn get_ensemble(&self, id: &str) -> Result<Option<Ensemble>> {
pub fn get_ensemble(&self, id: &str) -> DatabaseResult<Option<Ensemble>> {
let ensemble = ensembles::table
.filter(ensembles::id.eq(id))
.load::<Ensemble>(&self.connection)?
@ -38,14 +37,14 @@ impl Database {
}
/// Delete an existing ensemble.
pub fn delete_ensemble(&self, id: &str) -> Result<()> {
pub fn delete_ensemble(&self, id: &str) -> DatabaseResult<()> {
diesel::delete(ensembles::table.filter(ensembles::id.eq(id))).execute(&self.connection)?;
Ok(())
}
/// Get all existing ensembles.
pub fn get_ensembles(&self) -> Result<Vec<Ensemble>> {
pub fn get_ensembles(&self) -> DatabaseResult<Vec<Ensemble>> {
let ensembles = ensembles::table.load::<Ensemble>(&self.connection)?;
Ok(ensembles)