sova-db 0.1.4

SeaORM database plugin for Sova (postgres / sqlite / mysql)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

use sova_core::extend::ErrorResponse;
use sova_core::{Error, IntoResponse, Response};
use sea_orm::{DbErr, SqlErr};

/// Newtype around SeaORM [`DbErr`] mapped to HTTP via [`Error::Response`].
#[derive(Debug)]
pub struct DbError(pub DbErr);

impl From<DbErr> for DbError {
    fn from(err: DbErr) -> Self {
        Self(err)
    }
}

fn map_db_err(err: DbErr) -> Error {
    let (status, public) = if matches!(err, DbErr::RecordNotFound(_)) {
        (404, "Not Found")
    } else if let Some(sql) = err.sql_err() {
        match sql {
            SqlErr::UniqueConstraintViolation(_) => (409, "Conflict"),
            SqlErr::ForeignKeyConstraintViolation(_) => (422, "Unprocessable Entity"),
            _ => match &err {
                DbErr::ConnectionAcquire(_) | DbErr::Conn(_) => (503, "Service Unavailable"),
                _ => (500, "Internal Server Error"),
            },
        }
    } else {
        match &err {
            DbErr::ConnectionAcquire(_) | DbErr::Conn(_) => (503, "Service Unavailable"),
            _ => (500, "Internal Server Error"),
        }
    };
    if status >= 500 {
        tracing::error!(error = %err, "database error");
    }
    Error::Response(Box::new(Response::text(public).status(status)))
}

impl From<DbError> for Error {
    fn from(err: DbError) -> Self {
        map_db_err(err.0)
    }
}

impl IntoResponse for DbError {
    fn into_response(self) -> Response {
        Error::from(self).into_response()
    }
}

impl ErrorResponse for DbError {}