use std::error::Error;
use std::fmt;
#[cfg(feature = "diesel")]
use crate::error::ConstraintViolationType;
use crate::error::{ConstraintViolationError, InternalError, ResourceTemporarilyUnavailableError};
#[derive(Debug)]
pub enum SchemaStoreError {
InternalError(InternalError),
ConstraintViolationError(ConstraintViolationError),
ResourceTemporarilyUnavailableError(ResourceTemporarilyUnavailableError),
NotFoundError(String),
}
impl Error for SchemaStoreError {
fn source(&self) -> Option<&(dyn Error + 'static)> {
match self {
SchemaStoreError::InternalError(err) => Some(err),
SchemaStoreError::ConstraintViolationError(err) => Some(err),
SchemaStoreError::ResourceTemporarilyUnavailableError(err) => Some(err),
SchemaStoreError::NotFoundError(_) => None,
}
}
}
impl fmt::Display for SchemaStoreError {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {
SchemaStoreError::InternalError(err) => err.fmt(f),
SchemaStoreError::ConstraintViolationError(err) => err.fmt(f),
SchemaStoreError::ResourceTemporarilyUnavailableError(err) => err.fmt(f),
SchemaStoreError::NotFoundError(ref s) => write!(f, "Element not found: {}", s),
}
}
}
#[cfg(feature = "diesel")]
impl From<diesel::result::Error> for SchemaStoreError {
fn from(err: diesel::result::Error) -> Self {
match err {
diesel::result::Error::DatabaseError(
diesel::result::DatabaseErrorKind::UniqueViolation,
_,
) => SchemaStoreError::ConstraintViolationError(
ConstraintViolationError::from_source_with_violation_type(
ConstraintViolationType::Unique,
Box::new(err),
),
),
diesel::result::Error::DatabaseError(
diesel::result::DatabaseErrorKind::ForeignKeyViolation,
_,
) => SchemaStoreError::ConstraintViolationError(
ConstraintViolationError::from_source_with_violation_type(
ConstraintViolationType::ForeignKey,
Box::new(err),
),
),
_ => SchemaStoreError::InternalError(InternalError::from_source(Box::new(err))),
}
}
}
#[cfg(feature = "diesel")]
impl From<diesel::r2d2::PoolError> for SchemaStoreError {
fn from(err: diesel::r2d2::PoolError) -> SchemaStoreError {
SchemaStoreError::ResourceTemporarilyUnavailableError(
ResourceTemporarilyUnavailableError::from_source(Box::new(err)),
)
}
}