about summary refs log blame commit diff
path: root/users/tazjin/finito/finito-postgres/src/error.rs
blob: e130d18361f1f277a6d6ab40ac06da487e200a1b (plain) (tree)
1
2
3
4
5
6
7
8
9



                                                                      
             
               
                                  

                   
                               

                                            


                                              
                












                                                               


                                                 






                                         









                                                             


                                                                   















                                                           




















                                                 





                                          


                                                            
                                                           


                                                                
                                                            





                                              
//! This module defines error types and conversions for issue that can
//! occur while dealing with persisted state machines.

use std::result;
use std::fmt;
use uuid::Uuid;
use std::error::Error as StdError;

// errors to chain:
use postgres::Error as PgError;
use r2d2_postgres::r2d2::Error as PoolError;
use serde_json::Error as JsonError;

pub type Result<T> = result::Result<T, Error>;

#[derive(Debug)]
pub struct Error {
    pub kind: ErrorKind,
    pub context: Option<String>,
}

#[derive(Debug)]
pub enum ErrorKind {
    /// Errors occuring during JSON serialization of FSM types.
    Serialization(String),

    /// Errors occuring during communication with the database.
    Database(String),

    /// Errors with the database connection pool.
    DBPool(String),

    /// State machine could not be found.
    FSMNotFound(Uuid),

    /// Action could not be found.
    ActionNotFound(Uuid),
}

impl fmt::Display for Error {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        use ErrorKind::*;
        let msg = match &self.kind {
            Serialization(err) =>
                format!("JSON serialization error: {}", err),

            Database(err) =>
                format!("PostgreSQL error: {}", err),

            DBPool(err) =>
                format!("Database connection pool error: {}", err),

            FSMNotFound(id) =>
                format!("FSM with ID {} not found", id),

            ActionNotFound(id) =>
                format!("Action with ID {} not found", id),
        };

        match &self.context {
            None => write!(f, "{}", msg),
            Some(ctx) => write!(f, "{}: {}", ctx, msg),
        }
    }
}

impl StdError for Error {}

impl <E: Into<ErrorKind>> From<E> for Error {
    fn from(err: E) -> Error {
        Error {
            kind: err.into(),
            context: None,
        }
    }
}

impl From<JsonError> for ErrorKind {
    fn from(err: JsonError) -> ErrorKind {
        ErrorKind::Serialization(err.to_string())
    }
}

impl From<PgError> for ErrorKind {
    fn from(err: PgError) -> ErrorKind {
        ErrorKind::Database(err.to_string())
    }
}

impl From<PoolError> for ErrorKind {
    fn from(err: PoolError) -> ErrorKind {
        ErrorKind::DBPool(err.to_string())
    }
}

/// Helper trait that makes it possible to supply contextual
/// information with an error.
pub trait ResultExt<T> {
    fn context<C: fmt::Display>(self, ctx: C) -> Result<T>;
}

impl <T, E: Into<Error>> ResultExt<T> for result::Result<T, E> {
    fn context<C: fmt::Display>(self, ctx: C) -> Result<T> {
        self.map_err(|err| Error {
            context: Some(format!("{}", ctx)),
            .. err.into()
        })
    }
}