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
use std::sync::Arc;
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[non_exhaustive]
pub enum Feature {
Snappy,
Brotli,
Gzip,
Lz4,
Zstd,
}
#[derive(Debug, Clone)]
#[non_exhaustive]
pub enum Error {
General(String),
FeatureNotActive(Feature, String),
OutOfSpec(String),
External(String, Arc<dyn std::error::Error + Send + Sync>),
}
impl std::error::Error for Error {}
impl std::fmt::Display for Error {
fn fmt(&self, fmt: &mut std::fmt::Formatter) -> std::fmt::Result {
match self {
Error::General(message) => {
write!(fmt, "{}", message)
}
Error::FeatureNotActive(feature, reason) => {
write!(
fmt,
"The feature \"{:?}\" needs to be active to {}",
feature, reason
)
}
Error::OutOfSpec(message) => {
write!(fmt, "{}", message)
}
Error::External(message, err) => {
write!(fmt, "{}: {}", message, err)
}
}
}
}
#[cfg(feature = "snappy")]
impl From<snap::Error> for Error {
fn from(e: snap::Error) -> Error {
Error::General(format!("underlying snap error: {}", e))
}
}
#[cfg(feature = "lz4_flex")]
impl From<lz4_flex::block::DecompressError> for Error {
fn from(e: lz4_flex::block::DecompressError) -> Error {
Error::General(format!("underlying lz4_flex error: {}", e))
}
}
#[cfg(feature = "lz4_flex")]
impl From<lz4_flex::block::CompressError> for Error {
fn from(e: lz4_flex::block::CompressError) -> Error {
Error::General(format!("underlying lz4_flex error: {}", e))
}
}
impl From<parquet_format_async_temp::thrift::Error> for Error {
fn from(e: parquet_format_async_temp::thrift::Error) -> Error {
Error::General(format!("underlying thrift error: {}", e))
}
}
impl From<std::io::Error> for Error {
fn from(e: std::io::Error) -> Error {
Error::General(format!("underlying IO error: {}", e))
}
}
impl From<std::num::TryFromIntError> for Error {
fn from(e: std::num::TryFromIntError) -> Error {
Error::OutOfSpec(format!("Number must be zero or positive: {}", e))
}
}
impl From<std::array::TryFromSliceError> for Error {
fn from(e: std::array::TryFromSliceError) -> Error {
Error::OutOfSpec(format!("Can't deserialize to parquet native type: {}", e))
}
}
pub type Result<T> = std::result::Result<T, Error>;
macro_rules! general_err {
($fmt:expr) => (Error::General($fmt.to_owned()));
($fmt:expr, $($args:expr),*) => (Error::General(format!($fmt, $($args),*)));
($e:expr, $fmt:expr) => (Error::General($fmt.to_owned(), $e));
($e:ident, $fmt:expr, $($args:tt),*) => (
Error::General(&format!($fmt, $($args),*), $e));
}