postgres_to_polars 0.7.2

A Rust library for direct and performant conversion of PostgreSQL data to Polars dataframes.
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

#[cfg(feature = "execution")]
use postgres_protocol::message::backend;
use std::array::TryFromSliceError;
use thiserror::Error;

use crate::models::postgres_error::PostgresError;

#[derive(Debug, Error)]
pub enum PgToPlError {
    #[error("I/O error {0}")]
    Io(#[from] std::io::Error),
    #[error("Only 1D arrays supported")]
    OnlyOneDimensionArraySupported,
    #[error("Not enough bytes")]
    NotEnoughBytes,
    #[error("Bind error")]
    BindError,
    #[error("Row has fewer fields ({0}) than expected ({1})")]
    TooFewField(usize, usize),
    #[error("Row has more fields than expected ({0})")]
    TooManyField(usize),
    #[error("Ping failed : {0}")]
    PingFailed(String),
    #[error("Connection closed")]
    ConnectionClosed,
    #[error("Pool error {0}")]
    PoolError(String),
    #[error("Parameter type mismatch")]
    ParamTypeMismatch,
    #[error("Query error: {0}")]
    QueryError(PostgresError),
    #[error("Connection broken")]
    ConnectionBroken,
    #[cfg(feature = "execution")]
    #[error("Polars error: {0}")]
    PolarsError(#[from] polars::error::PolarsError),
    #[error("Column type error: {type_name}<{bytes:?}> {error}")]
    ColumnTypeError {
        bytes: Vec<u8>,
        type_name: String,
        error: TryFromSliceError,
    },
    #[error("Timeout")]
    Timeout,
    #[error(
        "Unsupported SASL mechanism: server offered {offered:?}, client supports SCRAM-SHA-256"
    )]
    UnsupportedSaslMechanism { offered: Vec<String> },
    #[error("SASL protocol error: received {message} without prior initialization")]
    SaslStateError { message: &'static str },
    #[error("SASL authentication failed: {0}")]
    SaslAuthenticationFailed(#[source] std::io::Error),
}

pub type PgToPlResult<T> = Result<T, PgToPlError>;

#[cfg(feature = "execution")]
pub trait MessageX {
    fn message_name(&self) -> &'static str;
}

#[cfg(feature = "execution")]
impl MessageX for backend::Message {
    fn message_name(&self) -> &'static str {
        match self {
            backend::Message::AuthenticationCleartextPassword => "AuthenticationCleartextPassword",
            backend::Message::AuthenticationGss => "AuthenticationGss",
            backend::Message::AuthenticationKerberosV5 => "AuthenticationKerberosV5",
            backend::Message::AuthenticationMd5Password(_) => "AuthenticationMd5Password",
            backend::Message::AuthenticationOk => "AuthenticationOk",
            backend::Message::AuthenticationScmCredential => "AuthenticationScmCredential",
            backend::Message::AuthenticationSspi => "AuthenticationSspi",
            backend::Message::AuthenticationGssContinue(_) => "AuthenticationGssContinue",
            backend::Message::AuthenticationSasl(_) => "AuthenticationSasl",
            backend::Message::AuthenticationSaslContinue(_) => "AuthenticationSaslContinue",
            backend::Message::AuthenticationSaslFinal(_) => "AuthenticationSaslFinal",
            backend::Message::BackendKeyData(_) => "BackendKeyData",
            backend::Message::BindComplete => "BindComplete",
            backend::Message::CloseComplete => "CloseComplete",
            backend::Message::CommandComplete(_) => "CommandComplete",
            backend::Message::CopyData(_) => "CopyData",
            backend::Message::CopyDone => "CopyDone",
            backend::Message::CopyInResponse(_) => "CopyInResponse",
            backend::Message::CopyOutResponse(_) => "CopyOutResponse",
            backend::Message::DataRow(_) => "DataRow",
            backend::Message::EmptyQueryResponse => "EmptyQueryResponse",
            backend::Message::ErrorResponse(_) => "ErrorResponse",
            backend::Message::NoData => "NoData",
            backend::Message::NoticeResponse(_) => "NoticeResponse",
            backend::Message::NotificationResponse(_) => "NotificationResponse",
            backend::Message::ParameterDescription(_) => "ParameterDescription",
            backend::Message::ParameterStatus(_) => "ParameterStatus",
            backend::Message::ParseComplete => "ParseComplete",
            backend::Message::PortalSuspended => "PortalSuspended",
            backend::Message::ReadyForQuery(_) => "ReadyForQuery",
            backend::Message::RowDescription(_) => "RowDescription",
            _ => "UnknownMessage",
        }
    }
}