sqlx-exasol-impl 0.9.2-alpha

Driver implementation for sqlx-exasol. Not meant to be used directly.
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

use std::fmt::{Debug, Display};

use async_tungstenite::tungstenite::{protocol::CloseFrame, Error as WsError};
use rsa::errors::Error as RsaError;
use serde_json::error::Error as JsonError;
use thiserror::Error as ThisError;

use crate::SqlxError;

/// Enum representing protocol implementation errors.
#[derive(Debug, ThisError)]
pub enum ExaProtocolError {
    #[error("JSON error: {0}")]
    Json(#[from] JsonError),
    #[error("expected {0} parameter sets; found a mismatch of length {1}")]
    ParameterLengthMismatch(usize, usize),
    #[error("transaction already open")]
    TransactionAlreadyOpen,
    #[error("not ready to send data")]
    SendNotReady,
    #[error("no response received")]
    NoResponse,
    #[error("server closed connection; info: {0}")]
    WebSocketClosed(CloseError),
    #[error("feature 'compression' must be enabled to use compression")]
    CompressionDisabled,
}

#[derive(Debug)]
pub struct CloseError(Option<CloseFrame>);

impl Display for CloseError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match &self.0 {
            Some(c) => write!(f, "{c}"),
            None => write!(f, "unknown reason"),
        }
    }
}

impl From<Option<CloseFrame>> for ExaProtocolError {
    fn from(value: Option<CloseFrame>) -> Self {
        Self::WebSocketClosed(CloseError(value))
    }
}

impl From<ExaProtocolError> for SqlxError {
    fn from(value: ExaProtocolError) -> Self {
        Self::Protocol(value.to_string())
    }
}

/// Helper trait used for converting errors from various underlying libraries to `SQLx`.
pub trait ToSqlxError {
    fn to_sqlx_err(self) -> SqlxError;
}

impl ToSqlxError for WsError {
    fn to_sqlx_err(self) -> SqlxError {
        match self {
            WsError::ConnectionClosed => SqlxError::Protocol(WsError::ConnectionClosed.to_string()),
            WsError::AlreadyClosed => SqlxError::Protocol(WsError::AlreadyClosed.to_string()),
            WsError::Io(e) => SqlxError::Io(e),
            WsError::Tls(e) => SqlxError::Tls(e.into()),
            WsError::Capacity(e) => SqlxError::Protocol(e.to_string()),
            WsError::Protocol(e) => SqlxError::Protocol(e.to_string()),
            WsError::WriteBufferFull(e) => SqlxError::Protocol(e.to_string()),
            WsError::Utf8 => SqlxError::Protocol(WsError::Utf8.to_string()),
            WsError::Url(e) => SqlxError::Configuration(e.into()),
            WsError::Http(r) => SqlxError::Protocol(format!("HTTP error: {}", r.status())),
            WsError::HttpFormat(e) => SqlxError::Protocol(e.to_string()),
            WsError::AttackAttempt => SqlxError::Tls(WsError::AttackAttempt.into()),
        }
    }
}

impl ToSqlxError for RsaError {
    fn to_sqlx_err(self) -> SqlxError {
        SqlxError::Protocol(self.to_string())
    }
}