use std::error::Error;
use csv as csv_crate;
#[derive(Debug, Clone, PartialEq)]
pub enum ArrowError {
MemoryError(String),
ParseError(String),
ComputeError(String),
DivideByZero,
CsvError(String),
JsonError(String),
IoError(String),
InvalidArgumentError(String),
}
impl From<::std::io::Error> for ArrowError {
fn from(error: ::std::io::Error) -> Self {
ArrowError::IoError(error.description().to_string())
}
}
impl From<csv_crate::Error> for ArrowError {
fn from(error: csv_crate::Error) -> Self {
match error.kind() {
csv_crate::ErrorKind::Io(error) => {
ArrowError::CsvError(error.description().to_string())
}
csv_crate::ErrorKind::Utf8 { pos: _, err } => ArrowError::CsvError(format!(
"Encountered UTF-8 error while reading CSV file: {:?}",
err.description()
)),
csv_crate::ErrorKind::UnequalLengths {
pos: _,
expected_len,
len,
} => ArrowError::CsvError(format!(
"Encountered unequal lengths between records on CSV file. Expected {} \
records, found {} records",
len, expected_len
)),
_ => ArrowError::CsvError("Error reading CSV file".to_string()),
}
}
}
impl From<::std::string::FromUtf8Error> for ArrowError {
fn from(error: ::std::string::FromUtf8Error) -> Self {
ArrowError::ParseError(error.description().to_string())
}
}
pub type Result<T> = ::std::result::Result<T, ArrowError>;