use std::ffi::CString;
use std::{error, fmt, result};
use crate::call::RpcStatus;
use crate::grpc_sys::grpc_call_error;
#[cfg(feature = "protobuf-codec")]
use protobuf::ProtobufError;
#[cfg(feature = "protobufv3-codec")]
use protobufv3::Error as ProtobufError;
#[derive(Debug)]
pub enum Error {
Codec(Box<dyn error::Error + Send + Sync>),
CallFailure(grpc_call_error),
RpcFailure(RpcStatus),
RpcFinished(Option<RpcStatus>),
RemoteStopped,
ShutdownFailed,
BindFail(CString),
QueueShutdown,
GoogleAuthenticationFailed,
InvalidMetadata(String),
}
impl fmt::Display for Error {
fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
Error::RpcFailure(s) => {
if s.message().is_empty() {
write!(fmt, "RpcFailure: {}", s.code())
} else {
write!(fmt, "RpcFailure: {} {}", s.code(), s.message())
}
}
other_error => write!(fmt, "{other_error:?}"),
}
}
}
impl error::Error for Error {
fn source(&self) -> Option<&(dyn error::Error + 'static)> {
match *self {
Error::Codec(ref e) => Some(e.as_ref()),
_ => None,
}
}
}
#[cfg(any(feature = "protobuf-codec", feature = "protobufv3-codec"))]
impl From<ProtobufError> for Error {
fn from(e: ProtobufError) -> Error {
Error::Codec(Box::new(e))
}
}
#[cfg(feature = "prost-codec")]
impl From<prost::DecodeError> for Error {
fn from(e: prost::DecodeError) -> Error {
Error::Codec(Box::new(e))
}
}
#[cfg(feature = "prost-codec")]
impl From<prost::EncodeError> for Error {
fn from(e: prost::EncodeError) -> Error {
Error::Codec(Box::new(e))
}
}
pub type Result<T> = result::Result<T, Error>;
#[cfg(all(test, feature = "protobuf-codec"))]
mod tests {
use std::error::Error as StdError;
use protobuf::error::WireError;
use protobuf::ProtobufError;
use super::Error;
#[test]
fn test_convert() {
let error = ProtobufError::WireError(WireError::UnexpectedEof);
let e: Error = error.into();
assert_eq!(e.to_string(), "Codec(WireError(UnexpectedEof))");
assert!(e.source().is_some());
}
}