harald/chainerror
1
0
Fork 0
mirror of https://github.com/haraldh/chainerror.git synced 2025-06-03 06:54:36 +02:00
Code Issues Projects Releases Packages Wiki Activity

add ChainError Iterator

Browse source
This commit is contained in:
Harald Hoyer 2019-01-25 13:53:30 +01:00
parent 49515b86f9
commit d14af67560
2 changed files with 91 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

52
src/lib.rs
View file

@ -214,11 +214,7 @@ impl<T: 'static + Display + Debug> ChainError<T> {
/// return the root cause of the error chain, if any exists
pub fn root_cause(&self) -> Option<&(dyn Error + 'static)> {
let mut cause = self as &(dyn Error + 'static);
while let Some(c) = cause.source() {
cause = c;
}
Some(cause)
self.iter().last()
}
/** find the first error cause of type U, if any exists
@ -270,17 +266,7 @@ impl<T: 'static + Display + Debug> ChainError<T> {
~~~
**/
pub fn find_cause<U: Error + 'static>(&self) -> Option<&U> {
let mut cause = self as &(dyn Error + 'static);
loop {
if cause.is::<U>() {
return cause.downcast_ref::<U>();
}
match cause.source() {
Some(c) => cause = c,
None => return None,
}
}
self.iter().filter_map(Error::downcast_ref::<U>()).next()
}
/** find the first error cause of type ChainError<U>, if any exists
@ -299,17 +285,9 @@ impl<T: 'static + Display + Debug> ChainError<T> {
**/
pub fn find_chain_cause<U: Error + 'static>(&self) -> Option<&ChainError<U>> {
let mut cause = self as &(dyn Error + 'static);
loop {
if cause.is::<ChainError<U>>() {
return cause.downcast_ref::<ChainError<U>>();
}
match cause.source() {
Some(c) => cause = c,
None => return None,
}
}
self.iter()
.filter_map(Error::downcast_ref::<ChainError<U>>())
.next()
}
/** return a reference to T of `ChainError<T>`
@ -374,6 +352,26 @@ impl<T: 'static + Display + Debug> ChainError<T> {
pub fn kind(&self) -> &T {
&self.kind
}
pub fn iter(&self) -> impl Iterator<Item = &(dyn Error + 'static)> {
ErrorIter {
current: Some(self),
}
}
}
struct ErrorIter<'a> {
current: Option<&'a (dyn Error + 'static)>,
}
impl<'a> Iterator for ErrorIter<'a> {
type Item = &'a (dyn Error + 'static);
fn next(&mut self) -> Option<Self::Item> {
let current = self.current;
self.current = self.current.and_then(Error::source);
current
}
}
/** convenience trait to hide the `ChainError<T>` implementation internals