use std::fmt;
use crate::proto::error::*;
use thiserror::Error;
#[cfg(feature = "backtrace")]
use crate::proto::{trace, ExtBacktrace};
pub type Result<T> = ::std::result::Result<T, Error>;
#[derive(Debug, Error)]
#[non_exhaustive]
pub enum ErrorKind {
#[error("error recovering from journal: {}", _0)]
Recovery(&'static str),
#[error("wrong insert count: {} expect: {}", got, expect)]
WrongInsertCount {
got: usize,
expect: usize,
},
#[error("proto error: {0}")]
Proto(#[from] ProtoError),
#[cfg(feature = "sqlite")]
#[cfg_attr(docsrs, doc(cfg(feature = "sqlite")))]
#[error("sqlite error: {0}")]
Sqlite(#[from] rusqlite::Error),
#[error("request timed out")]
Timeout,
}
#[derive(Debug, Error)]
pub struct Error {
kind: ErrorKind,
#[cfg(feature = "backtrace")]
backtrack: Option<ExtBacktrace>,
}
impl Error {
pub fn kind(&self) -> &ErrorKind {
&self.kind
}
}
impl fmt::Display for Error {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
cfg_if::cfg_if! {
if #[cfg(feature = "backtrace")] {
if let Some(ref backtrace) = self.backtrack {
fmt::Display::fmt(&self.kind, f)?;
fmt::Debug::fmt(backtrace, f)
} else {
fmt::Display::fmt(&self.kind, f)
}
} else {
fmt::Display::fmt(&self.kind, f)
}
}
}
}
impl From<ErrorKind> for Error {
fn from(kind: ErrorKind) -> Self {
Self {
kind,
#[cfg(feature = "backtrace")]
backtrack: trace!(),
}
}
}
impl From<ProtoError> for Error {
fn from(e: ProtoError) -> Self {
match *e.kind() {
ProtoErrorKind::Timeout => ErrorKind::Timeout.into(),
_ => ErrorKind::from(e).into(),
}
}
}
#[cfg(feature = "sqlite")]
#[cfg_attr(docsrs, doc(cfg(feature = "sqlite")))]
impl From<rusqlite::Error> for Error {
fn from(e: rusqlite::Error) -> Self {
ErrorKind::from(e).into()
}
}