| // Licensed to the Apache Software Foundation (ASF) under one |
| // or more contributor license agreements. See the NOTICE file |
| // distributed with this work for additional information |
| // regarding copyright ownership. The ASF licenses this file |
| // to you under the Apache License, Version 2.0 (the |
| // "License"); you may not use this file except in compliance |
| // with the License. You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, |
| // software distributed under the License is distributed on an |
| // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| // KIND, either express or implied. See the License for the |
| // specific language governing permissions and limitations |
| // under the License. |
| |
| //! Defines `ArrowError` for representing failures in various Arrow operations. |
| use std::fmt::{Debug, Display, Formatter}; |
| use std::io::Write; |
| |
| use csv as csv_crate; |
| use std::error::Error; |
| |
| /// Many different operations in the `arrow` crate return this error type. |
| #[derive(Debug)] |
| pub enum ArrowError { |
| /// Returned when functionality is not yet available. |
| NotYetImplemented(String), |
| ExternalError(Box<dyn Error + Send + Sync>), |
| CastError(String), |
| MemoryError(String), |
| ParseError(String), |
| SchemaError(String), |
| ComputeError(String), |
| DivideByZero, |
| CsvError(String), |
| JsonError(String), |
| IoError(String), |
| InvalidArgumentError(String), |
| ParquetError(String), |
| /// Error during import or export to/from the C Data Interface |
| CDataInterface(String), |
| DictionaryKeyOverflowError, |
| } |
| |
| impl ArrowError { |
| /// Wraps an external error in an `ArrowError`. |
| pub fn from_external_error( |
| error: Box<dyn ::std::error::Error + Send + Sync>, |
| ) -> Self { |
| Self::ExternalError(error) |
| } |
| } |
| |
| impl From<::std::io::Error> for ArrowError { |
| fn from(error: std::io::Error) -> Self { |
| ArrowError::IoError(error.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.to_string()), |
| csv_crate::ErrorKind::Utf8 { pos: _, err } => ArrowError::CsvError(format!( |
| "Encountered UTF-8 error while reading CSV file: {}", |
| err.to_string() |
| )), |
| csv_crate::ErrorKind::UnequalLengths { |
| 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.to_string()) |
| } |
| } |
| |
| impl From<serde_json::Error> for ArrowError { |
| fn from(error: serde_json::Error) -> Self { |
| ArrowError::JsonError(error.to_string()) |
| } |
| } |
| |
| impl<W: Write> From<::std::io::IntoInnerError<W>> for ArrowError { |
| fn from(error: std::io::IntoInnerError<W>) -> Self { |
| ArrowError::IoError(error.to_string()) |
| } |
| } |
| |
| impl Display for ArrowError { |
| fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result { |
| match self { |
| ArrowError::NotYetImplemented(source) => { |
| write!(f, "Not yet implemented: {}", &source) |
| } |
| ArrowError::ExternalError(source) => write!(f, "External error: {}", &source), |
| ArrowError::CastError(desc) => write!(f, "Cast error: {}", desc), |
| ArrowError::MemoryError(desc) => write!(f, "Memory error: {}", desc), |
| ArrowError::ParseError(desc) => write!(f, "Parser error: {}", desc), |
| ArrowError::SchemaError(desc) => write!(f, "Schema error: {}", desc), |
| ArrowError::ComputeError(desc) => write!(f, "Compute error: {}", desc), |
| ArrowError::DivideByZero => write!(f, "Divide by zero error"), |
| ArrowError::CsvError(desc) => write!(f, "Csv error: {}", desc), |
| ArrowError::JsonError(desc) => write!(f, "Json error: {}", desc), |
| ArrowError::IoError(desc) => write!(f, "Io error: {}", desc), |
| ArrowError::InvalidArgumentError(desc) => { |
| write!(f, "Invalid argument error: {}", desc) |
| } |
| ArrowError::ParquetError(desc) => { |
| write!(f, "Parquet argument error: {}", desc) |
| } |
| ArrowError::CDataInterface(desc) => { |
| write!(f, "C Data interface error: {}", desc) |
| } |
| ArrowError::DictionaryKeyOverflowError => { |
| write!(f, "Dictionary key bigger than the key type") |
| } |
| } |
| } |
| } |
| |
| impl Error for ArrowError {} |
| |
| pub type Result<T> = std::result::Result<T, ArrowError>; |