trino-rust-client 0.8.0

A trino client library
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

use reqwest::header::HeaderName;
use reqwest::StatusCode;
use serde::Deserialize;
use thiserror::Error;

use crate::models::QueryError;

#[derive(Error, Debug)]
#[allow(clippy::result_large_err)]
pub enum Error {
    #[error("invalid catalog")]
    InvalidCatalog,
    #[error("catalog not found")]
    CatalogNotFound,
    #[error("invalid schema")]
    InvalidSchema,
    #[error("schema not found")]
    SchemaNotFound,
    #[error("schema already exists")]
    SchemaAlreadyExists,
    #[error("invalid source")]
    InvalidSource,
    #[error("invalid user")]
    InvalidUser,
    #[error("invalid properties")]
    InvalidProperties,
    #[error("invalid table property: {0}")]
    InvalidTableProperty(String),
    #[error("table not found")]
    TableNotFound,
    #[error("table already exists")]
    TableAlreadyExists,
    #[error("duplicate header")]
    DuplicateHeader(HeaderName),
    #[error("invalid empty auth")]
    EmptyAuth,
    #[error("forbidden: {message}")]
    Forbidden { message: String },
    #[error("basic auth can not be used with http")]
    BasicAuthWithHttp,
    #[error("http error, reason: {0}")]
    HttpError(#[from] reqwest::Error),
    #[error("http not ok, code: {0}, reason: {1}")]
    HttpNotOk(StatusCode, String),
    #[error("query error, reason: {0}")]
    QueryError(#[from] QueryError),
    #[error("inconsistent data")]
    InconsistentData,
    #[error("empty data")]
    EmptyData,
    #[error("reach max attempt: {0}")]
    ReachMaxAttempt(usize),
    #[error("invalid host: {0}")]
    InvalidHost(String),
    #[error("internal error: {0}")]
    InternalError(String),
}

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

#[derive(Debug, Deserialize)]
pub struct TrinoRetryResult {
    pub id: String,
    #[serde(rename = "infoUri")]
    pub info_uri: String,
    pub stats: TrinoStats,
    pub error: Option<TrinoError>,
    #[serde(rename = "updateType")]
    pub update_type: Option<String>,
    #[serde(rename = "updateCount")]
    pub update_count: Option<u64>,
}

#[derive(Debug, Deserialize)]
pub struct TrinoStats {
    pub state: String,
}

#[derive(Debug, Deserialize)]
pub struct TrinoError {
    pub message: String,
    #[serde(rename = "errorCode")]
    pub error_code: i64,
    #[serde(rename = "errorName")]
    pub error_name: String,
    #[serde(rename = "errorType")]
    pub error_type: String,
    #[serde(rename = "errorLocation")]
    pub error_location: Option<TrinoErrorLocation>,
}

#[derive(Debug, Deserialize)]
pub struct TrinoErrorLocation {
    #[serde(rename = "lineNumber")]
    pub line_number: i64,
    #[serde(rename = "columnNumber")]
    pub column_number: i64,
}

impl From<TrinoError> for Error {
    fn from(error: TrinoError) -> Self {
        match error.error_name.as_str() {
            // CATALOG ERRORS
            "CATALOG_NOT_FOUND" => Error::CatalogNotFound,
            "MISSING_CATALOG_NAME" => Error::InvalidCatalog,

            // SCHEMA ERRORS
            "SCHEMA_NOT_FOUND" => Error::SchemaNotFound,
            "MISSING_SCHEMA_NAME" => Error::InvalidSchema,
            "SCHEMA_ALREADY_EXISTS" => Error::SchemaAlreadyExists,

            // TABLE ERRORS
            "INVALID_TABLE_PROPERTY" => Error::InvalidTableProperty(error.message),
            "TABLE_NOT_FOUND" => Error::TableNotFound,
            "TABLE_ALREADY_EXISTS" => Error::TableAlreadyExists,

            // OTHER ERRORS
            _ => Error::InternalError(format!(
                "Trino error: {} - {}",
                error.error_name, error.message
            )),
        }
    }
}