You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
1.7 KiB
Rust
72 lines
1.7 KiB
Rust
// flotte-user-management server for managing users, roles and permissions
|
|
// Copyright (C) 2020 trivernis
|
|
// See LICENSE for more information
|
|
|
|
use std::error;
|
|
use std::fmt::{self, Display, Formatter};
|
|
|
|
use r2d2::Error;
|
|
use serde_postgres::DeError;
|
|
|
|
#[derive(Debug)]
|
|
pub enum DBError {
|
|
Postgres(PostgresError),
|
|
Pool(r2d2::Error),
|
|
RecordExists,
|
|
RecordDoesNotExist,
|
|
BCryptError,
|
|
DeserializeError(serde_postgres::DeError),
|
|
GenericError(String),
|
|
}
|
|
|
|
impl Display for DBError {
|
|
fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
|
|
write!(f, "{}", self.to_string())
|
|
}
|
|
}
|
|
|
|
impl error::Error for DBError {}
|
|
|
|
impl DBError {
|
|
pub fn to_string(&self) -> String {
|
|
match self {
|
|
DBError::GenericError(g) => g.clone(),
|
|
DBError::RecordExists => "Record exists".to_string(),
|
|
DBError::Postgres(p) => p.to_string(),
|
|
DBError::DeserializeError(de) => de.to_string(),
|
|
DBError::BCryptError => "BCrypt Hash creation error".to_string(),
|
|
DBError::Pool(p) => p.to_string(),
|
|
DBError::RecordDoesNotExist => "Record does not exist".to_string(),
|
|
}
|
|
}
|
|
}
|
|
|
|
pub type DatabaseResult<T> = Result<T, DBError>;
|
|
|
|
impl From<PostgresError> for DBError {
|
|
fn from(other: PostgresError) -> Self {
|
|
Self::Postgres(other)
|
|
}
|
|
}
|
|
|
|
impl From<r2d2::Error> for DBError {
|
|
fn from(other: Error) -> Self {
|
|
Self::Pool(other)
|
|
}
|
|
}
|
|
|
|
impl From<serde_postgres::DeError> for DBError {
|
|
fn from(other: DeError) -> Self {
|
|
Self::DeserializeError(other)
|
|
}
|
|
}
|
|
|
|
impl From<String> for DBError {
|
|
fn from(other: String) -> Self {
|
|
Self::GenericError(other)
|
|
}
|
|
}
|
|
|
|
pub type DatabaseClient = postgres::Client;
|
|
pub type PostgresError = postgres::Error;
|