sz-orm-sqlx 1.2.0

sqlx adapter: MySQL, PostgreSQL, SQLite backend via sqlx crate
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

//! sqlx 错误到 sz-orm-core DbError 的映射

use sz_orm_core::{DbError, PoolError};

/// 将 sqlx::Error 转换为 DbError
pub fn map_sqlx_error(e: sqlx::Error) -> DbError {
    match e {
        sqlx::Error::Database(db_err) => {
            // 用 code() 和 message() 判断错误类型
            let msg = db_err.message().to_string();
            let code = db_err.code().map(|c| c.into_owned()).unwrap_or_default();
            // PostgreSQL SQLSTATE codes
            if code == "23505"
                || msg.contains("Duplicate entry")
                || msg.contains("unique constraint")
            {
                // 优先映射到细粒度的 UniqueViolation（唯一约束），保留 AlreadyExists 向后兼容
                DbError::UniqueViolation(msg)
            } else if code == "23503" || msg.contains("foreign key constraint") {
                DbError::ForeignKeyViolation(msg)
            } else if code.starts_with("23") || msg.contains("constraint") {
                DbError::ConstraintViolation(msg)
            } else if code.starts_with("42") || msg.contains("syntax") {
                DbError::InvalidInput(msg)
            } else {
                DbError::QueryError(msg)
            }
        }
        sqlx::Error::PoolClosed => {
            DbError::PoolError(PoolError::Internal("sqlx pool closed".to_string()))
        }
        sqlx::Error::PoolTimedOut => DbError::PoolError(PoolError::Timeout),
        sqlx::Error::Io(io) => DbError::IoError(io.to_string()),
        sqlx::Error::Tls(tls) => DbError::ConnectionError(tls.to_string()),
        sqlx::Error::Protocol(p) => DbError::ConnectionError(p.to_string()),
        sqlx::Error::RowNotFound => DbError::NotFound("row not found".to_string()),
        other => DbError::Internal(other.to_string()),
    }
}